Align pool_maxsize for different connection pool implementations. (#535)

* Align pool_maxsize for different connection pool implementations. Signed-off-by: dblock <[email protected]> * Document connection classes and settings. Signed-off-by: dblock <[email protected]> * Undo change in async for backwards compatibility. Signed-off-by: dblock <[email protected]> * Fix: typo. Signed-off-by: dblock <[email protected]> --------- Signed-off-by: dblock <[email protected]>
2023-10-12 18:55:31 -04:00
parent 62b408bbd1
commit 7a638cdafc
9 changed files with 166 additions and 5 deletions
@@ -0,0 +1,32 @@
+# SPDX-License-Identifier: Apache-2.0
+#
+# The OpenSearch Contributors require contributions made to
+# this file be licensed under the Apache-2.0 license or a
+# compatible open source license.
+#
+# Modifications Copyright OpenSearch Contributors. See
+# GitHub history for details.
+
+from unittest import TestCase
+
+from opensearchpy import OpenSearch, RequestsHttpConnection
+
+
+class TestRequests(TestCase):
+    def test_connection_class(self):
+        client = OpenSearch(connection_class=RequestsHttpConnection)
+        self.assertEqual(client.transport.pool_maxsize, None)
+        self.assertEqual(client.transport.connection_class, RequestsHttpConnection)
+        self.assertIsInstance(
+            client.transport.connection_pool.connections[0], RequestsHttpConnection
+        )
+
+    def test_pool_maxsize(self):
+        client = OpenSearch(connection_class=RequestsHttpConnection, pool_maxsize=42)
+        self.assertEqual(client.transport.pool_maxsize, 42)
+        self.assertEqual(
+            client.transport.connection_pool.connections[0]
+            .session.adapters["https://"]
+            ._pool_maxsize,
+            42,
+        )