diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_utils.py | 12 | ||||
| -rw-r--r-- | test/test_websockets.py | 8 |
2 files changed, 17 insertions, 3 deletions
diff --git a/test/test_utils.py b/test/test_utils.py index 0cdd3fae..942136fd 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -5,9 +5,10 @@ import tutils def test_bidi(): b = utils.BiDi(a=1, b=2) assert b.a == 1 - assert b[1] == "a" + assert b.get_name(1) == "a" + assert b.get_name(5) is None tutils.raises(AttributeError, getattr, b, "c") - tutils.raises(KeyError, b.__getitem__, 5) + tutils.raises(ValueError, utils.BiDi, one=1, two=1) def test_hexdump(): @@ -19,3 +20,10 @@ def test_cleanBin(): assert utils.cleanBin("\00ne") == ".ne" assert utils.cleanBin("\nne") == "\nne" assert utils.cleanBin("\nne", True) == ".ne" + + +def test_pretty_size(): + 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) == "1MB" diff --git a/test/test_websockets.py b/test/test_websockets.py index 9266d93e..d8e56a8f 100644 --- a/test/test_websockets.py +++ b/test/test_websockets.py @@ -185,7 +185,13 @@ class TestFrameHeader: round(masking_key="test") def test_human_readable(self): - f = websockets.FrameHeader(masking_key="test", mask=False) + f = websockets.FrameHeader( + masking_key="test", + fin=True, + payload_length=10 + ) + assert f.human_readable() + f = websockets.FrameHeader() assert f.human_readable() def test_funky(self): |