def test_sets_headers(self):
agent = _RequestHeaderSettingAgent(
self.agent,
Headers({'X-Test-Header': ['Test-Header-Value']}))
agent.request('method', 'uri')
self.agent.request.assert_called_once_with(
'method', 'uri',
headers=Headers({'X-Test-Header': ['Test-Header-Value']}),
bodyProducer=None
)
def test_sets_headers(self):
agent = _RequestHeaderSettingAgent(
self.agent,
Headers({b'X-Test-Header': [b'Test-Header-Value']}))
agent.request('method', 'uri')
self.agent.request.assert_called_once_with(
'method', 'uri',
headers=Headers({b'X-Test-Header': [b'Test-Header-Value']}),
bodyProducer=None
)
def test_overrides_per_request_headers(self):
agent = _RequestHeaderSettingAgent(
self.agent,
Headers({b'X-Test-Header': [b'Test-Header-Value']})
)
agent.request(
'method', 'uri',
Headers({b'X-Test-Header': [b'Unwanted-Value']})
)
self.agent.request.assert_called_once_with(
'method', 'uri',
headers=Headers({b'X-Test-Header': [b'Test-Header-Value']}),
bodyProducer=None
)
def test_overrides_per_request_headers(self):
agent = _RequestHeaderSettingAgent(
self.agent,
Headers({'X-Test-Header': ['Test-Header-Value']})
)
agent.request(
'method', 'uri',
Headers({'X-Test-Header': ['Unwanted-Value']})
)
self.agent.request.assert_called_once_with(
'method', 'uri',
headers=Headers({'X-Test-Header': ['Test-Header-Value']}),
bodyProducer=None
)