mitmproxy/test/netlib/test_utils.py

# coding=utf-8

from netlib import utils, tutils


def test_is_valid_host():
    assert not utils.is_valid_host(b"")
    assert utils.is_valid_host(b"one.two")
    assert not utils.is_valid_host(b"one"*255)
    assert utils.is_valid_host(b"one.two.")


def test_bidi():
    b = utils.BiDi(a=1, b=2)
    assert b.a == 1
    assert b.get_name(1) == "a"
    assert b.get_name(5) is None
    tutils.raises(AttributeError, getattr, b, "c")
    tutils.raises(ValueError, utils.BiDi, one=1, two=1)
Sanitize Print (#1135) * sanitize strings with shell control characters * netlib: add utilities to safe-print bytes * escaped str: add TODO for multi-byte chars 2016-05-12 17:03:57 +00:00			`# coding=utf-8`
more style cleanup 2016-05-29 11:33:20 +00:00
headers: adjust everything 2015-09-05 16:15:47 +00:00			`from netlib import utils, tutils`
Add a tiny utility class for keeping bi-directional mappings. Use it in websocket and socks. 2015-04-29 21:04:22 +00:00
more style cleanup 2016-05-29 11:33:20 +00:00
Fix netlib.utils.is_valid_host - Don't crash when passed an empty string. This translated into an actual core crash, discovered while fuzzing with afl. - Taking a slice of length one out of bytes returns an integer, so the check for trailing period in this function never worked on Python3. - Add unit tests. 2016-07-20 22:38:37 +00:00			`def test_is_valid_host():`
			`assert not utils.is_valid_host(b"")`
			`assert utils.is_valid_host(b"one.two")`
			`assert not utils.is_valid_host(b"one"*255)`
			`assert utils.is_valid_host(b"one.two.")`


Add a tiny utility class for keeping bi-directional mappings. Use it in websocket and socks. 2015-04-29 21:04:22 +00:00			`def test_bidi():`
			`b = utils.BiDi(a=1, b=2)`
			`assert b.a == 1`
websockets: more compact and legible human_readable 2015-04-30 00:10:08 +00:00			`assert b.get_name(1) == "a"`
			`assert b.get_name(5) is None`
Add a tiny utility class for keeping bi-directional mappings. Use it in websocket and socks. 2015-04-29 21:04:22 +00:00			`tutils.raises(AttributeError, getattr, b, "c")`
websockets: more compact and legible human_readable 2015-04-30 00:10:08 +00:00			`tutils.raises(ValueError, utils.BiDi, one=1, two=1)`