diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2015-04-24 15:23:00 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2015-04-24 15:23:00 +1200 |
commit | def93ea8cae69676a91b01e149e8a406fa03eacd (patch) | |
tree | a63986489656694fdbc4a0eb1083f43ca2810c71 /test/test_websockets.py | |
parent | f22bc0b4c74776bcc312fed1f4ceede83f869a6e (diff) | |
download | mitmproxy-def93ea8cae69676a91b01e149e8a406fa03eacd.tar.gz mitmproxy-def93ea8cae69676a91b01e149e8a406fa03eacd.tar.bz2 mitmproxy-def93ea8cae69676a91b01e149e8a406fa03eacd.zip |
websockets: remove validation
We don't really need this any more. The interface is much less error prone
because bit flags are no longer integers, we have a range check on opcode on
header instantiation, and we've deferred length code calculation and so forth
into the byte render methods.
Diffstat (limited to 'test/test_websockets.py')
-rw-r--r-- | test/test_websockets.py | 26 |
1 files changed, 8 insertions, 18 deletions
diff --git a/test/test_websockets.py b/test/test_websockets.py index 06876e0b..215b3958 100644 --- a/test/test_websockets.py +++ b/test/test_websockets.py @@ -4,6 +4,7 @@ import os from nose.tools import raises from netlib import tcp, test, websockets, http +import tutils class WebSocketsEchoHandler(tcp.BaseHandler): @@ -106,25 +107,7 @@ class TestWebSockets(test.ServerTestBase): """ msg = self.random_bytes() client_frame = websockets.Frame.default(msg, from_client = True) - server_frame = websockets.Frame.default(msg, from_client = False) - assert server_frame.is_valid() - - def test_is_valid(self): - def f(): - return websockets.Frame.default(self.random_bytes(10), True) - - frame = f() - assert frame.is_valid() - - frame = f() - frame.header.fin = 2 - assert not frame.is_valid() - - frame = f() - frame.header.mask_bit = 1 - frame.header.masking_key = "foobbarboo" - assert not frame.is_valid() def test_serialization_bijection(self): """ @@ -207,6 +190,9 @@ class TestFrameHeader: f2 = websockets.FrameHeader.from_file(cStringIO.StringIO(bytes)) assert not f2.mask + def test_violations(self): + tutils.raises("opcode", websockets.FrameHeader, opcode=17) + class TestFrame: def test_roundtrip(self): @@ -216,3 +202,7 @@ class TestFrame: f2 = websockets.Frame.from_file(cStringIO.StringIO(bytes)) assert f == f2 round("test") + + def test_human_readable(self): + f = websockets.Frame() + assert f.human_readable() |