aboutsummaryrefslogtreecommitdiffstats
path: root/test/pathod/test_language_websocket.py
diff options
context:
space:
mode:
authorThomas Kriechbaumer <thomas@kriechbaumer.name>2017-02-14 22:48:43 +0100
committerThomas Kriechbaumer <thomas@kriechbaumer.name>2017-02-14 23:31:37 +0100
commita12c3d3f8ea255dd03bb7e993fa85eb00f47ab29 (patch)
tree82595e20d3fbbbb568de9178b7645a40fb7b0923 /test/pathod/test_language_websocket.py
parent04748e6f3f654733afafea3701d4ed03b08b8167 (diff)
downloadmitmproxy-a12c3d3f8ea255dd03bb7e993fa85eb00f47ab29.tar.gz
mitmproxy-a12c3d3f8ea255dd03bb7e993fa85eb00f47ab29.tar.bz2
mitmproxy-a12c3d3f8ea255dd03bb7e993fa85eb00f47ab29.zip
restructure and move test files
add empty test files to satisfy linter
Diffstat (limited to 'test/pathod/test_language_websocket.py')
-rw-r--r--test/pathod/test_language_websocket.py142
1 files changed, 0 insertions, 142 deletions
diff --git a/test/pathod/test_language_websocket.py b/test/pathod/test_language_websocket.py
deleted file mode 100644
index e5046591..00000000
--- a/test/pathod/test_language_websocket.py
+++ /dev/null
@@ -1,142 +0,0 @@
-import pytest
-
-from pathod import language
-from pathod.language import websockets
-import mitmproxy.net.websockets
-
-from . import tservers
-
-
-def parse_request(s):
- return next(language.parse_pathoc(s))
-
-
-class TestWebsocketFrame:
-
- def _test_messages(self, specs, message_klass):
- for i in specs:
- wf = parse_request(i)
- assert isinstance(wf, message_klass)
- assert wf
- assert wf.values(language.Settings())
- assert wf.resolve(language.Settings())
-
- spec = wf.spec()
- wf2 = parse_request(spec)
- assert wf2.spec() == spec
-
- def test_server_values(self):
- specs = [
- "wf",
- "wf:dr",
- "wf:b'foo'",
- "wf:mask:r'foo'",
- "wf:l1024:b'foo'",
- "wf:cbinary",
- "wf:c1",
- "wf:mask:knone",
- "wf:fin",
- "wf:fin:rsv1:rsv2:rsv3:mask",
- "wf:-fin:-rsv1:-rsv2:-rsv3:-mask",
- "wf:k@4",
- "wf:x10",
- ]
- self._test_messages(specs, websockets.WebsocketFrame)
-
- def test_parse_websocket_frames(self):
- wf = language.parse_websocket_frame("wf:x10")
- assert len(list(wf)) == 10
- with pytest.raises(language.ParseException):
- language.parse_websocket_frame("wf:x")
-
- def test_client_values(self):
- specs = [
- "wf:f'wf'",
- ]
- self._test_messages(specs, websockets.WebsocketClientFrame)
-
- def test_nested_frame(self):
- wf = parse_request("wf:f'wf'")
- assert wf.nested_frame
-
- def test_flags(self):
- wf = parse_request("wf:fin:mask:rsv1:rsv2:rsv3")
- frm = mitmproxy.net.websockets.Frame.from_bytes(tservers.render(wf))
- assert frm.header.fin
- assert frm.header.mask
- assert frm.header.rsv1
- assert frm.header.rsv2
- assert frm.header.rsv3
-
- wf = parse_request("wf:-fin:-mask:-rsv1:-rsv2:-rsv3")
- frm = mitmproxy.net.websockets.Frame.from_bytes(tservers.render(wf))
- assert not frm.header.fin
- assert not frm.header.mask
- assert not frm.header.rsv1
- assert not frm.header.rsv2
- assert not frm.header.rsv3
-
- def fr(self, spec, **kwargs):
- settings = language.base.Settings(**kwargs)
- wf = parse_request(spec)
- return mitmproxy.net.websockets.Frame.from_bytes(tservers.render(wf, settings))
-
- def test_construction(self):
- assert self.fr("wf:c1").header.opcode == 1
- assert self.fr("wf:c0").header.opcode == 0
- assert self.fr("wf:cbinary").header.opcode ==\
- mitmproxy.net.websockets.OPCODE.BINARY
- assert self.fr("wf:ctext").header.opcode ==\
- mitmproxy.net.websockets.OPCODE.TEXT
-
- def test_rawbody(self):
- frm = self.fr("wf:mask:r'foo'")
- assert len(frm.payload) == 3
- assert frm.payload != b"foo"
-
- assert self.fr("wf:r'foo'").payload == b"foo"
-
- def test_construction_2(self):
- # Simple server frame
- frm = self.fr("wf:b'foo'")
- assert not frm.header.mask
- assert not frm.header.masking_key
-
- # Simple client frame
- frm = self.fr("wf:b'foo'", is_client=True)
- assert frm.header.mask
- assert frm.header.masking_key
- frm = self.fr("wf:b'foo':k'abcd'", is_client=True)
- assert frm.header.mask
- assert frm.header.masking_key == b'abcd'
-
- # Server frame, mask explicitly set
- frm = self.fr("wf:b'foo':mask")
- assert frm.header.mask
- assert frm.header.masking_key
- frm = self.fr("wf:b'foo':k'abcd'")
- assert frm.header.mask
- assert frm.header.masking_key == b'abcd'
-
- # Client frame, mask explicitly unset
- frm = self.fr("wf:b'foo':-mask", is_client=True)
- assert not frm.header.mask
- assert not frm.header.masking_key
-
- frm = self.fr("wf:b'foo':-mask:k'abcd'", is_client=True)
- assert not frm.header.mask
- # We're reading back a corrupted frame - the first 3 characters of the
- # mask is mis-interpreted as the payload
- assert frm.payload == b"abc"
-
- def test_knone(self):
- with pytest.raises(Exception, match="Expected 4 bytes"):
- self.fr("wf:b'foo':mask:knone")
-
- def test_length(self):
- assert self.fr("wf:l3:b'foo'").header.payload_length == 3
- frm = self.fr("wf:l2:b'foo'")
- assert frm.header.payload_length == 2
- assert frm.payload == b"fo"
- with pytest.raises(Exception, match="Expected 1024 bytes"):
- self.fr("wf:l1024:b'foo'")