From 8eeab37a07a00a0c85067ed58885233e426067f7 Mon Sep 17 00:00:00 2001 From: Thomas Kriechbaumer Date: Sun, 29 May 2016 11:43:29 +0200 Subject: more style cleanup Use this to check: flake8 --count mitmproxy netlib pathod examples test --- test/netlib/websockets/test_websockets.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'test/netlib/websockets/test_websockets.py') 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") -- cgit v1.2.3