From 6b86c996bb26a426aefc60f27b574ad8f1016ffc Mon Sep 17 00:00:00 2001
From: Ankit Chaurasia <8670962+sunank200@users.noreply.github.com>
Date: Thu, 6 Feb 2025 18:57:37 +0545
Subject: [PATCH 1/2] Add logical in the tests

---
 providers/tests/amazon/aws/operators/test_dms.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/providers/tests/amazon/aws/operators/test_dms.py b/providers/tests/amazon/aws/operators/test_dms.py
index 0b414234d9fa2..058b09d8b1c5c 100644
--- a/providers/tests/amazon/aws/operators/test_dms.py
+++ b/providers/tests/amazon/aws/operators/test_dms.py
@@ -477,13 +477,13 @@ def test_init(self):
     @pytest.mark.db_test
     @mock.patch.object(DmsHook, "conn")
     def test_template_fields_native(self, mock_conn, session):
-        execution_date = timezone.datetime(2020, 1, 1)
+        logical_date = timezone.datetime(2020, 1, 1)
         Variable.set("test_filter", self.filter, session=session)
 
         dag = DAG(
             "test_dms",
             schedule=None,
-            start_date=execution_date,
+            start_date=logical_date,
             render_template_as_native_obj=True,
         )
         op = DmsDescribeReplicationConfigsOperator(
@@ -495,6 +495,7 @@ def test_template_fields_native(self, mock_conn, session):
             run_id="test",
             run_type=DagRunType.MANUAL,
             state=DagRunState.RUNNING,
+            logical_date=logical_date,
         )
         ti = TaskInstance(task=op)
         ti.dag_run = dag_run

From 6f02b84fb114f780bdd02b97e68ebf9cb458f076 Mon Sep 17 00:00:00 2001
From: Ankit Chaurasia <8670962+sunank200@users.noreply.github.com>
Date: Thu, 6 Feb 2025 19:36:01 +0545
Subject: [PATCH 2/2] Fix compact test

---
 .../tests/amazon/aws/operators/test_dms.py    | 23 +++++++++++++------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/providers/tests/amazon/aws/operators/test_dms.py b/providers/tests/amazon/aws/operators/test_dms.py
index 058b09d8b1c5c..f128ab75f2b20 100644
--- a/providers/tests/amazon/aws/operators/test_dms.py
+++ b/providers/tests/amazon/aws/operators/test_dms.py
@@ -490,13 +490,22 @@ def test_template_fields_native(self, mock_conn, session):
             task_id="test_task", filter="{{ var.value.test_filter }}", dag=dag
         )
 
-        dag_run = DagRun(
-            dag_id=dag.dag_id,
-            run_id="test",
-            run_type=DagRunType.MANUAL,
-            state=DagRunState.RUNNING,
-            logical_date=logical_date,
-        )
+        if AIRFLOW_V_3_0_PLUS:
+            dag_run = DagRun(
+                dag_id=dag.dag_id,
+                run_id="test",
+                run_type=DagRunType.MANUAL,
+                state=DagRunState.RUNNING,
+                logical_date=logical_date,
+            )
+        else:
+            dag_run = DagRun(
+                dag_id=dag.dag_id,
+                run_id="test",
+                run_type=DagRunType.MANUAL,
+                state=DagRunState.RUNNING,
+                execution_date=logical_date,
+            )
         ti = TaskInstance(task=op)
         ti.dag_run = dag_run
         session.add(ti)