diff options
| author | Maximilian Hils <git@maximilianhils.com> | 2015-09-17 02:13:28 +0200 |
|---|---|---|
| committer | Maximilian Hils <git@maximilianhils.com> | 2015-09-17 02:13:28 +0200 |
| commit | 14a5f405fdf3973f05ca11182e927d145d10d8d6 (patch) | |
| tree | 2843010882d64d928c0f53edffa9f8d48cee6da0 /test/test_proxy.py | |
| parent | 8fbed971ae4a84e875f1072805b17cc7ada39fad (diff) | |
| download | mitmproxy-14a5f405fdf3973f05ca11182e927d145d10d8d6.tar.gz mitmproxy-14a5f405fdf3973f05ca11182e927d145d10d8d6.tar.bz2 mitmproxy-14a5f405fdf3973f05ca11182e927d145d10d8d6.zip | |
use new netlib exceptions
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): |