From f123fb600dfae40223db44bfda9dba87d8c4eda7 Mon Sep 17 00:00:00 2001 From: Thomas Kriechbaumer Date: Wed, 27 Dec 2017 21:44:10 +0100 Subject: websocket: docs++ and kill messages fixes #2724 --- test/mitmproxy/test_websocket.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'test') diff --git a/test/mitmproxy/test_websocket.py b/test/mitmproxy/test_websocket.py index 7c53a4b0..0a37f370 100644 --- a/test/mitmproxy/test_websocket.py +++ b/test/mitmproxy/test_websocket.py @@ -71,3 +71,9 @@ class TestWebSocketFlow: d = tflow.twebsocketflow().handshake_flow.get_state() tnetstring.dump(d, b) assert b.getvalue() + + def test_message_kill(self): + f = tflow.twebsocketflow() + assert not f.messages[-1].killed + f.messages[-1].kill() + assert f.messages[-1].killed -- cgit v1.2.3