def test_default_headers_useragent(self):
msg = protocol.Request(self.transport, 'GET', '/')
msg._add_default_headers()
headers = [r for r, _ in msg.headers]
self.assertNotIn('SERVER', headers)
self.assertIn('USER-AGENT', headers)
def test_start_request(transport):
msg = protocol.Request(
transport, 'GET', '/index.html', close=True)
assert msg.transport is transport
assert msg.closing
assert msg.status_line == 'GET /index.html HTTP/1.1\r\n'
def test_start_request(self):
msg = protocol.Request(
self.transport, 'GET', '/index.html', close=True)
self.assertIs(msg.transport, self.transport)
self.assertIsNone(msg.status)
self.assertTrue(msg.closing)
self.assertEqual(msg.status_line, 'GET /index.html HTTP/1.1\r\n')
def test_default_headers_useragent_custom(self):
msg = protocol.Request(self.transport, 'GET', '/')
msg.add_headers(('user-agent', 'my custom agent'))
msg._add_default_headers()
headers = [r for r, _ in msg.headers
if r.lower() == 'user-agent']
self.assertEqual(len(headers), 1)
def test_dont_override_request_headers_with_default_values(self):
msg = protocol.Request(self.transport,
'GET',
'/index.html',
close=True)
msg.add_header('USER-AGENT', 'custom')
msg._add_default_headers()
self.assertEqual('custom', msg.headers['USER-AGENT'])
def test_dont_override_request_headers_with_default_values(transport):
msg = protocol.Request(
transport, 'GET', '/index.html', close=True)
msg.add_header('USER-AGENT', 'custom')
msg._add_default_headers()
assert 'custom' == msg.headers['USER-AGENT']
