add move tests and code from mitmproxy

1 files changed, 25 insertions, 4 deletions

diff --git a/test/http/test_exceptions.py b/test/http/test_exceptions.py
index aa57f831..0131c7ef 100644
--- a/test/http/test_exceptions.py
+++ b/test/http/test_exceptions.py
@@ -1,6 +1,27 @@
 from netlib.http.exceptions import *
+from netlib import odict
 
-def test_HttpAuthenticationError():
-    x = HttpAuthenticationError({"foo": "bar"})
-    assert str(x)
-    assert "foo" in x.headers
+class TestHttpError:
+    def test_simple(self):
+        e = HttpError(404, "Not found")
+        assert str(e)
+
+class TestHttpAuthenticationError:
+    def test_init(self):
+        headers = odict.ODictCaseless([("foo", "bar")])
+        x = HttpAuthenticationError(headers)
+        assert str(x)
+        assert isinstance(x.headers, odict.ODictCaseless)
+        assert x.code == 407
+        assert x.headers == headers
+        print(x.headers.keys())
+        assert "foo" in x.headers.keys()
+
+    def test_header_conversion(self):
+        headers = {"foo": "bar"}
+        x = HttpAuthenticationError(headers)
+        assert isinstance(x.headers, odict.ODictCaseless)
+        assert x.headers.lst == headers.items()
+
+    def test_repr(self):
+        assert repr(HttpAuthenticationError()) == "Proxy Authentication Required"