fix #1162

2016-05-29 13:58:14 +02:00 · 2016-05-29 13:58:14 +02:00 · 01fcaa64c3
parent e2447406cc
commit 01fcaa64c3
1 changed files with 5 additions and 2 deletions
--- a/test/mitmproxy/test_protocol_http2.py
+++ b/test/mitmproxy/test_protocol_http2.py
@ -483,8 +483,11 @@ class TestConnectionLost(_Http2TestBase, _Http2ServerBase):
                h2_conn.receive_data(raw)
            except:
                break
-            client.wfile.write(h2_conn.data_to_send())
-            client.wfile.flush()
+            try:
+                client.wfile.write(h2_conn.data_to_send())
+                client.wfile.flush()
+            except:
+                break

        if len(self.master.state.flows) == 1:
            assert self.master.state.flows[0].response is None