mitmproxy/test/test_encoding.py
2011-07-17 20:16:47 -07:00

32 lines
1.0 KiB
Python

from libmproxy import encoding
import libpry
import cStringIO
import gzip, zlib
class uidentity(libpry.AutoTree):
def test_simple(self):
assert "string" == encoding.decode("identity", "string")
assert "string" == encoding.encode("identity", "string")
def test_fallthrough(self):
assert "string" == encoding.decode("nonexistent encoding", "string")
assert "string" == encoding.encode("nonexistent encoding", "string")
class ugzip(libpry.AutoTree):
def test_simple(self):
assert "string" == encoding.decode("gzip", encoding.encode("gzip", "string"))
assert None == encoding.decode("gzip", "bogus")
class udeflate(libpry.AutoTree):
def test_simple(self):
assert "string" == encoding.decode("deflate", encoding.encode("deflate", "string"))
assert "string" == encoding.decode("deflate", encoding.encode("deflate", "string")[2:-4])
assert None == encoding.decode("deflate", "bogus")
tests = [
uidentity(),
ugzip(),
udeflate()
]