aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_language_websocket.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2015-06-04 18:14:25 +1200
committerAldo Cortesi <aldo@nullcube.com>2015-06-04 18:14:25 +1200
commit5a33f310412d57c975ff26ddc5653c6cca05a433 (patch)
tree1283088d2be001b2cf3cfa9b65412bd48bf7bdf3 /test/test_language_websocket.py
parent9fda74c65a632bda5176e2ccafbbcab2af27d77d (diff)
downloadmitmproxy-5a33f310412d57c975ff26ddc5653c6cca05a433.tar.gz
mitmproxy-5a33f310412d57c975ff26ddc5653c6cca05a433.tar.bz2
mitmproxy-5a33f310412d57c975ff26ddc5653c6cca05a433.zip
Clarify nomenclature, complete nested frame language spec.
Diffstat (limited to 'test/test_language_websocket.py')
-rw-r--r--test/test_language_websocket.py29
1 files changed, 19 insertions, 10 deletions
diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py
index f55e6e37..70a626e6 100644
--- a/test/test_language_websocket.py
+++ b/test/test_language_websocket.py
@@ -10,7 +10,19 @@ def parse_request(s):
class TestWebsocketFrame:
- def test_values(self):
+ 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",
@@ -25,16 +37,13 @@ class TestWebsocketFrame:
"wf:-fin:-rsv1:-rsv2:-rsv3:-mask",
"wf:k@4",
]
- for i in specs:
- wf = parse_request(i)
- assert isinstance(wf, websockets.WebsocketFrame)
- assert wf
- assert wf.values(language.Settings())
- assert wf.resolve(language.Settings())
+ self._test_messages(specs, websockets.WebsocketFrame)
- spec = wf.spec()
- wf2 = parse_request(spec)
- assert wf2.spec() == spec
+ def test_client_values(self):
+ specs = [
+ "wf:f'wf'",
+ ]
+ self._test_messages(specs, websockets.WebsocketClientFrame)
def test_flags(self):
wf = parse_request("wf:fin:mask:rsv1:rsv2:rsv3")