aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_websockets.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2015-06-22 14:52:23 +1200
committerAldo Cortesi <aldo@nullcube.com>2015-06-22 14:52:23 +1200
commit2aa1b98fbf8d03005e022da86e3e534cf25ebf62 (patch)
treeed0dc9ec2ed2cd6f43ea56eec940b759c711626c /test/test_websockets.py
parentf5c5deb2aea047394238f3b993ddf24c60845768 (diff)
downloadmitmproxy-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.py8
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