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') +