fixed feedback from review

1 files changed, 6 insertions, 3 deletions

diff --git a/test/http2_test/test_max_streams.py b/test/http2_test/test_max_streams.py
index deb26770c3..10a9da3deb 100644
--- a/test/http2_test/test_max_streams.py
+++ b/test/http2_test/test_max_streams.py
@@ -1,6 +1,7 @@
+import hyperframe.frame
 import logging
+
 import http2_base_server
-from hyperframe.frame import SettingsFrame
 
 class TestcaseSettingsMaxStreams(object):
   """
@@ -18,7 +19,8 @@ class TestcaseSettingsMaxStreams(object):
   def on_connection_made(self):
     logging.info('Connection Made')
     self._base_server._conn.initiate_connection()
-    self._base_server._conn.update_settings({SettingsFrame.MAX_CONCURRENT_STREAMS: 1})
+    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())
 
@@ -26,6 +28,7 @@ class TestcaseSettingsMaxStreams(object):
     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)
+      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