netlib.utils.BiDi -> mitmproxy.types.bidi.BiDi

This commit is contained in:
Aldo Cortesi
2016-10-20 10:32:09 +13:00
parent a684585e7c
commit 9555126585
7 changed files with 53 additions and 49 deletions

View File

@@ -0,0 +1,11 @@
from mitmproxy.types import bidi
from netlib import tutils
def test_bidi():
b = bidi.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, bidi.BiDi, one=1, two=1)

View File

@@ -1,6 +1,6 @@
# coding=utf-8
from netlib import utils, tutils
from netlib import utils
def test_is_valid_host():
@@ -8,12 +8,3 @@ def test_is_valid_host():
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)