integ test debug

Author: faviansamatha

tests/integration/container/tortoise/test_tortoise_config.py

@@ -22,8 +22,10 @@
 from tests.integration.container.tortoise.test_tortoise_common import \
     reset_tortoise
 from tests.integration.container.utils.conditions import (disable_on_engines,
-                                                          disable_on_features)
+                                                          disable_on_features,
+                                                          disable_on_deployments)
 from tests.integration.container.utils.database_engine import DatabaseEngine
+from tests.integration.container.utils.database_engine_deployment import DatabaseEngineDeployment
 from tests.integration.container.utils.test_environment_features import \
     TestEnvironmentFeatures
 
@@ -43,17 +45,18 @@ async def _clear_all_test_models(self):
     async def setup_tortoise_dict_config(self, conn_utils):
         """Setup Tortoise with dictionary configuration instead of URL."""
         # Ensure clean state
+        host = conn_utils.writer_cluster_host if conn_utils.writer_cluster_host else conn_utils.writer_host
         config = {
             "connections": {
                 "default": {
                     "engine": "aws_advanced_python_wrapper.tortoise_orm.backends.mysql",
                     "credentials": {
-                        "host": conn_utils.writer_cluster_host,
+                        "host": host,
                         "port": conn_utils.port,
                         "user": conn_utils.user,
                         "password": conn_utils.password,
                         "database": conn_utils.dbname,
-                        "plugins": "aurora_connection_tracker,failover",
+                        "plugins": "aurora_connection_tracker",
                     }
                 }
             },
@@ -80,29 +83,29 @@ async def setup_tortoise_multi_db(self, conn_utils):
         # Create second database name
         original_db = conn_utils.dbname
         second_db = f"{original_db}_test2"
-
+        host = conn_utils.writer_cluster_host if conn_utils.writer_cluster_host else conn_utils.writer_host
         config = {
             "connections": {
                 "default": {
                     "engine": "aws_advanced_python_wrapper.tortoise_orm.backends.mysql",
                     "credentials": {
-                        "host": conn_utils.writer_cluster_host,
+                        "host": host,
                         "port": conn_utils.port,
                         "user": conn_utils.user,
                         "password": conn_utils.password,
                         "database": original_db,
-                        "plugins": "aurora_connection_tracker,failover",
+                        "plugins": "aurora_connection_tracker",
                     }
                 },
                 "second_db": {
                     "engine": "aws_advanced_python_wrapper.tortoise_orm.backends.mysql",
                     "credentials": {
-                        "host": conn_utils.writer_cluster_host,
+                        "host": host,
                         "port": conn_utils.port,
                         "user": conn_utils.user,
                         "password": conn_utils.password,
                         "database": second_db,
-                        "plugins": "aurora_connection_tracker,failover"
+                        "plugins": "aurora_connection_tracker"
                     }
                 }
             },
@@ -142,17 +145,18 @@ async def setup_tortoise_multi_db(self, conn_utils):
     @pytest_asyncio.fixture
     async def setup_tortoise_with_router(self, conn_utils):
         """Setup Tortoise with router configuration."""
+        host = conn_utils.writer_cluster_host if conn_utils.writer_cluster_host else conn_utils.writer_host
         config = {
             "connections": {
                 "default": {
                     "engine": "aws_advanced_python_wrapper.tortoise_orm.backends.mysql",
                     "credentials": {
-                        "host": conn_utils.writer_cluster_host,
+                        "host": host,
                         "port": conn_utils.port,
                         "user": conn_utils.user,
                         "password": conn_utils.password,
                         "database": conn_utils.dbname,
-                        "plugins": "aurora_connection_tracker,failover"
+                        "plugins": "aurora_connection_tracker"
                     }
                 }
             },
@@ -210,6 +214,7 @@ async def test_dict_config_write_operations(self, setup_tortoise_dict_config):
         with pytest.raises(Exception):
             await User.get(id=user.id)
 
+    @disable_on_deployments([DatabaseEngineDeployment.DOCKER])
     @pytest.mark.asyncio
     async def test_multi_db_operations(self, setup_tortoise_multi_db):
         """Test operations with multiple databases using same backend."""

tests/integration/container/utils/conditions.py

@@ -87,3 +87,11 @@ def disable_on_features(disable_on_test_features: List[TestEnvironmentFeatures])
         disable_test,
         reason="The current test environment contains test features for which this test is disabled"
     )
+
+
+def disable_on_deployments(requested_deployments: List[DatabaseEngineDeployment]):
+    current_deployment = TestEnvironment.get_current().get_deployment()
+    return pytest.mark.skipif(
+        current_deployment in requested_deployments,
+        reason=f"This test is not supported for {current_deployment.value} deployments"
+    )

tests/integration/container/utils/connection_utils.py

@@ -109,7 +109,6 @@ def get_aws_tortoise_url(
             dbname: Optional[str] = None,
             **kwargs) -> str:
         """Build AWS MySQL connection URL for Tortoise ORM with query parameters."""
-
         env_host = self.writer_cluster_host if self.writer_cluster_host else self.writer_host
         host = env_host if host is None else host
         port = self.port if port is None else port