1 files changed, 37 insertions, 0 deletions
diff --git a/netlib/test/test_encoding.py b/netlib/test/test_encoding.py
new file mode 100644
index 00000000..0ff1aad1
--- /dev/null
+++ b/netlib/test/test_encoding.py
@@ -0,0 +1,37 @@
+from netlib import encoding
+
+
+def test_identity():
+    assert b"string" == encoding.decode("identity", b"string")
+    assert b"string" == encoding.encode("identity", b"string")
+    assert not encoding.encode("nonexistent", b"string")
+    assert not encoding.decode("nonexistent encoding", b"string")
+
+
+def test_gzip():
+    assert b"string" == encoding.decode(
+        "gzip",
+        encoding.encode(
+            "gzip",
+            b"string"
+        )
+    )
+    assert encoding.decode("gzip", b"bogus") is None
+
+
+def test_deflate():
+    assert b"string" == encoding.decode(
+        "deflate",
+        encoding.encode(
+            "deflate",
+            b"string"
+        )
+    )
+    assert b"string" == encoding.decode(
+        "deflate",
+        encoding.encode(
+            "deflate",
+            b"string"
+        )[2:-4]
+    )
+    assert encoding.decode("deflate", b"bogus") is None