TST small cleanup of a cookie test

This commit is contained in:
Mikhail Korobov 2014-09-21 05:31:34 +06:00
parent c543fe6e4c
commit 49645d4bf9
1 changed files with 7 additions and 6 deletions

View File

@ -8,12 +8,6 @@ from scrapy.contrib.downloadermiddleware.cookies import CookiesMiddleware
class CookiesMiddlewareTest(TestCase):
def test_local_domain(self):
request = Request("http://example-host/", cookies={'currencyCookie': 'USD'})
assert self.mw.process_request(request, self.spider) is None
assert 'Cookie' in request.headers
assert 'currencyCookie' in request.headers['Cookie']
def assertCookieValEqual(self, first, second, msg=None):
cookievaleq = lambda cv: re.split(';\s*', cv)
return self.assertEqual(
@ -153,3 +147,10 @@ class CookiesMiddlewareTest(TestCase):
req6 = Request('file:///scrapy/sometempfile')
assert self.mw.process_request(req6, self.spider) is None
self.assertEquals(req6.headers.get('Cookie'), None)
def test_local_domain(self):
request = Request("http://example-host/", cookies={'currencyCookie': 'USD'})
assert self.mw.process_request(request, self.spider) is None
self.assertIn('Cookie', request.headers)
self.assertIn('currencyCookie', request.headers['Cookie'])