diff options
| author |  Makarand Dharmapurikar <makarandd@google.com> | 2016-12-01 14:24:22 -0800 |
|---|---|---|
| committer | Makarand Dharmapurikar <makarandd@google.com> | 2016-12-01 14:24:22 -0800 |
| commit | 4350e748e470be92abab9bb1f72ad8eb1ede5486 (patch) | |
| tree | c1fb6e1df569295aa46742ac3d7935ef815e82d7 /test/http2_test/test_rst_after_data.py | |
| parent | 28d198008a6da19d28f0c34f7b3a4441f9c9d9ff (diff) | |
ability to deal with multiple streams in flight.
Diffstat (limited to 'test/http2_test/test_rst_after_data.py')
| -rw-r--r-- | test/http2_test/test_rst_after_data.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/test/http2_test/test_rst_after_data.py b/test/http2_test/test_rst_after_data.py index ef8d4084d9..c4ff56c889 100644 --- a/test/http2_test/test_rst_after_data.py +++ b/test/http2_test/test_rst_after_data.py @@ -14,10 +14,10 @@ class TestcaseRstStreamAfterData(object): def on_data_received(self, event): self._base_server.on_data_received_default(event) - sr = self._base_server.parse_received_data(self._base_server._recv_buffer) - assert(sr is not None) - response_data = self._base_server.default_response_data(sr.response_size) - self._ready_to_send = True - self._base_server.setup_send(response_data) - # send reset stream - self._base_server.send_reset_stream() + sr = self._base_server.parse_received_data(event.stream_id) + if sr: + response_data = self._base_server.default_response_data(sr.response_size) + self._ready_to_send = True + self._base_server.setup_send(response_data, event.stream_id) + # send reset stream + self._base_server.send_reset_stream() |