Added support for opensearch-py client connections urllib3httpconnect…

…ion, requestshttpconnection. Signed-off-by: saimedhi <[email protected]>
opensearch-project · Jan 23, 2024 · e43f9bf · e43f9bf
1 parent cba1df9
commit e43f9bf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/osbenchmark/client.py b/osbenchmark/client.py
@@ -286,10 +286,10 @@ def create(self):
 
         if "connection_class" in self.client_options:
             connection = self.client_options.pop("connection_class", None)
-            if connection.lower() in {"urllib3httpconnection", "urllib3"}:
-                connection_class = osbenchmark.sync_connection.Urllib3HttpConnection
-            elif connection.lower() in {"requestshttpconnection", "requests"}:
+            if connection.lower() in {"requestshttpconnection", "requests"}:
                 connection_class = osbenchmark.sync_connection.RequestsHttpConnection
+            else:
+                connection_class = osbenchmark.sync_connection.Urllib3HttpConnection
         else:
             connection_class = osbenchmark.sync_connection.Urllib3HttpConnection