From 083635ebafe19403cae7945310b13e8acb0b9f4b Mon Sep 17 00:00:00 2001
From: Ismael Carnales <icarnales@gmail.com>
Date: Mon, 7 Sep 2009 11:14:43 -0300
Subject: [PATCH] added depth spider middleware test

---
 scrapy/tests/test_spidermiddleware_depth.py | 53 +++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 scrapy/tests/test_spidermiddleware_depth.py

diff --git a/scrapy/tests/test_spidermiddleware_depth.py b/scrapy/tests/test_spidermiddleware_depth.py
new file mode 100644
index 000000000..580e331e8
--- /dev/null
+++ b/scrapy/tests/test_spidermiddleware_depth.py
@@ -0,0 +1,53 @@
+from unittest import TestCase
+
+from scrapy.contrib.spidermiddleware.depth import DepthMiddleware 
+from scrapy.conf import settings
+from scrapy.http import Response, Request
+from scrapy.spider import BaseSpider
+from scrapy.stats import stats
+
+
+class TestDepthMiddleware(TestCase):
+
+    def setUp(self):
+        settings.disabled = False
+        settings.overrides['DEPTH_LIMIT'] = 1
+        settings.overrides['DEPTH_STATS'] = True
+
+        self.spider = BaseSpider()
+        self.spider.domain_name = 'scrapytest.org'
+
+        stats.open_domain(self.spider.domain_name)
+
+        self.mw = DepthMiddleware()
+        self.assertEquals(stats.get_value('envinfo/request_depth_limit'), 1)
+
+    def test_process_spider_output(self):
+        req = Request('http://scrapytest.org')
+        resp = Response('http://scrapytest.org')
+        resp.request = req
+        result = [Request('http://scrapytest.org')]
+
+        out = list(self.mw.process_spider_output(resp, result, self.spider))
+        self.assertEquals(out, result)
+
+        rdc = stats.get_value('request_depth_count/1',
+                              domain=self.spider.domain_name)
+        self.assertEquals(rdc, 1)
+
+        req.meta['depth'] = 1
+
+        out2 = list(self.mw.process_spider_output(resp, result, self.spider))
+        self.assertEquals(out2, [])
+
+        rdm = stats.get_value('request_depth_max',
+                              domain=self.spider.domain_name)
+        self.assertEquals(rdm, 1)
+ 
+    def tearDown(self):
+        del settings.overrides['DEPTH_LIMIT']
+        del settings.overrides['DEPTH_STATS']
+        settings.disabled = True
+
+        stats.close_domain(self.spider.domain_name, '')
+