diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/http/http1/test_assemble.py | 11 | ||||
| -rw-r--r-- | test/http/test_models.py | 11 |
2 files changed, 8 insertions, 14 deletions
diff --git a/test/http/http1/test_assemble.py b/test/http/http1/test_assemble.py index cdc8bda9..2d250909 100644 --- a/test/http/http1/test_assemble.py +++ b/test/http/http1/test_assemble.py @@ -79,15 +79,10 @@ def test_assemble_request_headers(): r = treq(body=b"") r.headers[b"Transfer-Encoding"] = b"chunked" c = _assemble_request_headers(r) - assert b"Content-Length" in c - assert b"Transfer-Encoding" not in c + assert b"Transfer-Encoding" in c assert b"Host" in _assemble_request_headers(treq(headers=Headers())) - assert b"Proxy-Connection" not in _assemble_request_headers( - treq(headers=Headers(Proxy_Connection="42")) - ) - def test_assemble_response_headers(): # https://github.com/mitmproxy/mitmproxy/issues/186 @@ -95,7 +90,3 @@ def test_assemble_response_headers(): r.headers["Transfer-Encoding"] = b"chunked" c = _assemble_response_headers(r) assert b"Transfer-Encoding" in c - - assert b"Proxy-Connection" not in _assemble_response_headers( - tresp(headers=Headers(Proxy_Connection=b"42")) - ) diff --git a/test/http/test_models.py b/test/http/test_models.py index c3ab4d0f..6970a6e4 100644 --- a/test/http/test_models.py +++ b/test/http/test_models.py @@ -442,13 +442,16 @@ class TestHeaders(object): def test_str(self): headers = Headers(Host="example.com") - assert bytes(headers) == "Host: example.com\r\n" + assert bytes(headers) == b"Host: example.com\r\n" headers = Headers([ - ["Host", "example.com"], - ["Accept", "text/plain"] + [b"Host", b"example.com"], + [b"Accept", b"text/plain"] ]) - assert str(headers) == "Host: example.com\r\nAccept: text/plain\r\n" + assert bytes(headers) == b"Host: example.com\r\nAccept: text/plain\r\n" + + headers = Headers() + assert bytes(headers) == b"" def test_setitem(self): headers = Headers() |