diff options
Diffstat (limited to 'test/test_websockets.py')
| -rw-r--r-- | test/test_websockets.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_websockets.py b/test/test_websockets.py index 8ed14708..9956543b 100644 --- a/test/test_websockets.py +++ b/test/test_websockets.py @@ -2,8 +2,8 @@ import os from nose.tools import raises -from netlib import tcp, test, websockets, http -import tutils +from netlib import tcp, websockets, http +from . import tutils, tservers class WebSocketsEchoHandler(tcp.BaseHandler): @@ -75,7 +75,7 @@ class WebSocketsClient(tcp.TCPClient): frame.to_file(self.wfile) -class TestWebSockets(test.ServerTestBase): +class TestWebSockets(tservers.ServerTestBase): handler = WebSocketsEchoHandler def random_bytes(self, n=100): @@ -155,7 +155,7 @@ class BadHandshakeHandler(WebSocketsEchoHandler): self.handshake_done = True -class TestBadHandshake(test.ServerTestBase): +class TestBadHandshake(tservers.ServerTestBase): """ Ensure that the client disconnects if the server handshake is malformed |