aboutsummaryrefslogtreecommitdiffhomepage
path: root/test/http2_test/test_rst_after_data.py
diff options
context:
space:
mode:
authorGravatar Makarand Dharmapurikar <makarandd@google.com>2016-12-01 16:56:07 -0800
committerGravatar Makarand Dharmapurikar <makarandd@google.com>2016-12-01 16:56:07 -0800
commit5b7070a15b7cd13115a15a619d447792d99d78a6 (patch)
treec0fadfdbbec9348351bceb0e59e330f24a007a62 /test/http2_test/test_rst_after_data.py
parentecc32885f097ece2c46187166c496e3dd2d70229 (diff)
bugfix in rst_after_data
Diffstat (limited to 'test/http2_test/test_rst_after_data.py')
-rw-r--r--test/http2_test/test_rst_after_data.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/test/http2_test/test_rst_after_data.py b/test/http2_test/test_rst_after_data.py
index c4ff56c889..30163c2ceb 100644
--- a/test/http2_test/test_rst_after_data.py
+++ b/test/http2_test/test_rst_after_data.py
@@ -8,6 +8,7 @@ class TestcaseRstStreamAfterData(object):
def __init__(self):
self._base_server = http2_base_server.H2ProtocolBaseServer()
self._base_server._handlers['DataReceived'] = self.on_data_received
+ self._base_server._handlers['SendDone'] = self.on_send_done
def get_base_server(self):
return self._base_server
@@ -20,4 +21,7 @@ class TestcaseRstStreamAfterData(object):
self._ready_to_send = True
self._base_server.setup_send(response_data, event.stream_id)
# send reset stream
- self._base_server.send_reset_stream()
+
+ def on_send_done(self, stream_id):
+ self._base_server.send_reset_stream()
+ self._base_server._stream_status[stream_id] = False