test: add timeouts to prevent integration tests from hanging

Author: karenc-bq

tests/integration/container/test_aurora_failover.py

@@ -56,7 +56,15 @@ def aurora_utility(self):
 
     @pytest.fixture(scope='class')
     def props(self):
-        p: Properties = Properties({"plugins": "failover", "connect_timeout": 60, "topology_refresh_ms": 10, "autocommit": True})
+        p: Properties = Properties({
+            "plugins": "failover",
+            "socket_timeout": 30,
+            "connect_timeout": 10,
+            "monitoring-connect_timeout": 5,
+            "monitoring-socket_timeout": 5,
+            "topology_refresh_ms": 10,
+            "autocommit": True
+        })
 
         features = TestEnvironment.get_current().get_features()
         if TestEnvironmentFeatures.TELEMETRY_TRACES_ENABLED in features \

tests/integration/container/test_basic_connectivity.py

@@ -131,11 +131,15 @@ def test_proxied_wrapper_connection_failed(
     @enable_on_deployments([DatabaseEngineDeployment.AURORA, DatabaseEngineDeployment.RDS_MULTI_AZ_CLUSTER])
     @enable_on_features([TestEnvironmentFeatures.ABORT_CONNECTION_SUPPORTED])
     def test_wrapper_connection_reader_cluster_with_efm_enabled(self, test_driver: TestDriver, conn_utils, plugins):
+        props: Properties = Properties({
+            WrapperProperties.PLUGINS.name: plugins,
+            "socket_timeout": 5,
+            "connect_timeout": 5,
+            "monitoring-connect_timeout": 3,
+            "monitoring-socket_timeout": 3,
+            "autocommit": True})
         target_driver_connect = DriverHelper.get_connect_func(test_driver)
-        conn = AwsWrapperConnection.connect(
-            target_driver_connect,
-            **conn_utils.get_connect_params(conn_utils.reader_cluster_host),
-            plugins=plugins, connect_timeout=10)
+        conn = AwsWrapperConnection.connect(target_driver_connect, **conn_utils.get_connect_params(conn_utils.reader_cluster_host), **props)
         cursor = conn.cursor()
         cursor.execute("SELECT 1")
         result = cursor.fetchone()

tests/integration/container/test_host_monitoring_v2.py

@@ -22,7 +22,7 @@
 from aws_advanced_python_wrapper.utils.properties import (Properties,
                                                           WrapperProperties)
 from tests.integration.container.utils.conditions import (
-    disable_on_features, enable_on_deployments, disable_on_engines)
+    disable_on_engines, disable_on_features, enable_on_deployments)
 from tests.integration.container.utils.database_engine import DatabaseEngine
 from tests.integration.container.utils.database_engine_deployment import \
     DatabaseEngineDeployment

tests/integration/container/test_read_write_splitting.py

@@ -82,7 +82,13 @@ def props(self):
     @pytest.fixture(scope='class')
     def failover_props(self):
         return {
-            "plugins": "read_write_splitting,failover", "connect_timeout": 10, "autocommit": True}
+            "plugins": "read_write_splitting,failover",
+            "socket_timeout": 30,
+            "connect_timeout": 10,
+            "monitoring-connect_timeout": 5,
+            "monitoring-socket_timeout": 5,
+            "autocommit": True
+        }
 
     @pytest.fixture(scope='class')
     def proxied_props(self, props, conn_utils):