200ms timeout for client.poll in ensure_active_group and admin client

Author: dpkp

kafka/admin/client.py

@@ -390,7 +390,7 @@ def _send_request_to_node(self, node_id, request, wakeup=True):
         while not self._client.ready(node_id):
             # poll until the connection to broker is ready, otherwise send()
             # will fail with NodeNotReadyError
-            self._client.poll()
+            self._client.poll(timeout_ms=200)
         return self._client.send(node_id, request, wakeup)
 
     def _send_request_to_controller(self, request):

kafka/coordinator/base.py

@@ -371,7 +371,7 @@ def ensure_active_group(self):
                 while not self.coordinator_unknown():
                     if not self._client.in_flight_request_count(self.coordinator_id):
                         break
-                    self._client.poll()
+                    self._client.poll(timeout_ms=200)
                 else:
                     continue