diff options
| author |  Makarand Dharmapurikar <makarandd@google.com> | 2016-12-01 16:56:07 -0800 |
|---|---|---|
| committer | Makarand Dharmapurikar <makarandd@google.com> | 2016-12-01 16:56:07 -0800 |
| commit | 5b7070a15b7cd13115a15a619d447792d99d78a6 (patch) | |
| tree | c0fadfdbbec9348351bceb0e59e330f24a007a62 /test/http2_test | |
| parent | ecc32885f097ece2c46187166c496e3dd2d70229 (diff) | |
bugfix in rst_after_data
Diffstat (limited to 'test/http2_test')
| -rw-r--r-- | test/http2_test/http2_test_server.py | 2 | ||||
| -rw-r--r-- | test/http2_test/test_rst_after_data.py | 6 |
2 files changed, 6 insertions, 2 deletions
diff --git a/test/http2_test/http2_test_server.py b/test/http2_test/http2_test_server.py index 35308c4787..c74fc4b1fb 100644 --- a/test/http2_test/http2_test_server.py +++ b/test/http2_test/http2_test_server.py @@ -31,7 +31,7 @@ class H2Factory(Factory): elif self._testcase == 'ping': t = test_ping.TestcasePing() elif self._testcase == 'max_streams': - t = TestcaseSettingsMaxStreams(self._num_streams) + t = test_max_streams.TestcaseSettingsMaxStreams() else: logging.error('Unknown test case: %s'%self._testcase) assert(0) 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 |