aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_proxy.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_proxy.py')
-rw-r--r--test/test_proxy.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/test/test_proxy.py b/test/test_proxy.py
index 5828d077..c5640d8a 100644
--- a/test/test_proxy.py
+++ b/test/test_proxy.py
@@ -56,7 +56,7 @@ class TestServerConnection:
sc = proxy.ServerConnection(proxy.ProxyConfig(), "http", self.d.IFACE, self.d.port, "host.com")
sc.connect()
sc.connection = mock.Mock()
- sc.connection.close = mock.Mock(side_effect=IOError)
+ sc.connection.flush = mock.Mock(side_effect=tcp.NetLibDisconnect)
sc.terminate()