diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2015-06-04 18:14:25 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2015-06-04 18:14:25 +1200 |
commit | 5a33f310412d57c975ff26ddc5653c6cca05a433 (patch) | |
tree | 1283088d2be001b2cf3cfa9b65412bd48bf7bdf3 /test/test_language_websocket.py | |
parent | 9fda74c65a632bda5176e2ccafbbcab2af27d77d (diff) | |
download | mitmproxy-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.py | 29 |
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") |