import textwrap, cStringIO, os, time, re, json import libpry from libmproxy import utils utils.CERT_SLEEP_TIME = 0 class uformat_timestamp(libpry.AutoTree): def test_simple(self): assert utils.format_timestamp(utils.timestamp()) class uisBin(libpry.AutoTree): def test_simple(self): assert not utils.isBin("testing\n\r") assert utils.isBin("testing\x01") assert utils.isBin("testing\x0e") assert utils.isBin("testing\x7f") class uisXML(libpry.AutoTree): def test_simple(self): assert not utils.isXML("foo") assert utils.isXML("") assert f(r"") assert f(r"< body/>") assert f(r"< body/ >") assert f(r"< body / >") assert f(r"") assert f(r"") assert f(r"") assert f(r'') assert f('') assert f('

') def test_all(self): def isbalanced(ret): # The last tag should have no indent assert ret[-1].strip() == ret[-1] s = "

one

" ret = utils.pretty_xmlish(s) isbalanced(ret) s = r""" """ isbalanced(utils.pretty_xmlish(textwrap.dedent(s))) s = r"""

""" isbalanced(utils.pretty_xmlish(textwrap.dedent(s))) s = r""" """ ret = utils.pretty_xmlish(textwrap.dedent(s)) isbalanced(ret) s = "

one

" ret = utils.pretty_xmlish(s) assert len(ret) == 6 isbalanced(ret) s = "gobbledygook" assert utils.pretty_xmlish(s) == ["gobbledygook"] class upretty_json(libpry.AutoTree): def test_one(self): s = json.dumps({"foo": 1}) assert utils.pretty_json(s) assert not utils.pretty_json("moo") class u_urldecode(libpry.AutoTree): def test_one(self): s = "one=two&three=four" assert len(utils.urldecode(s)) == 2 class udummy_ca(libpry.AutoTree): def test_all(self): d = self.tmpdir() path = os.path.join(d, "foo/cert.cnf") assert utils.dummy_ca(path) assert os.path.exists(path) path = os.path.join(d, "foo/cert2.pem") assert utils.dummy_ca(path) assert os.path.exists(path) assert os.path.exists(os.path.join(d, "foo/cert2-cert.pem")) assert os.path.exists(os.path.join(d, "foo/cert2-cert.p12")) class udummy_cert(libpry.AutoTree): def test_with_ca(self): d = self.tmpdir() cacert = os.path.join(d, "foo/cert.cnf") assert utils.dummy_ca(cacert) p = utils.dummy_cert( os.path.join(d, "foo"), cacert, "foo.com" ) assert os.path.exists(p) # Short-circuit assert utils.dummy_cert( os.path.join(d, "foo"), cacert, "foo.com" ) def test_no_ca(self): d = self.tmpdir() p = utils.dummy_cert( d, None, "foo.com" ) assert os.path.exists(p) class uLRUCache(libpry.AutoTree): def test_one(self): class Foo: ran = False @utils.LRUCache(2) def one(self, x): self.ran = True return x f = Foo() assert f.one(1) == 1 assert f.ran f.ran = False assert f.one(1) == 1 assert not f.ran f.ran = False assert f.one(1) == 1 assert not f.ran assert f.one(2) == 2 assert f.one(3) == 3 assert f.ran f.ran = False assert f.one(1) == 1 assert f.ran assert len(f._cached_one) == 2 assert len(f._cachelist_one) == 2 class u_parse_url(libpry.AutoTree): def test_simple(self): assert not utils.parse_url("") u = "http://foo.com:8888/test" s, h, po, pa = utils.parse_url(u) assert s == "http" assert h == "foo.com" assert po == 8888 assert pa == "/test" s, h, po, pa = utils.parse_url("http://foo/bar") assert s == "http" assert h == "foo" assert po == 80 assert pa == "/bar" s, h, po, pa = utils.parse_url("http://foo") assert pa == "/" s, h, po, pa = utils.parse_url("https://foo") assert po == 443 assert not utils.parse_url("https://foo:bar") assert not utils.parse_url("https://foo:") class u_parse_size(libpry.AutoTree): def test_simple(self): assert utils.parse_size("1") == 1 assert utils.parse_size("1k") == 1024 assert utils.parse_size("1m") == 1024**2 assert utils.parse_size("1g") == 1024**3 libpry.raises(ValueError, utils.parse_size, "1f") libpry.raises(ValueError, utils.parse_size, "ak") tests = [ uformat_timestamp(), uisBin(), uisXML(), uhexdump(), upretty_size(), uData(), upretty_xmlish(), upretty_json(), u_urldecode(), udel_all(), udummy_ca(), udummy_cert(), uLRUCache(), u_parse_url(), u_parse_size() ]