This is an automated email from the ASF dual-hosted git repository.
vincbeck pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new bbe3991d216 Add missing Amazon utility tests (#72985)
bbe3991d216 is described below
commit bbe3991d21638388bdad2ca06b0ed3c22d32016b
Author: Vincent Hsiao <[email protected]>
AuthorDate: Mon Sep 14 22:18:49 2026 +0800
Add missing Amazon utility tests (#72985)
---
.../tests/unit/always/test_project_structure.py | 3 ---
...trigger.py => test_sagemaker_unified_studio.py} | 0
.../amazon/tests/unit/amazon/aws/utils/test_rds.py | 26 +++++++++++++++++++++
.../tests/unit/amazon/aws/utils/test_sagemaker.py | 27 ++++++++++++++++++++++
4 files changed, 53 insertions(+), 3 deletions(-)
diff --git a/airflow-core/tests/unit/always/test_project_structure.py
b/airflow-core/tests/unit/always/test_project_structure.py
index 7ac2256b27f..1bbad7dca94 100644
--- a/airflow-core/tests/unit/always/test_project_structure.py
+++ b/airflow-core/tests/unit/always/test_project_structure.py
@@ -72,9 +72,6 @@ class TestProjectStructure:
"providers/amazon/tests/unit/amazon/aws/operators/test_sagemaker.py",
"providers/amazon/tests/unit/amazon/aws/sensors/test_emr.py",
"providers/amazon/tests/unit/amazon/aws/sensors/test_sagemaker.py",
-
"providers/amazon/tests/unit/amazon/aws/triggers/test_sagemaker_unified_studio.py",
- "providers/amazon/tests/unit/amazon/aws/utils/test_rds.py",
- "providers/amazon/tests/unit/amazon/aws/utils/test_sagemaker.py",
"providers/apache/hdfs/tests/unit/apache/hdfs/hooks/test_hdfs.py",
"providers/apache/hdfs/tests/unit/apache/hdfs/sensors/test_hdfs.py",
"providers/apache/hive/tests/unit/apache/hive/plugins/test_hive.py",
diff --git
a/providers/amazon/tests/unit/amazon/aws/triggers/test_sagemaker_unified_studio_trigger.py
b/providers/amazon/tests/unit/amazon/aws/triggers/test_sagemaker_unified_studio.py
similarity index 100%
rename from
providers/amazon/tests/unit/amazon/aws/triggers/test_sagemaker_unified_studio_trigger.py
rename to
providers/amazon/tests/unit/amazon/aws/triggers/test_sagemaker_unified_studio.py
diff --git a/providers/amazon/tests/unit/amazon/aws/utils/test_rds.py
b/providers/amazon/tests/unit/amazon/aws/utils/test_rds.py
new file mode 100644
index 00000000000..8614e0614e5
--- /dev/null
+++ b/providers/amazon/tests/unit/amazon/aws/utils/test_rds.py
@@ -0,0 +1,26 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from airflow.providers.amazon.aws.utils.rds import RdsDbType
+
+
+def test_rds_db_type_values():
+ assert {db_type.name: db_type.value for db_type in RdsDbType} == {
+ "INSTANCE": "instance",
+ "CLUSTER": "cluster",
+ }
diff --git a/providers/amazon/tests/unit/amazon/aws/utils/test_sagemaker.py
b/providers/amazon/tests/unit/amazon/aws/utils/test_sagemaker.py
new file mode 100644
index 00000000000..380ca991842
--- /dev/null
+++ b/providers/amazon/tests/unit/amazon/aws/utils/test_sagemaker.py
@@ -0,0 +1,27 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from airflow.providers.amazon.aws.utils.sagemaker import ApprovalStatus
+
+
+def test_approval_status_values():
+ assert {status.name: status.value for status in ApprovalStatus} == {
+ "APPROVED": "Approved",
+ "REJECTED": "Rejected",
+ "PENDING_MANUAL_APPROVAL": "PendingManualApproval",
+ }
