コード例 #1
0
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
コード例 #2
0
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)