Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2025-01-30 23:09:44 +00:00
Code Issues Projects Releases Wiki Activity

improve test reliability

Browse Source
This commit is contained in:
Thomas Kriechbaumer 2016-02-01 23:27:50 +01:00
parent b007ff3f9b
commit 738094e167
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/test_protocol_http2.py
View File

@ -48,7 +48,10 @@ class _Http2ServerBase(netlib_tservers.ServerTestBase):
self.wfile.flush()
while True:
events = h2_conn.receive_data(b''.join(http2_read_raw_frame(self.rfile)))
try:
events = h2_conn.receive_data(b''.join(http2_read_raw_frame(self.rfile)))
except:
break
self.wfile.write(h2_conn.data_to_send())
self.wfile.flush()
@ -362,8 +365,8 @@ class TestPushPromise(_Http2TestBase, _Http2ServerBase):
('foo', 'bar')
])
done = False
while not done:
streams = 0
while streams != 3:
try:
events = h2_conn.receive_data(b''.join(http2_read_raw_frame(client.rfile)))
except:
@ -373,8 +376,9 @@ class TestPushPromise(_Http2TestBase, _Http2ServerBase):
for event in events:
if isinstance(event, h2.events.StreamEnded) and event.stream_id == 1:
done = True
streams += 1
elif isinstance(event, h2.events.PushedStreamReceived):
streams += 1
h2_conn.reset_stream(event.pushed_stream_id, error_code=0x8)
client.wfile.write(h2_conn.data_to_send())
client.wfile.flush()