Add AWS project structure tests (re: AIP-47) (#23630)

apache · May 17, 2022 · 64d0d9c · 64d0d9c
1 parent 3993cb8
commit 64d0d9c
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 1 deletion.
diff --git a/airflow/providers/amazon/aws/sensors/s3.py b/airflow/providers/amazon/aws/sensors/s3.py
@@ -155,7 +155,7 @@ def __init__(
     ):
         warnings.warn(
             """
-            S3PrefixSensor is deprecated.
+            S3KeySizeSensor is deprecated.
             Please use `airflow.providers.amazon.aws.sensors.s3.S3KeySensor`.
             """,
             DeprecationWarning,

diff --git a/tests/always/test_project_structure.py b/tests/always/test_project_structure.py
@@ -419,6 +419,64 @@ def test_missing_assets(self):
         super().test_missing_assets()
 
 
+class TestAmazonProviderProjectStructure(ExampleCoverageTest):
+    PROVIDER = "amazon"
+    CLASS_DIRS = ProjectStructureTest.CLASS_DIRS
+
+    BASE_CLASSES = {
+        'airflow.providers.amazon.aws.operators.rds.RdsBaseOperator',
+        'airflow.providers.amazon.aws.operators.sagemaker.SageMakerBaseOperator',
+        'airflow.providers.amazon.aws.sensors.dms.DmsTaskBaseSensor',
+        'airflow.providers.amazon.aws.sensors.emr.EmrBaseSensor',
+        'airflow.providers.amazon.aws.sensors.rds.RdsBaseSensor',
+        'airflow.providers.amazon.aws.sensors.sagemaker.SageMakerBaseSensor',
+    }
+
+    MISSING_EXAMPLES_FOR_CLASSES = {
+        # DMS examples DAGs are currently in development
+        'airflow.providers.amazon.aws.operators.dms.DmsDescribeTasksOperator',
+        'airflow.providers.amazon.aws.operators.dms.DmsStopTaskOperator',
+        # EMR legitimately missing, needs development
+        'airflow.providers.amazon.aws.operators.emr.EmrModifyClusterOperator',
+        'airflow.providers.amazon.aws.sensors.emr.EmrContainerSensor',
+        # S3 Exasol transfer difficult to test, see: https://github.com/apache/airflow/issues/22632
+        'airflow.providers.amazon.aws.transfers.exasol_to_s3.ExasolToS3Operator',
+        # S3 legitimately missing, needs development
+        'airflow.providers.amazon.aws.transfers.gcs_to_s3.GCSToS3Operator',
+        # Glue Catalog sensor difficult to test
+        'airflow.providers.amazon.aws.sensors.glue_catalog_partition.GlueCatalogPartitionSensor',
+    }
+
+    DEPRECATED_CLASSES = {
+        'airflow.providers.amazon.aws.operators.athena.AWSAthenaOperator',
+        'airflow.providers.amazon.aws.operators.batch.AwsBatchOperator',
+        'airflow.providers.amazon.aws.operators.datasync.AWSDataSyncOperator',
+        'airflow.providers.amazon.aws.operators.ecs.ECSOperator',
+        'airflow.providers.amazon.aws.operators.eks.EKSCreateClusterOperator',
+        'airflow.providers.amazon.aws.operators.eks.EKSCreateFargateProfileOperator',
+        'airflow.providers.amazon.aws.operators.eks.EKSCreateNodegroupOperator',
+        'airflow.providers.amazon.aws.operators.eks.EKSDeleteClusterOperator',
+        'airflow.providers.amazon.aws.operators.eks.EKSDeleteFargateProfileOperator',
+        'airflow.providers.amazon.aws.operators.eks.EKSDeleteNodegroupOperator',
+        'airflow.providers.amazon.aws.operators.eks.EKSPodOperator',
+        'airflow.providers.amazon.aws.operators.emr_containers.EMRContainerOperator',
+        'airflow.providers.amazon.aws.operators.glue.AwsGlueJobOperator',
+        'airflow.providers.amazon.aws.operators.glue_crawler.AwsGlueCrawlerOperator',
+        'airflow.providers.amazon.aws.operators.sqs.SQSPublishOperator',
+        'airflow.providers.amazon.aws.sensors.eks.EKSClusterStateSensor',
+        'airflow.providers.amazon.aws.sensors.eks.EKSFargateProfileStateSensor',
+        'airflow.providers.amazon.aws.sensors.eks.EKSNodegroupStateSensor',
+        'airflow.providers.amazon.aws.sensors.emr_containers.EMRContainerSensor',
+        'airflow.providers.amazon.aws.sensors.glue.AwsGlueJobSensor',
+        'airflow.providers.amazon.aws.sensors.glue_catalog_partition.AwsGlueCatalogPartitionSensor',
+        'airflow.providers.amazon.aws.sensors.glue_crawler.AwsGlueCrawlerSensor',
+        'airflow.providers.amazon.aws.sensors.s3.S3KeySizeSensor',
+        'airflow.providers.amazon.aws.sensors.s3.S3PrefixSensor',
+        'airflow.providers.amazon.aws.sensors.sqs.SQSSensor',
+        'airflow.providers.amazon.aws.transfers.mysql_to_s3.MySQLToS3Operator',
+    }
+
+
 class TestElasticsearchProviderProjectStructure(ExampleCoverageTest):
     PROVIDER = "elasticsearch"
     CLASS_DIRS = {"hooks"}