mirror of
https://github.com/scrapy/scrapy.git
synced 2025-02-24 12:03:47 +00:00
22 lines
639 B
Python
22 lines
639 B
Python
from unittest import TestCase
|
|
|
|
from scrapy.http import Response, Request
|
|
from scrapy.spider import Spider
|
|
from scrapy.contrib.spidermiddleware.referer import RefererMiddleware
|
|
|
|
|
|
class TestRefererMiddleware(TestCase):
|
|
|
|
def setUp(self):
|
|
self.spider = Spider('foo')
|
|
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')
|
|
|