From 43bd00dea2226ab7fd1ba36476bfea462111bfc3 Mon Sep 17 00:00:00 2001 From: Ismael Carnales Date: Mon, 7 Sep 2009 11:14:46 -0300 Subject: [PATCH] added referer spider middleware test --- scrapy/tests/test_spidermiddleware_referer.py | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 scrapy/tests/test_spidermiddleware_referer.py diff --git a/scrapy/tests/test_spidermiddleware_referer.py b/scrapy/tests/test_spidermiddleware_referer.py new file mode 100644 index 000000000..ec11b0489 --- /dev/null +++ b/scrapy/tests/test_spidermiddleware_referer.py @@ -0,0 +1,21 @@ +from unittest import TestCase + +from scrapy.http import Response, Request +from scrapy.spider import BaseSpider +from scrapy.contrib.spidermiddleware.referer import RefererMiddleware + + +class TestRefererMiddleware(TestCase): + + def setUp(self): + self.spider = BaseSpider() + self.mw = RefererMiddleware() + + def test_process_spider_output(self): + res = Response('http://scrapytest.org') + reqs = [Request('http://scrapytest.org/')] + + out = list(self.mw.process_spider_output(res, reqs, self.spider)) + self.assertEquals(out[0].headers.get('Referer'), + 'http://scrapytest.org') +