1# Copyright 2016 gRPC authors. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14 15import logging 16 17import http2_base_server 18import hyperframe.frame 19 20 21class TestcaseSettingsMaxStreams(object): 22 """ 23 This test sets MAX_CONCURRENT_STREAMS to 1 and asserts that at any point 24 only 1 stream is active. 25 """ 26 27 def __init__(self): 28 self._base_server = http2_base_server.H2ProtocolBaseServer() 29 self._base_server._handlers["DataReceived"] = self.on_data_received 30 self._base_server._handlers["ConnectionMade"] = self.on_connection_made 31 32 def get_base_server(self): 33 return self._base_server 34 35 def on_connection_made(self): 36 logging.info("Connection Made") 37 self._base_server._conn.initiate_connection() 38 self._base_server._conn.update_settings( 39 {hyperframe.frame.SettingsFrame.MAX_CONCURRENT_STREAMS: 1} 40 ) 41 self._base_server.transport.setTcpNoDelay(True) 42 self._base_server.transport.write( 43 self._base_server._conn.data_to_send() 44 ) 45 46 def on_data_received(self, event): 47 self._base_server.on_data_received_default(event) 48 sr = self._base_server.parse_received_data(event.stream_id) 49 if sr: 50 logging.info("Creating response of size = %s" % sr.response_size) 51 response_data = self._base_server.default_response_data( 52 sr.response_size 53 ) 54 self._base_server.setup_send(response_data, event.stream_id) 55 # TODO (makdharma): Add assertion to check number of live streams 56