diff options
Diffstat (limited to 'test/http/http2/test_protocol.py')
| -rw-r--r-- | test/http/http2/test_protocol.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/http/http2/test_protocol.py b/test/http/http2/test_protocol.py index a369eb49..598b5cd7 100644 --- a/test/http/http2/test_protocol.py +++ b/test/http/http2/test_protocol.py @@ -2,6 +2,7 @@ import OpenSSL import mock from netlib import tcp, http, tutils +from netlib.exceptions import TcpDisconnect from netlib.http import Headers from netlib.http.http2.connections import HTTP2Protocol, TCPHandler from netlib.http.http2.frame import * @@ -127,7 +128,7 @@ class TestPerformServerConnectionPreface(tservers.ServerTestBase): protocol.perform_server_connection_preface() assert protocol.connection_preface_performed - tutils.raises(tcp.NetLibDisconnect, protocol.perform_server_connection_preface, force=True) + tutils.raises(TcpDisconnect, protocol.perform_server_connection_preface, force=True) class TestPerformClientConnectionPreface(tservers.ServerTestBase): |