diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2015-05-17 11:04:53 +1200 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2015-05-17 11:04:53 +1200 |
| commit | cd2fb13b3e9ced39059f3d6572fd8806d0026d2e (patch) | |
| tree | d22f5d48705272516cf04e9793db33371c29abc7 /test/test_language_websocket.py | |
| parent | e4feba54330e1afcfb8d48bce8c474659aba281c (diff) | |
| download | mitmproxy-cd2fb13b3e9ced39059f3d6572fd8806d0026d2e.tar.gz mitmproxy-cd2fb13b3e9ced39059f3d6572fd8806d0026d2e.tar.bz2 mitmproxy-cd2fb13b3e9ced39059f3d6572fd8806d0026d2e.zip | |
websockets: replace the "raw" with a simpler mechanism
You can now say "knone" to specifiy that no key should be generated
under any circumstances.
Diffstat (limited to 'test/test_language_websocket.py')
| -rw-r--r-- | test/test_language_websocket.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py index 5e2ccb88..f9e8310a 100644 --- a/test/test_language_websocket.py +++ b/test/test_language_websocket.py @@ -17,7 +17,7 @@ class TestWebsocketFrame: "wf:b'foo'", "wf:cbinary", "wf:c1", - "wf:r", + "wf:mask:knone", "wf:fin", "wf:fin:rsv1:rsv2:rsv3:mask", "wf:-fin:-rsv1:-rsv2:-rsv3:-mask", @@ -67,7 +67,7 @@ class TestWebsocketFrame: assert self.fr("wf:ctext").header.opcode ==\ netlib.websockets.OPCODE.TEXT - def test_auto_raw(self): + def test_construction(self): # Simple server frame frm = self.fr("wf:b'foo'") assert not frm.header.mask @@ -99,3 +99,10 @@ class TestWebsocketFrame: # We're reading back a corrupted frame - the first 3 characters of the # mask is mis-interpreted as the payload assert frm.payload == "abc" + + def test_knone(self): + tutils.raises( + "expected 4 bytes", + self.fr, + "wf:b'foo':mask:knone", + ) |