Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2025-02-01 07:49:10 +00:00
Code Issues Projects Releases Wiki Activity

http2: improve header tests (#1143) (#1144)

Browse Source
This commit is contained in:
Thomas Kriechbaumer 2016-05-14 10:14:42 -07:00
parent 7fdc0a94db
commit f4a13d6f9f
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
test/mitmproxy/test_protocol_http2.py
View File

@ -162,16 +162,19 @@ class TestSimple(_Http2TestBase, _Http2ServerBase):
if isinstance(event, h2.events.ConnectionTerminated): if isinstance(event, h2.events.ConnectionTerminated):
return False return False
elif isinstance(event, h2.events.RequestReceived): elif isinstance(event, h2.events.RequestReceived):
h2_conn.send_headers(1, [ assert ('client-foo', 'client-bar-1') in event.headers
assert ('client-foo', 'client-bar-2') in event.headers
h2_conn.send_headers(event.stream_id, [
(':status', '200'), (':status', '200'),
('foo', 'bar'), ('server-foo', 'server-bar'),
('föo', 'bär'), ('föo', 'bär'),
('X-Stream-ID', str(event.stream_id)),
]) ])
h2_conn.send_data(1, b'foobar') h2_conn.send_data(event.stream_id, b'foobar')
h2_conn.end_stream(1) h2_conn.end_stream(event.stream_id)
wfile.write(h2_conn.data_to_send()) wfile.write(h2_conn.data_to_send())
wfile.flush() wfile.flush()
return True return True
def test_simple(self): def test_simple(self):
@ -182,6 +185,8 @@ class TestSimple(_Http2TestBase, _Http2ServerBase):
(':method', 'GET'), (':method', 'GET'),
(':scheme', 'https'), (':scheme', 'https'),
(':path', '/'), (':path', '/'),
('ClIeNt-FoO', 'client-bar-1'),
('ClIeNt-FoO', 'client-bar-2'),
], body='my request body echoed back to me') ], body='my request body echoed back to me')
done = False done = False
@ -203,7 +208,7 @@ class TestSimple(_Http2TestBase, _Http2ServerBase):
assert len(self.master.state.flows) == 1 assert len(self.master.state.flows) == 1
assert self.master.state.flows[0].response.status_code == 200 assert self.master.state.flows[0].response.status_code == 200
assert self.master.state.flows[0].response.headers['foo'] == 'bar' assert self.master.state.flows[0].response.headers['server-foo'] == 'server-bar'
assert self.master.state.flows[0].response.headers['föo'] == 'bär' assert self.master.state.flows[0].response.headers['föo'] == 'bär'
assert self.master.state.flows[0].response.body == b'foobar' assert self.master.state.flows[0].response.body == b'foobar'