class DefaultHeadersTest(unittest.TestCase): def setUp(self): self.dh = DefaultHeaders(get_engine()) self.defaults = {} for k, v in self.dh.headers.iteritems(): self.defaults[k] = [v] def test_process_request(self): req = Request('http://github.com/') req = self.dh.process_request(req) self.assertDictEqual(req.headers, self.defaults) def test_update_headers(self): headers = {'Accept-Language': ['es'], 'Test-Header': ['test']} req = Request('http://github.com/', headers=headers) self.assertDictEqual(req.headers, headers) req = self.dh.process_request(req) self.defaults.update(headers) self.assertDictEqual(req.headers, self.defaults)
def setUp(self): self.dh = DefaultHeaders(get_engine()) self.defaults = {} for k, v in self.dh.headers.iteritems(): self.defaults[k] = [v]