diff options
| author | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2017-02-08 15:00:31 +0100 |
|---|---|---|
| committer | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2017-02-09 16:08:35 +0100 |
| commit | 1847cf175c8a45359eef08f5cf2bfc414b059dbe (patch) | |
| tree | 9a564e33d3ea99c48b959e0e232768fd54d88ff6 /mitmproxy/test | |
| parent | 2134b5b06abf354a595076da5ff2d43e455e1e90 (diff) | |
| download | mitmproxy-1847cf175c8a45359eef08f5cf2bfc414b059dbe.tar.gz mitmproxy-1847cf175c8a45359eef08f5cf2bfc414b059dbe.tar.bz2 mitmproxy-1847cf175c8a45359eef08f5cf2bfc414b059dbe.zip | |
websockets, tcp, version: coverage++
Diffstat (limited to 'mitmproxy/test')
| -rw-r--r-- | mitmproxy/test/tflow.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/mitmproxy/test/tflow.py b/mitmproxy/test/tflow.py index a5670538..6d330840 100644 --- a/mitmproxy/test/tflow.py +++ b/mitmproxy/test/tflow.py @@ -1,3 +1,4 @@ +from mitmproxy.net import websockets from mitmproxy.test import tutils from mitmproxy import tcp from mitmproxy import websocket @@ -70,8 +71,8 @@ def twebsocketflow(client_conn=True, server_conn=True, messages=True, err=None, if messages is True: messages = [ - websocket.WebSocketBinaryMessage(f, True, b"hello binary"), - websocket.WebSocketTextMessage(f, False, "hello text".encode()), + websocket.WebSocketMessage(websockets.OPCODE.BINARY, True, b"hello binary"), + websocket.WebSocketMessage(websockets.OPCODE.TEXT, False, "hello text".encode()), ] if err is True: err = terr() |