aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_encoding.py
blob: d9e6a85c9795980704412888dfab85af7d19e024 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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 None == encoding.decode("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()
]