2 files changed, 32 insertions, 1 deletions
diff --git a/test/test_encoding.py b/test/test_encoding.py
new file mode 100644
index 00000000..ba0755d6
--- /dev/null
+++ b/test/test_encoding.py
@@ -0,0 +1,31 @@
+from libmproxy import encoding
+import libpry
+
+import cStringIO
+import gzip, zlib
+
+class udecode_identity(libpry.AutoTree):
+    def test_decode(self):
+        assert 'string' == encoding.decode('identity', 'string')
+
+    def test_fallthrough(self):
+        assert 'string' == encoding.decode('nonexistent encoding', 'string')
+
+class udecode_gzip(libpry.AutoTree):
+    def test_simple(self):
+        s = cStringIO.StringIO()
+        gf = gzip.GzipFile(fileobj=s, mode='wb')
+        gf.write('string')
+        gf.close()
+        assert 'string' == encoding.decode('gzip', s.getvalue())
+
+class udecode_deflate(libpry.AutoTree):
+    def test_simple(self):
+        assert 'string' == encoding.decode('deflate', zlib.compress('string'))
+        assert 'string' == encoding.decode('deflate', zlib.compress('string')[2:-4])
+
+tests = [
+    udecode_identity(),
+    udecode_gzip(),
+    udecode_deflate()
+]
diff --git a/test/test_utils.py b/test/test_utils.py
index 2b0f4342..2ff951d4 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -98,7 +98,7 @@ class uHeaders(libpry.AutoTree):
         out = repr(self.hd)
         for i in expected:
             assert out.find(i) >= 0
-        
+
     def test_dictToHeader2(self):
         self.hd["one"] = ["uno"]
         expected1 = "one: uno\r\n"