mitmproxy/test/test_utils.py

import textwrap, 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("<foo")
        assert utils.isXML("  \n<foo")


class uhexdump(libpry.AutoTree):
    def test_simple(self):
        assert utils.hexdump("one\0"*10)


class udel_all(libpry.AutoTree):
    def test_simple(self):
        d = dict(a=1, b=2, c=3)
        utils.del_all(d, ["a", "x", "b"])
        assert d.keys() == ["c"]


class uclean_hanging_newline(libpry.AutoTree):
    def test_simple(self):
        s = "foo\n"
        assert utils.clean_hanging_newline(s) == "foo"
        assert utils.clean_hanging_newline("foo") == "foo"


class upretty_size(libpry.AutoTree):
    def test_simple(self):
        assert utils.pretty_size(100) == "100B"
        assert utils.pretty_size(1024) == "1kB"
        assert utils.pretty_size(1024 + (1024/2.0)) == "1.5kB"
        assert utils.pretty_size(1024*1024) == "1M"


class uData(libpry.AutoTree):
    def test_nonexistent(self):
        assert utils.pkg_data.path("console")
        libpry.raises("does not exist", utils.pkg_data.path, "nonexistent")


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 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_proxy_spec(libpry.AutoTree):
    def test_simple(self):
        assert not utils.parse_proxy_spec("")
        assert utils.parse_proxy_spec("http://foo.com:88") == ("http", "foo.com", 88)
        assert utils.parse_proxy_spec("http://foo.com") == ("http", "foo.com", 80)
        assert not utils.parse_proxy_spec("foo.com")
        assert not utils.parse_proxy_spec("http://")


class u_unparse_url(libpry.AutoTree):
    def test_simple(self):
        assert utils.unparse_url("http", "foo.com", 99, "") == "http://foo.com:99"
        assert utils.unparse_url("http", "foo.com", 80, "") == "http://foo.com"
        assert utils.unparse_url("https", "foo.com", 80, "") == "https://foo.com:80"
        assert utils.unparse_url("https", "foo.com", 443, "") == "https://foo.com"


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 not utils.parse_size("")
        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")


class u_parse_content_type(libpry.AutoTree):
    def test_simple(self):
        p = utils.parse_content_type
        assert p("text/html") == ("text", "html", {})
        assert p("text") == None

        v = p("text/html; charset=UTF-8")
        assert v == ('text', 'html', {'charset': 'UTF-8'})


class u_cleanBin(libpry.AutoTree):
    def test_simple(self):
        assert utils.cleanBin("one") == "one"
        assert utils.cleanBin("\00ne") == ".ne"
        assert utils.cleanBin("\nne") == "\nne"
        assert utils.cleanBin("\nne", True) == ".ne"


class u_safe_subn(libpry.AutoTree):
    def test_simple(self):
        assert utils.safe_subn("foo", u"bar", "\xc2foo")


tests = [
    u_safe_subn(),
    u_cleanBin(),
    u_parse_content_type(),
    uformat_timestamp(),
    uisBin(),
    uisXML(),
    uhexdump(),
    upretty_size(),
    uData(),
    upretty_json(),
    u_urldecode(),
    udel_all(),
    uLRUCache(),
    u_parse_url(),
    u_parse_proxy_spec(),
    u_unparse_url(),
    u_parse_size(),
    uclean_hanging_newline()
]