Disable failover tests based on conditions

Author: faviansamatha

tests/integration/container/test_aws_secrets_manager.py

@@ -23,7 +23,9 @@
 from aws_advanced_python_wrapper.errors import (AwsWrapperError,
                                                 FailoverSuccessError)
 from aws_advanced_python_wrapper.utils.properties import Properties
-from .utils.conditions import enable_on_features
+from .utils.conditions import (disable_on_features, enable_on_deployments,
+                               enable_on_features, enable_on_num_instances)
+from .utils.database_engine_deployment import DatabaseEngineDeployment
 from .utils.driver_helper import DriverHelper
 from .utils.rds_test_utility import RdsTestUtility
 from .utils.test_environment import TestEnvironment
@@ -196,7 +198,12 @@ def test_incorrect_region(self, test_driver, conn_utils, create_secret, props):
             ) as conn:
                 conn.cursor()
 
-    @enable_on_features([TestEnvironmentFeatures.FAILOVER_SUPPORTED])
+    @enable_on_num_instances(min_instances=2)
+    @enable_on_deployments([DatabaseEngineDeployment.AURORA, DatabaseEngineDeployment.RDS_MULTI_AZ_CLUSTER])
+    @disable_on_features([TestEnvironmentFeatures.RUN_AUTOSCALING_TESTS_ONLY,
+                          TestEnvironmentFeatures.BLUE_GREEN_DEPLOYMENT,
+                          TestEnvironmentFeatures.PERFORMANCE])
+    @enable_on_features([TestEnvironmentFeatures.FAILOVER_SUPPORTED, TestEnvironmentFeatures.IAM])
     def test_failover_with_secrets_manager(
             self, test_driver, props, conn_utils, create_secret):
         region = TestEnvironment.get_current().get_info().get_region()

tests/integration/container/test_iam_authentication.py

@@ -33,8 +33,11 @@
 from aws_advanced_python_wrapper import AwsWrapperConnection
 from aws_advanced_python_wrapper.errors import (AwsWrapperError,
                                                 FailoverSuccessError)
-from tests.integration.container.utils.conditions import (disable_on_features,
-                                                          enable_on_features)
+from tests.integration.container.utils.conditions import (
+    disable_on_features, enable_on_deployments, enable_on_features,
+    enable_on_num_instances)
+from tests.integration.container.utils.database_engine_deployment import \
+    DatabaseEngineDeployment
 from tests.integration.container.utils.driver_helper import DriverHelper
 from tests.integration.container.utils.rds_test_utility import RdsTestUtility
 from tests.integration.container.utils.test_environment import TestEnvironment
@@ -127,6 +130,11 @@ def test_iam_valid_connection_properties_no_password(
 
         self.validate_connection(target_driver_connect, **params, **props)
 
+    @enable_on_num_instances(min_instances=2)
+    @enable_on_deployments([DatabaseEngineDeployment.AURORA, DatabaseEngineDeployment.RDS_MULTI_AZ_CLUSTER])
+    @disable_on_features([TestEnvironmentFeatures.RUN_AUTOSCALING_TESTS_ONLY,
+                          TestEnvironmentFeatures.BLUE_GREEN_DEPLOYMENT,
+                          TestEnvironmentFeatures.PERFORMANCE])
     @enable_on_features([TestEnvironmentFeatures.FAILOVER_SUPPORTED, TestEnvironmentFeatures.IAM])
     def test_failover_with_iam(
             self, test_environment: TestEnvironment, test_driver: TestDriver, props, conn_utils):