diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/mitmproxy/proxy/protocol/test_websocket.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/mitmproxy/proxy/protocol/test_websocket.py b/test/mitmproxy/proxy/protocol/test_websocket.py index 490d31a0..d9389faf 100644 --- a/test/mitmproxy/proxy/protocol/test_websocket.py +++ b/test/mitmproxy/proxy/protocol/test_websocket.py @@ -236,8 +236,8 @@ class TestKillFlow(_WebSocketTest): self.master.addons.add(KillFlow()) self.setup_connection() - frame = websockets.Frame.from_file(self.client.rfile) - assert frame.payload == b'foo' + with pytest.raises(exceptions.TcpDisconnect): + websockets.Frame.from_file(self.client.rfile) class TestSimpleTLS(_WebSocketTest): |