diff options
| author | Maximilian Hils <git@maximilianhils.com> | 2015-09-17 02:14:14 +0200 |
|---|---|---|
| committer | Maximilian Hils <git@maximilianhils.com> | 2015-09-17 02:14:14 +0200 |
| commit | dad9f06cb9403ac88d31d0ba8422034df2bc5078 (patch) | |
| tree | d69ea509a4cfd9549f4a602966bd9309f8853b3c /test/http/http2/test_protocol.py | |
| parent | e1659f3fcf83b5993b776a4ef3d2de70fbe27aa2 (diff) | |
| download | mitmproxy-dad9f06cb9403ac88d31d0ba8422034df2bc5078.tar.gz mitmproxy-dad9f06cb9403ac88d31d0ba8422034df2bc5078.tar.bz2 mitmproxy-dad9f06cb9403ac88d31d0ba8422034df2bc5078.zip | |
organize exceptions, improve content-length handling
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): |