Skip to content
Snippets Groups Projects
Commit 5b7070a1 authored by Makarand Dharmapurikar's avatar Makarand Dharmapurikar
Browse files

bugfix in rst_after_data

parent ecc32885
No related branches found
No related tags found
No related merge requests found
...@@ -31,7 +31,7 @@ class H2Factory(Factory): ...@@ -31,7 +31,7 @@ class H2Factory(Factory):
elif self._testcase == 'ping': elif self._testcase == 'ping':
t = test_ping.TestcasePing() t = test_ping.TestcasePing()
elif self._testcase == 'max_streams': elif self._testcase == 'max_streams':
t = TestcaseSettingsMaxStreams(self._num_streams) t = test_max_streams.TestcaseSettingsMaxStreams()
else: else:
logging.error('Unknown test case: %s'%self._testcase) logging.error('Unknown test case: %s'%self._testcase)
assert(0) assert(0)
......
...@@ -8,6 +8,7 @@ class TestcaseRstStreamAfterData(object): ...@@ -8,6 +8,7 @@ class TestcaseRstStreamAfterData(object):
def __init__(self): def __init__(self):
self._base_server = http2_base_server.H2ProtocolBaseServer() self._base_server = http2_base_server.H2ProtocolBaseServer()
self._base_server._handlers['DataReceived'] = self.on_data_received self._base_server._handlers['DataReceived'] = self.on_data_received
self._base_server._handlers['SendDone'] = self.on_send_done
def get_base_server(self): def get_base_server(self):
return self._base_server return self._base_server
...@@ -20,4 +21,7 @@ class TestcaseRstStreamAfterData(object): ...@@ -20,4 +21,7 @@ class TestcaseRstStreamAfterData(object):
self._ready_to_send = True self._ready_to_send = True
self._base_server.setup_send(response_data, event.stream_id) self._base_server.setup_send(response_data, event.stream_id)
# send reset stream # 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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment