diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2015-06-22 14:52:23 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2015-06-22 14:52:23 +1200 |
commit | 2aa1b98fbf8d03005e022da86e3e534cf25ebf62 (patch) | |
tree | ed0dc9ec2ed2cd6f43ea56eec940b759c711626c /test/test_websockets.py | |
parent | f5c5deb2aea047394238f3b993ddf24c60845768 (diff) | |
download | mitmproxy-2aa1b98fbf8d03005e022da86e3e534cf25ebf62.tar.gz mitmproxy-2aa1b98fbf8d03005e022da86e3e534cf25ebf62.tar.bz2 mitmproxy-2aa1b98fbf8d03005e022da86e3e534cf25ebf62.zip |
netlib/test.py -> test/tservers.py
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 |