From 99f8d5733a1630afd5ddd3c6c6af92905d55bc32 Mon Sep 17 00:00:00 2001
From: Pablo Hoffman <pablo@pablohoffman.com>
Date: Mon, 6 May 2013 20:04:15 -0300
Subject: [PATCH] added mock tests for retries (more to come)

---
 scrapy/tests/test_crawl.py | 35 ++++++++++++++++++++++++++++++++++-
 scrapy/utils/test.py       | 11 +++++++++++
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/scrapy/tests/test_crawl.py b/scrapy/tests/test_crawl.py
index 8125ae17e..eb4e00d75 100644
--- a/scrapy/tests/test_crawl.py
+++ b/scrapy/tests/test_crawl.py
@@ -5,7 +5,7 @@ from subprocess import Popen, PIPE
 from scrapy.spider import BaseSpider
 from scrapy.http import Request
 from scrapy.contrib.linkextractors.sgml import SgmlLinkExtractor
-from scrapy.utils.test import get_crawler, get_testenv
+from scrapy.utils.test import get_crawler, get_testenv, get_testlog
 
 class FollowAllSpider(BaseSpider):
 
@@ -43,6 +43,16 @@ class DelaySpider(BaseSpider):
     def errback(self, failure):
         self.t2_err = time.time()
 
+class SimpleSpider(BaseSpider):
+
+    name = 'simple'
+
+    def __init__(self, url="http://localhost:8998"):
+        self.start_urls = [url]
+
+    def parse(self, response):
+        self.log("Got response %d" % response.status)
+
 def docrawl(spider, settings=None):
     crawler = get_crawler(settings)
     crawler.configure()
@@ -92,3 +102,26 @@ class CrawlTestCase(TestCase):
         self.assertTrue(spider.t2 == 0)
         self.assertTrue(spider.t2_err > 0)
         self.assertTrue(spider.t2_err > spider.t1)
+
+    @defer.inlineCallbacks
+    def test_retry_503(self):
+        spider = SimpleSpider("http://localhost:8998/status?n=503")
+        yield docrawl(spider)
+        self._assert_retried()
+
+    @defer.inlineCallbacks
+    def test_retry_conn_failed(self):
+        spider = SimpleSpider("http://localhost:65432/status?n=503")
+        yield docrawl(spider)
+        self._assert_retried()
+
+    @defer.inlineCallbacks
+    def test_retry_dns_error(self):
+        spider = SimpleSpider("http://localhost666/status?n=503")
+        yield docrawl(spider)
+        self._assert_retried()
+
+    def _assert_retried(self):
+        log = get_testlog()
+        self.assertEqual(log.count("Retrying"), 2)
+        self.assertEqual(log.count("Gave up retrying"), 1)
diff --git a/scrapy/utils/test.py b/scrapy/utils/test.py
index 66e045919..f612f5a70 100644
--- a/scrapy/utils/test.py
+++ b/scrapy/utils/test.py
@@ -75,6 +75,17 @@ def get_testenv():
     env['PYTHONPATH'] = get_pythonpath()
     return env
 
+def get_testlog():
+    """Get Scrapy log of current test, ignoring the rest"""
+    thistest = []
+    loglines = open("test.log").readlines()
+    for l in loglines[::-1]:
+        thistest.append(l)
+        if "[-] -->" in l:
+            break
+    return "".join(thistest[::-1])
+
+
 def assert_samelines(testcase, text1, text2, msg=None):
     """Asserts text1 and text2 have the same lines, ignoring differences in
     line endings between platforms