diff --git a/providers/amazon/tests/unit/amazon/aws/triggers/test_bedrock.py b/providers/amazon/tests/unit/amazon/aws/triggers/test_bedrock.py index 51f0dad49b5be..390ef64d72893 100644 --- a/providers/amazon/tests/unit/amazon/aws/triggers/test_bedrock.py +++ b/providers/amazon/tests/unit/amazon/aws/triggers/test_bedrock.py @@ -33,6 +33,8 @@ BASE_TRIGGER_CLASSPATH = "airflow.providers.amazon.aws.triggers.bedrock." +pytest.importorskip("aiobotocore") + class TestBaseBedrockTrigger: EXPECTED_WAITER_NAME: str | None = None diff --git a/providers/amazon/tests/unit/amazon/aws/triggers/test_comprehend.py b/providers/amazon/tests/unit/amazon/aws/triggers/test_comprehend.py index bf58a9853f235..cd98606697cdf 100644 --- a/providers/amazon/tests/unit/amazon/aws/triggers/test_comprehend.py +++ b/providers/amazon/tests/unit/amazon/aws/triggers/test_comprehend.py @@ -31,6 +31,8 @@ BASE_TRIGGER_CLASSPATH = "airflow.providers.amazon.aws.triggers.comprehend." +pytest.importorskip("aiobotocore") + class TestBaseComprehendTrigger: EXPECTED_WAITER_NAME: str | None = None diff --git a/providers/amazon/tests/unit/amazon/aws/triggers/test_glue.py b/providers/amazon/tests/unit/amazon/aws/triggers/test_glue.py index 2be4db335b1ac..5bb3b44fe86ab 100644 --- a/providers/amazon/tests/unit/amazon/aws/triggers/test_glue.py +++ b/providers/amazon/tests/unit/amazon/aws/triggers/test_glue.py @@ -36,6 +36,8 @@ BASE_TRIGGER_CLASSPATH = "airflow.providers.amazon.aws.triggers.glue." +pytest.importorskip("aiobotocore") + class TestGlueJobTrigger: @pytest.mark.asyncio diff --git a/providers/amazon/tests/unit/amazon/aws/triggers/test_glue_crawler.py b/providers/amazon/tests/unit/amazon/aws/triggers/test_glue_crawler.py index b4a6ef4aa6883..c21e982bf8f22 100644 --- a/providers/amazon/tests/unit/amazon/aws/triggers/test_glue_crawler.py +++ b/providers/amazon/tests/unit/amazon/aws/triggers/test_glue_crawler.py @@ -26,6 +26,8 @@ from airflow.triggers.base import TriggerEvent from unit.amazon.aws.utils.test_waiter import assert_expected_waiter_type +pytest.importorskip("aiobotocore") + class TestGlueCrawlerCompleteTrigger: def test_serialization(self): diff --git a/providers/amazon/tests/unit/amazon/aws/triggers/test_kinesis_analytics.py b/providers/amazon/tests/unit/amazon/aws/triggers/test_kinesis_analytics.py index dc6cc20937f7a..494f287783eb2 100644 --- a/providers/amazon/tests/unit/amazon/aws/triggers/test_kinesis_analytics.py +++ b/providers/amazon/tests/unit/amazon/aws/triggers/test_kinesis_analytics.py @@ -30,6 +30,8 @@ BASE_TRIGGER_CLASSPATH = "airflow.providers.amazon.aws.triggers.kinesis_analytics." +pytest.importorskip("aiobotocore") + class TestKinesisAnalyticsV2ApplicationOperationCompleteTrigger: APPLICATION_NAME = "demo" diff --git a/providers/amazon/tests/unit/amazon/aws/triggers/test_opensearch_serverless.py b/providers/amazon/tests/unit/amazon/aws/triggers/test_opensearch_serverless.py index b96c0134e0b1b..d4707f89a5360 100644 --- a/providers/amazon/tests/unit/amazon/aws/triggers/test_opensearch_serverless.py +++ b/providers/amazon/tests/unit/amazon/aws/triggers/test_opensearch_serverless.py @@ -31,6 +31,8 @@ BASE_TRIGGER_CLASSPATH = "airflow.providers.amazon.aws.triggers.opensearch_serverless." +pytest.importorskip("aiobotocore") + class TestBaseBedrockTrigger(TestAwsBaseWaiterTrigger): EXPECTED_WAITER_NAME: str | None = None