Imported Upstream version 1.27.0

[platform/upstream/grpc.git] / test / http2_test / test_max_streams.py

diff --git a/test/http2_test/test_max_streams.py b/test/http2_test/test_max_streams.py
index 7af2e28..4013074 100644 (file)
--- a/test/http2_test/test_max_streams.py
+++ b/test/http2_test/test_max_streams.py
@@ -17,32 +17,36 @@ import logging
 
 import http2_base_server
 
+
 class TestcaseSettingsMaxStreams(object):
-  """
+    """
     This test sets MAX_CONCURRENT_STREAMS to 1 and asserts that at any point
     only 1 stream is active.
   """
-  def __init__(self):
-    self._base_server = http2_base_server.H2ProtocolBaseServer()
-    self._base_server._handlers['DataReceived'] = self.on_data_received
-    self._base_server._handlers['ConnectionMade'] = self.on_connection_made
-
-  def get_base_server(self):
-    return self._base_server
-
-  def on_connection_made(self):
-    logging.info('Connection Made')
-    self._base_server._conn.initiate_connection()
-    self._base_server._conn.update_settings(
-                  {hyperframe.frame.SettingsFrame.MAX_CONCURRENT_STREAMS: 1})
-    self._base_server.transport.setTcpNoDelay(True)
-    self._base_server.transport.write(self._base_server._conn.data_to_send())
-
-  def on_data_received(self, event):
-    self._base_server.on_data_received_default(event)
-    sr = self._base_server.parse_received_data(event.stream_id)
-    if sr:
-      logging.info('Creating response of size = %s' % sr.response_size)
-      response_data = self._base_server.default_response_data(sr.response_size)
-      self._base_server.setup_send(response_data, event.stream_id)
-    # TODO (makdharma): Add assertion to check number of live streams
+
+    def __init__(self):
+        self._base_server = http2_base_server.H2ProtocolBaseServer()
+        self._base_server._handlers['DataReceived'] = self.on_data_received
+        self._base_server._handlers['ConnectionMade'] = self.on_connection_made
+
+    def get_base_server(self):
+        return self._base_server
+
+    def on_connection_made(self):
+        logging.info('Connection Made')
+        self._base_server._conn.initiate_connection()
+        self._base_server._conn.update_settings(
+            {hyperframe.frame.SettingsFrame.MAX_CONCURRENT_STREAMS: 1})
+        self._base_server.transport.setTcpNoDelay(True)
+        self._base_server.transport.write(
+            self._base_server._conn.data_to_send())
+
+    def on_data_received(self, event):
+        self._base_server.on_data_received_default(event)
+        sr = self._base_server.parse_received_data(event.stream_id)
+        if sr:
+            logging.info('Creating response of size = %s' % sr.response_size)
+            response_data = self._base_server.default_response_data(
+                sr.response_size)
+            self._base_server.setup_send(response_data, event.stream_id)
+        # TODO (makdharma): Add assertion to check number of live streams