From 466888b01a361e46fb3d4e66afa2c6a0fd168c8e Mon Sep 17 00:00:00 2001
From: Maximilian Hils <git@maximilianhils.com>
Date: Sat, 26 Sep 2015 20:07:11 +0200
Subject: improve request tests, coverage++

---
 test/http/test_message.py | 136 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 136 insertions(+)
 create mode 100644 test/http/test_message.py

(limited to 'test/http/test_message.py')

diff --git a/test/http/test_message.py b/test/http/test_message.py
new file mode 100644
index 00000000..b0b7e27f
--- /dev/null
+++ b/test/http/test_message.py
@@ -0,0 +1,136 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import, print_function, division
+
+from netlib.http import decoded
+from netlib.tutils import tresp
+
+
+def _test_passthrough_attr(message, attr):
+    def t(self=None):
+        assert getattr(message, attr) == getattr(message.data, attr)
+        setattr(message, attr, "foo")
+        assert getattr(message.data, attr) == "foo"
+    return t
+
+
+def _test_decoded_attr(message, attr):
+    def t(self=None):
+        assert getattr(message, attr) == getattr(message.data, attr).decode("utf8")
+        # Set str, get raw bytes
+        setattr(message, attr, "foo")
+        assert getattr(message.data, attr) == b"foo"
+        # Set raw bytes, get decoded
+        setattr(message.data, attr, b"bar")
+        assert getattr(message, attr) == "bar"
+        # Set bytes, get raw bytes
+        setattr(message, attr, b"baz")
+        assert getattr(message.data, attr) == b"baz"
+
+        # Set UTF8
+        setattr(message, attr, "Non-Autorisé")
+        assert getattr(message.data, attr) == b"Non-Autoris\xc3\xa9"
+        # Don't fail on garbage
+        setattr(message.data, attr, b"foo\xFF\x00bar")
+        assert getattr(message, attr).startswith("foo")
+        assert getattr(message, attr).endswith("bar")
+        # foo.bar = foo.bar should not cause any side effects.
+        d = getattr(message, attr)
+        setattr(message, attr, d)
+        assert getattr(message.data, attr) == b"foo\xFF\x00bar"
+    return t
+
+
+class TestMessage(object):
+
+    def test_init(self):
+        resp = tresp()
+        assert resp.data
+
+    def test_eq_ne(self):
+        resp = tresp(timestamp_start=42, timestamp_end=42)
+        same = tresp(timestamp_start=42, timestamp_end=42)
+        assert resp == same
+        assert not resp != same
+
+        other = tresp(timestamp_start=0, timestamp_end=0)
+        assert not resp == other
+        assert resp != other
+
+        assert resp != 0
+
+    def test_content_length_update(self):
+        resp = tresp()
+        resp.content = b"foo"
+        assert resp.data.content == b"foo"
+        assert resp.headers["content-length"] == "3"
+        resp.content = b""
+        assert resp.data.content == b""
+        assert resp.headers["content-length"] == "0"
+
+    test_content_basic = _test_passthrough_attr(tresp(), "content")
+    test_headers = _test_passthrough_attr(tresp(), "headers")
+    test_timestamp_start = _test_passthrough_attr(tresp(), "timestamp_start")
+    test_timestamp_end = _test_passthrough_attr(tresp(), "timestamp_end")
+
+    test_http_version = _test_decoded_attr(tresp(), "http_version")
+
+
+class TestDecodedDecorator(object):
+
+    def test_simple(self):
+        r = tresp()
+        assert r.content == b"message"
+        assert "content-encoding" not in r.headers
+        assert r.encode("gzip")
+
+        assert r.headers["content-encoding"]
+        assert r.content != b"message"
+        with decoded(r):
+            assert "content-encoding" not in r.headers
+            assert r.content == b"message"
+        assert r.headers["content-encoding"]
+        assert r.content != b"message"
+
+    def test_modify(self):
+        r = tresp()
+        assert "content-encoding" not in r.headers
+        assert r.encode("gzip")
+
+        with decoded(r):
+            r.content = b"foo"
+
+        assert r.content != b"foo"
+        r.decode()
+        assert r.content == b"foo"
+
+    def test_unknown_ce(self):
+        r = tresp()
+        r.headers["content-encoding"] = "zopfli"
+        r.content = b"foo"
+        with decoded(r):
+            assert r.headers["content-encoding"]
+            assert r.content == b"foo"
+        assert r.headers["content-encoding"]
+        assert r.content == b"foo"
+
+    def test_cannot_decode(self):
+        r = tresp()
+        assert r.encode("gzip")
+        r.content = b"foo"
+        with decoded(r):
+            assert r.headers["content-encoding"]
+            assert r.content == b"foo"
+        assert r.headers["content-encoding"]
+        assert r.content != b"foo"
+        r.decode()
+        assert r.content == b"foo"
+
+    def test_cannot_encode(self):
+        r = tresp()
+        assert r.encode("gzip")
+        with decoded(r):
+            r.content = None
+
+        assert "content-encoding" not in r.headers
+        assert r.content is None
+
-- 
cgit v1.2.3


