diff options
| author | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2016-05-29 11:43:29 +0200 |
|---|---|---|
| committer | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2016-05-29 11:55:12 +0200 |
| commit | 8eeab37a07a00a0c85067ed58885233e426067f7 (patch) | |
| tree | afdff60b21b92adfa177232d04ef5583e005db60 /test/netlib/http/http2 | |
| parent | 0c2c017727c7be1eff3bad1766f3030be1f281ec (diff) | |
| download | mitmproxy-8eeab37a07a00a0c85067ed58885233e426067f7.tar.gz mitmproxy-8eeab37a07a00a0c85067ed58885233e426067f7.tar.bz2 mitmproxy-8eeab37a07a00a0c85067ed58885233e426067f7.zip | |
more style cleanup
Use this to check:
flake8 --count mitmproxy netlib pathod examples test
Diffstat (limited to 'test/netlib/http/http2')
| -rw-r--r-- | test/netlib/http/http2/test_connections.py | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/test/netlib/http/http2/test_connections.py b/test/netlib/http/http2/test_connections.py index 139d4fc1..ff462ba6 100644 --- a/test/netlib/http/http2/test_connections.py +++ b/test/netlib/http/http2/test_connections.py @@ -1,7 +1,7 @@ import mock import codecs -from hyperframe.frame import * +from hyperframe import frame from netlib import tcp, http, utils from netlib.tutils import raises @@ -10,6 +10,7 @@ from netlib.http.http2.connections import HTTP2Protocol, TCPHandler from ... import tservers + class TestTCPHandlerWrapper: def test_wrapped(self): h = TCPHandler(rfile='foo', wfile='bar') @@ -213,19 +214,19 @@ class TestApplySettings(tservers.ServerTestBase): protocol = HTTP2Protocol(c) protocol._apply_settings({ - SettingsFrame.ENABLE_PUSH: 'foo', - SettingsFrame.MAX_CONCURRENT_STREAMS: 'bar', - SettingsFrame.INITIAL_WINDOW_SIZE: 'deadbeef', + frame.SettingsFrame.ENABLE_PUSH: 'foo', + frame.SettingsFrame.MAX_CONCURRENT_STREAMS: 'bar', + frame.SettingsFrame.INITIAL_WINDOW_SIZE: 'deadbeef', }) assert c.rfile.safe_read(2) == b"OK" assert protocol.http2_settings[ - SettingsFrame.ENABLE_PUSH] == 'foo' + frame.SettingsFrame.ENABLE_PUSH] == 'foo' assert protocol.http2_settings[ - SettingsFrame.MAX_CONCURRENT_STREAMS] == 'bar' + frame.SettingsFrame.MAX_CONCURRENT_STREAMS] == 'bar' assert protocol.http2_settings[ - SettingsFrame.INITIAL_WINDOW_SIZE] == 'deadbeef' + frame.SettingsFrame.INITIAL_WINDOW_SIZE] == 'deadbeef' class TestCreateHeaders(object): @@ -257,7 +258,7 @@ class TestCreateHeaders(object): (b'server', b'version')]) protocol = HTTP2Protocol(self.c) - protocol.http2_settings[SettingsFrame.MAX_FRAME_SIZE] = 8 + protocol.http2_settings[frame.SettingsFrame.MAX_FRAME_SIZE] = 8 bytes = protocol._create_headers(headers, 1, end_stream=True) assert len(bytes) == 3 assert bytes[0] == codecs.decode('000008010100000001828487408294e783', 'hex_codec') @@ -280,7 +281,7 @@ class TestCreateBody(object): def test_create_body_multiple_frames(self): protocol = HTTP2Protocol(self.c) - protocol.http2_settings[SettingsFrame.MAX_FRAME_SIZE] = 5 + protocol.http2_settings[frame.SettingsFrame.MAX_FRAME_SIZE] = 5 bytes = protocol._create_body(b'foobarmehm42', 1) assert len(bytes) == 3 assert bytes[0] == codecs.decode('000005000000000001666f6f6261', 'hex_codec') |