From 825cec51ff84bcb1e6318f7c4cfeda35860f4d84 Mon Sep 17 00:00:00 2001
From: Pablo Hoffman <pablo@pablohoffman.com>
Date: Fri, 21 Aug 2009 23:23:46 -0300
Subject: [PATCH] fixed bug recently introduced in stats collector closing
 logic, and added unittests

---
 scrapy/stats/collector/__init__.py | 10 ++++------
 scrapy/tests/test_stats.py         | 31 ++++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+), 6 deletions(-)

diff --git a/scrapy/stats/collector/__init__.py b/scrapy/stats/collector/__init__.py
index 2551e35a0..66c23c2b4 100644
--- a/scrapy/stats/collector/__init__.py
+++ b/scrapy/stats/collector/__init__.py
@@ -53,20 +53,18 @@ class StatsCollector(object):
         send_catch_log(stats_domain_opened, domain=domain)
 
     def close_domain(self, domain, reason):
-        if self._dump:
-            log.msg("Dumping stats:\n" + pprint.pformat(self.get_stats(domain)), \
-                domain=domain)
+        send_catch_log(stats_domain_closing, domain=domain, reason=reason)
         stats = self._stats.pop(domain)
         send_catch_log(stats_domain_closed, domain=domain, reason=reason, \
             domain_stats=stats)
+        if self._dump:
+            log.msg("Dumping domain stats:\n" + pprint.pformat(stats), \
+                domain=domain)
 
     def engine_stopped(self):
         if self._dump:
             log.msg("Dumping global stats:\n" + pprint.pformat(self.get_stats()))
 
-    def _start_closing_domain(self, domain, reason):
-        send_catch_log(stats_domain_closing, domain=domain, reason=reason)
-        self.close_domain(domain, reason)
 
 class MemoryStatsCollector(StatsCollector):
 
diff --git a/scrapy/tests/test_stats.py b/scrapy/tests/test_stats.py
index 88a6da076..fa056c2b6 100644
--- a/scrapy/tests/test_stats.py
+++ b/scrapy/tests/test_stats.py
@@ -1,6 +1,9 @@
 import unittest 
 
+from scrapy.xlib.pydispatch import dispatcher
 from scrapy.stats.collector import StatsCollector, DummyStatsCollector
+from scrapy.stats.signals import stats_domain_opened, stats_domain_closing, \
+    stats_domain_closed
 
 class StatsCollectorTest(unittest.TestCase):
 
@@ -45,5 +48,33 @@ class StatsCollectorTest(unittest.TestCase):
         self.assertEqual(stats.get_stats(), {})
         self.assertEqual(stats.get_stats('a'), {})
 
+    def test_signals(self):
+        signals_catched = set()
+
+        def domain_open(domain):
+            assert domain == 'example.com'
+            signals_catched.add(stats_domain_opened)
+        def domain_closing(domain, reason):
+            assert domain == 'example.com'
+            assert reason == 'testing'
+            signals_catched.add(stats_domain_closing)
+        def domain_closed(domain, reason, domain_stats):
+            assert domain == 'example.com'
+            assert reason == 'testing'
+            assert domain_stats == {'test': 1}
+            signals_catched.add(stats_domain_closed)
+
+        dispatcher.connect(domain_open, signal=stats_domain_opened)
+        dispatcher.connect(domain_closing, signal=stats_domain_closing)
+        dispatcher.connect(domain_closed, signal=stats_domain_closed)
+
+        stats = StatsCollector()
+        stats.open_domain('example.com')
+        stats.set_value('test', 1, domain='example.com')
+        stats.close_domain('example.com', 'testing')
+        assert stats_domain_opened in signals_catched
+        assert stats_domain_closing in signals_catched
+        assert stats_domain_closed in signals_catched
+
 if __name__ == "__main__":
     unittest.main()