diff options
author | Maximilian Hils <git@maximilianhils.com> | 2016-08-23 00:39:30 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-08-23 00:39:30 -0700 |
commit | 06c499f76ebde90cee49871d9c4be1891f53f6c2 (patch) | |
tree | 59a76a43a8e3348f04ce09136c1ed703dde34a3f /test/netlib/http/test_response.py | |
parent | 798ce96bd0d22b642f8508c5d9a8e131bedb4896 (diff) | |
parent | f27028f58e50492c8c8e15e47278bf444d692246 (diff) | |
download | mitmproxy-06c499f76ebde90cee49871d9c4be1891f53f6c2.tar.gz mitmproxy-06c499f76ebde90cee49871d9c4be1891f53f6c2.tar.bz2 mitmproxy-06c499f76ebde90cee49871d9c4be1891f53f6c2.zip |
Merge pull request #1500 from mhils/simple-response-creation
Introduce Response.make for simple response creation
Diffstat (limited to 'test/netlib/http/test_response.py')
-rw-r--r-- | test/netlib/http/test_response.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/test/netlib/http/test_response.py b/test/netlib/http/test_response.py index b3c2f736..c7b1b646 100644 --- a/test/netlib/http/test_response.py +++ b/test/netlib/http/test_response.py @@ -5,6 +5,7 @@ import email import time from netlib.http import Headers +from netlib.http import Response from netlib.http.cookies import CookieAttrs from netlib.tutils import raises, tresp from .test_message import _test_passthrough_attr, _test_decoded_attr @@ -28,6 +29,25 @@ class TestResponseCore(object): response.content = None assert repr(response) == "Response(200 OK, no content)" + def test_make(self): + r = Response.make() + assert r.status_code == 200 + assert r.content == b"" + + Response.make(content=b"foo") + Response.make(content="foo") + with raises(TypeError): + Response.make(content=42) + + r = Response.make(headers=[(b"foo", b"bar")]) + assert r.headers["foo"] == "bar" + + r = Response.make(headers=({"foo": "baz"})) + assert r.headers["foo"] == "baz" + + with raises(TypeError): + Response.make(headers=42) + def test_status_code(self): _test_passthrough_attr(tresp(), "status_code") |