Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-30 03:14:22 +00:00
Code Issues Projects Releases Wiki Activity

websocket: test killable

Browse Source
This commit is contained in:
Thomas Kriechbaumer 2017-12-28 19:50:31 +01:00
parent f123fb600d
commit a35716a3a7
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
test/mitmproxy/test_websocket.py
View File

@ -3,6 +3,7 @@ import pytest
from mitmproxy.io import tnetstring from mitmproxy.io import tnetstring
from mitmproxy import flowfilter from mitmproxy import flowfilter
from mitmproxy.exceptions import Kill, ControlException
from mitmproxy.test import tflow from mitmproxy.test import tflow
@ -42,6 +43,20 @@ class TestWebSocketFlow:
assert f.error.get_state() == f2.error.get_state() assert f.error.get_state() == f2.error.get_state()
assert f.error is not f2.error assert f.error is not f2.error
def test_kill(self):
f = tflow.twebsocketflow()
with pytest.raises(ControlException):
f.intercept()
f.resume()
f.kill()
f = tflow.twebsocketflow()
f.intercept()
assert f.killable
f.kill()
assert not f.killable
assert f.reply.value == Kill
def test_match(self): def test_match(self):
f = tflow.twebsocketflow() f = tflow.twebsocketflow()
assert not flowfilter.match("~b nonexistent", f) assert not flowfilter.match("~b nonexistent", f)