diff options
Diffstat (limited to 'test/test_proxy.py')
-rw-r--r-- | test/test_proxy.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_proxy.py b/test/test_proxy.py index 76d8758c..172613aa 100644 --- a/test/test_proxy.py +++ b/test/test_proxy.py @@ -6,6 +6,7 @@ from libmproxy.proxy import ProxyConfig from libmproxy.proxy.config import process_proxy_options from libmproxy.models.connections import ServerConnection from libmproxy.proxy.server import DummyServer, ProxyServer, ConnectionHandler +from netlib.exceptions import TcpDisconnect import tutils from libpathod import test from netlib import http, tcp @@ -40,7 +41,7 @@ class TestServerConnection: sc.connect() sc.connection = mock.Mock() sc.connection.recv = mock.Mock(return_value=False) - sc.connection.flush = mock.Mock(side_effect=tcp.NetLibDisconnect) + sc.connection.flush = mock.Mock(side_effect=TcpDisconnect) sc.finish() def test_repr(self): |