mitmproxy/test/pathod/test_utils.py
2016-05-26 23:07:37 +05:30

39 lines
895 B
Python

from pathod import utils
import tutils
import six
def test_membool():
m = utils.MemBool()
assert not m.v
assert m(1)
assert m.v == 1
assert m(2)
assert m.v == 2
def test_parse_size():
assert utils.parse_size("100") == 100
assert utils.parse_size("100k") == 100 * 1024
tutils.raises("invalid size spec", utils.parse_size, "foo")
tutils.raises("invalid size spec", utils.parse_size, "100kk")
def test_parse_anchor_spec():
assert utils.parse_anchor_spec("foo=200") == ("foo", "200")
assert utils.parse_anchor_spec("foo") is None
def test_data_path():
tutils.raises(ValueError, utils.data.path, "nonexistent")
def test_escape_unprintables():
s = bytes(range(256))
if six.PY2:
s = "".join([chr(i) for i in range(255)])
e = utils.escape_unprintables(s)
assert e.encode('ascii')
assert "PATHOD_MARKER" not in e