diff options
| author | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2016-05-29 11:43:29 +0200 |
|---|---|---|
| committer | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2016-05-29 11:55:12 +0200 |
| commit | 8eeab37a07a00a0c85067ed58885233e426067f7 (patch) | |
| tree | afdff60b21b92adfa177232d04ef5583e005db60 /test/netlib/websockets/test_websockets.py | |
| parent | 0c2c017727c7be1eff3bad1766f3030be1f281ec (diff) | |
| download | mitmproxy-8eeab37a07a00a0c85067ed58885233e426067f7.tar.gz mitmproxy-8eeab37a07a00a0c85067ed58885233e426067f7.tar.bz2 mitmproxy-8eeab37a07a00a0c85067ed58885233e426067f7.zip | |
more style cleanup
Use this to check:
flake8 --count mitmproxy netlib pathod examples test
Diffstat (limited to 'test/netlib/websockets/test_websockets.py')
| -rw-r--r-- | test/netlib/websockets/test_websockets.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/netlib/websockets/test_websockets.py b/test/netlib/websockets/test_websockets.py index 4a6f0f9b..024544f5 100644 --- a/test/netlib/websockets/test_websockets.py +++ b/test/netlib/websockets/test_websockets.py @@ -7,10 +7,11 @@ from netlib import tutils from netlib import websockets from netlib.http import status_codes from netlib.tutils import treq -from netlib.exceptions import * +from netlib import exceptions from .. import tservers + class WebSocketsEchoHandler(tcp.BaseHandler): def __init__(self, connection, address, server): @@ -176,7 +177,7 @@ class TestBadHandshake(tservers.ServerTestBase): handler = BadHandshakeHandler def test(self): - with tutils.raises(TcpDisconnect): + with tutils.raises(exceptions.TcpDisconnect): client = WebSocketsClient(("127.0.0.1", self.port)) client.connect() client.send_message(b"hello") |