From 23d13e4c1282bc46c54222479c3b83032dad3335 Mon Sep 17 00:00:00 2001
From: Maximilian Hils <git@maximilianhils.com>
Date: Sun, 27 Sep 2015 00:49:41 +0200
Subject: test response model, push coverage to 100% branch cov

---
 test/http/test_message.py | 91 ++++++++++++++++++++++++++++-------------------
 1 file changed, 54 insertions(+), 37 deletions(-)

(limited to 'test/http/test_message.py')

diff --git a/test/http/test_message.py b/test/http/test_message.py
index b0b7e27f..2c37dc3e 100644
--- a/test/http/test_message.py
+++ b/test/http/test_message.py
@@ -1,43 +1,53 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, print_function, division
 
-from netlib.http import decoded
-from netlib.tutils import tresp
+from netlib.http import decoded, Headers
+from netlib.tutils import tresp, raises
 
 
 def _test_passthrough_attr(message, attr):
-    def t(self=None):
-        assert getattr(message, attr) == getattr(message.data, attr)
-        setattr(message, attr, "foo")
-        assert getattr(message.data, attr) == "foo"
-    return t
+    assert getattr(message, attr) == getattr(message.data, attr)
+    setattr(message, attr, "foo")
+    assert getattr(message.data, attr) == "foo"
 
 
 def _test_decoded_attr(message, attr):
-    def t(self=None):
-        assert getattr(message, attr) == getattr(message.data, attr).decode("utf8")
-        # Set str, get raw bytes
-        setattr(message, attr, "foo")
-        assert getattr(message.data, attr) == b"foo"
-        # Set raw bytes, get decoded
-        setattr(message.data, attr, b"bar")
-        assert getattr(message, attr) == "bar"
-        # Set bytes, get raw bytes
-        setattr(message, attr, b"baz")
-        assert getattr(message.data, attr) == b"baz"
-
-        # Set UTF8
-        setattr(message, attr, "Non-Autorisé")
-        assert getattr(message.data, attr) == b"Non-Autoris\xc3\xa9"
-        # Don't fail on garbage
-        setattr(message.data, attr, b"foo\xFF\x00bar")
-        assert getattr(message, attr).startswith("foo")
-        assert getattr(message, attr).endswith("bar")
-        # foo.bar = foo.bar should not cause any side effects.
-        d = getattr(message, attr)
-        setattr(message, attr, d)
-        assert getattr(message.data, attr) == b"foo\xFF\x00bar"
-    return t
+    assert getattr(message, attr) == getattr(message.data, attr).decode("utf8")
+    # Set str, get raw bytes
+    setattr(message, attr, "foo")
+    assert getattr(message.data, attr) == b"foo"
+    # Set raw bytes, get decoded
+    setattr(message.data, attr, b"bar")
+    assert getattr(message, attr) == "bar"
+    # Set bytes, get raw bytes
+    setattr(message, attr, b"baz")
+    assert getattr(message.data, attr) == b"baz"
+
+    # Set UTF8
+    setattr(message, attr, "Non-Autorisé")
+    assert getattr(message.data, attr) == b"Non-Autoris\xc3\xa9"
+    # Don't fail on garbage
+    setattr(message.data, attr, b"foo\xFF\x00bar")
+    assert getattr(message, attr).startswith("foo")
+    assert getattr(message, attr).endswith("bar")
+    # foo.bar = foo.bar should not cause any side effects.
+    d = getattr(message, attr)
+    setattr(message, attr, d)
+    assert getattr(message.data, attr) == b"foo\xFF\x00bar"
+
+
+class TestMessageData(object):
+    def test_eq_ne(self):
+        data = tresp(timestamp_start=42, timestamp_end=42).data
+        same = tresp(timestamp_start=42, timestamp_end=42).data
+        assert data == same
+        assert not data != same
+
+        other = tresp(content=b"foo").data
+        assert not data == other
+        assert data != other
+
+        assert data != 0
 
 
 class TestMessage(object):
@@ -67,12 +77,20 @@ class TestMessage(object):
         assert resp.data.content == b""
         assert resp.headers["content-length"] == "0"
 
-    test_content_basic = _test_passthrough_attr(tresp(), "content")
-    test_headers = _test_passthrough_attr(tresp(), "headers")
-    test_timestamp_start = _test_passthrough_attr(tresp(), "timestamp_start")
-    test_timestamp_end = _test_passthrough_attr(tresp(), "timestamp_end")
+    def test_content_basic(self):
+        _test_passthrough_attr(tresp(), "content")
+
+    def test_headers(self):
+        _test_passthrough_attr(tresp(), "headers")
 
-    test_http_version = _test_decoded_attr(tresp(), "http_version")
+    def test_timestamp_start(self):
+        _test_passthrough_attr(tresp(), "timestamp_start")
+
+    def test_timestamp_end(self):
+        _test_passthrough_attr(tresp(), "timestamp_end")
+
+    def teste_http_version(self):
+        _test_decoded_attr(tresp(), "http_version")
 
 
 class TestDecodedDecorator(object):
@@ -133,4 +151,3 @@ class TestDecodedDecorator(object):
 
         assert "content-encoding" not in r.headers
         assert r.content is None
-
-- 
cgit v1.2.3