diff options
| author | Maximilian Hils <git@maximilianhils.com> | 2016-12-01 09:28:18 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-12-01 09:28:18 +0100 |
| commit | d658783dec7e486003704b3d0cd24b9c5148e903 (patch) | |
| tree | d671480b506d9f6db29bbffed4a78e3792bdceca /test | |
| parent | 34bf3a2496ee82e1142547b95b28a0918a911bff (diff) | |
| parent | cc6aa1f54201aa0634da8a4fa4fafca36609a9c3 (diff) | |
| download | mitmproxy-d658783dec7e486003704b3d0cd24b9c5148e903.tar.gz mitmproxy-d658783dec7e486003704b3d0cd24b9c5148e903.tar.bz2 mitmproxy-d658783dec7e486003704b3d0cd24b9c5148e903.zip | |
Merge pull request #1797 from Kriechi/websocket++
Pass Autobahn WebSocket Test Suite
Diffstat (limited to 'test')
| -rw-r--r-- | test/mitmproxy/protocol/test_websocket.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/test/mitmproxy/protocol/test_websocket.py b/test/mitmproxy/protocol/test_websocket.py index e1c3e49a..e42250e0 100644 --- a/test/mitmproxy/protocol/test_websocket.py +++ b/test/mitmproxy/protocol/test_websocket.py @@ -276,6 +276,7 @@ class TestClose(_WebSocketTest): def handle_websockets(cls, rfile, wfile): frame = websockets.Frame.from_file(rfile) wfile.write(bytes(frame)) + wfile.write(bytes(websockets.Frame(fin=1, opcode=websockets.OPCODE.CLOSE))) wfile.flush() with pytest.raises(exceptions.TcpDisconnect): @@ -287,6 +288,7 @@ class TestClose(_WebSocketTest): client.wfile.write(bytes(websockets.Frame(fin=1, opcode=websockets.OPCODE.CLOSE))) client.wfile.flush() + websockets.Frame.from_file(client.rfile) with pytest.raises(exceptions.TcpDisconnect): websockets.Frame.from_file(client.rfile) @@ -296,6 +298,7 @@ class TestClose(_WebSocketTest): client.wfile.write(bytes(websockets.Frame(fin=1, opcode=websockets.OPCODE.CLOSE, payload=b'\00\42'))) client.wfile.flush() + websockets.Frame.from_file(client.rfile) with pytest.raises(exceptions.TcpDisconnect): websockets.Frame.from_file(client.rfile) @@ -305,6 +308,7 @@ class TestClose(_WebSocketTest): client.wfile.write(bytes(websockets.Frame(fin=1, opcode=websockets.OPCODE.CLOSE, payload=b'\00\42foobar'))) client.wfile.flush() + websockets.Frame.from_file(client.rfile) with pytest.raises(exceptions.TcpDisconnect): websockets.Frame.from_file(client.rfile) |