From dad9f06cb9403ac88d31d0ba8422034df2bc5078 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Thu, 17 Sep 2015 02:14:14 +0200 Subject: organize exceptions, improve content-length handling --- test/http/http2/test_protocol.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'test/http/http2/test_protocol.py') 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): -- cgit v1.2.3