blob: 00632c0e8731264d10c47ce99171e6b6b01ed9bc (
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
31
32
33
|
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())
assert None == encoding.decode("gzip", "bogus")
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])
assert None == encoding.decode("deflate", "bogus")
tests = [
udecode_identity(),
udecode_gzip(),
udecode_deflate()
]
|
