def test_set_request_headers_with_host(parse_headers): """ :type parse_headers: mock.MagicMock """ request = mock_http_request() headers = 'Host: localhost:8181' headers_bag = {'Host': 'localhost:8181'} parse_headers.return_value = headers_bag set_request_headers(request, headers) parse_headers.assert_called_once_with(headers) assert request.headers == headers_bag
def test_set_request_headers_without_host_must_raise_bad_request_exception(parse_headers): """ :type parse_headers: mock.MagicMock """ request = mock_http_request() headers = 'User-Agent: client/1.2.3' headers_bag = {'User-Agent': 'client/1.1.3'} parse_headers.return_value = headers_bag with pytest.raises(BadRequestHttpException): set_request_headers(request, headers) parse_headers.assert_called_once_with(headers)