aboutsummaryrefslogtreecommitdiffstats
path: root/test/netlib/http/test_headers.py
diff options
context:
space:
mode:
authorMaximilian Hils <git@maximilianhils.com>2016-09-21 19:21:32 -0700
committerMaximilian Hils <git@maximilianhils.com>2016-09-21 19:21:32 -0700
commit770936f1f9b56010011c2af7f5f854d88f1c1d4f (patch)
treeff67e18d3cd927a4f08b4be79b41d9c3033d52b3 /test/netlib/http/test_headers.py
parent1e5a5b03f8d56df62a04a368bd5eb2d59cb7582a (diff)
downloadmitmproxy-770936f1f9b56010011c2af7f5f854d88f1c1d4f.tar.gz
mitmproxy-770936f1f9b56010011c2af7f5f854d88f1c1d4f.tar.bz2
mitmproxy-770936f1f9b56010011c2af7f5f854d88f1c1d4f.zip
raise TypeError on invalid header assignment, fix #1562
Diffstat (limited to 'test/netlib/http/test_headers.py')
-rw-r--r--test/netlib/http/test_headers.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/test/netlib/http/test_headers.py b/test/netlib/http/test_headers.py
index ad2bc548..e8752c52 100644
--- a/test/netlib/http/test_headers.py
+++ b/test/netlib/http/test_headers.py
@@ -43,6 +43,15 @@ class TestHeaders(object):
with raises(TypeError):
Headers([[b"Host", u"not-bytes"]])
+ def test_set(self):
+ headers = Headers()
+ headers[u"foo"] = u"1"
+ headers[b"bar"] = b"2"
+ headers["baz"] = b"3"
+ with raises(TypeError):
+ headers["foobar"] = 42
+ assert len(headers) == 3
+
def test_bytes(self):
headers = Headers(Host="example.com")
assert bytes(headers) == b"Host: example.com\r\n"