aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_language_websocket.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2015-06-08 10:58:12 +1200
committerAldo Cortesi <aldo@nullcube.com>2015-06-08 10:58:12 +1200
commit7dff8f0dca936a58070f6957ab84ca15aa10076a (patch)
treeed00ec221a6eaf338f67276afb69c1fad1652f02 /test/test_language_websocket.py
parentdf962f8e7bab9d8cd1dc9143f44977b0711c5bd9 (diff)
downloadmitmproxy-7dff8f0dca936a58070f6957ab84ca15aa10076a.tar.gz
mitmproxy-7dff8f0dca936a58070f6957ab84ca15aa10076a.tar.bz2
mitmproxy-7dff8f0dca936a58070f6957ab84ca15aa10076a.zip
Convert parse_websocket_frame to a generator
Diffstat (limited to 'test/test_language_websocket.py')
-rw-r--r--test/test_language_websocket.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py
index e7f333b4..107f3427 100644
--- a/test/test_language_websocket.py
+++ b/test/test_language_websocket.py
@@ -36,9 +36,19 @@ class TestWebsocketFrame:
"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
+ tutils.raises(
+ language.ParseException,
+ language.parse_websocket_frame,
+ "wf:x"
+ )
+
def test_client_values(self):
specs = [
"wf:f'wf'",