diff options
| author | Maximilian Hils <git@maximilianhils.com> | 2015-09-27 00:49:41 +0200 |
|---|---|---|
| committer | Maximilian Hils <git@maximilianhils.com> | 2015-09-27 00:49:41 +0200 |
| commit | 23d13e4c1282bc46c54222479c3b83032dad3335 (patch) | |
| tree | 051bbe79f815cbdb7b8d2d76ceb0e1e4a31e4dbd /test/http/http1/test_assemble.py | |
| parent | 466888b01a361e46fb3d4e66afa2c6a0fd168c8e (diff) | |
| download | mitmproxy-23d13e4c1282bc46c54222479c3b83032dad3335.tar.gz mitmproxy-23d13e4c1282bc46c54222479c3b83032dad3335.tar.bz2 mitmproxy-23d13e4c1282bc46c54222479c3b83032dad3335.zip | |
test response model, push coverage to 100% branch cov
Diffstat (limited to 'test/http/http1/test_assemble.py')
| -rw-r--r-- | test/http/http1/test_assemble.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/test/http/http1/test_assemble.py b/test/http/http1/test_assemble.py index 460e22c5..ed94292d 100644 --- a/test/http/http1/test_assemble.py +++ b/test/http/http1/test_assemble.py @@ -78,10 +78,19 @@ def test_assemble_request_headers(): # https://github.com/mitmproxy/mitmproxy/issues/186 r = treq(content=b"") r.headers["Transfer-Encoding"] = "chunked" - c = _assemble_request_headers(r) + c = _assemble_request_headers(r.data) assert b"Transfer-Encoding" in c - assert b"host" in _assemble_request_headers(treq(headers=Headers())) + +def test_assemble_request_headers_host_header(): + r = treq() + r.headers = Headers() + c = _assemble_request_headers(r.data) + assert b"host" in c + + r.host = None + c = _assemble_request_headers(r.data) + assert b"host" not in c def test_assemble_response_headers(): |