Resolve core operators deprecations in tests (apache#40267)

romsharon98 · Jul 26, 2024 · 0fda8fb · 0fda8fb
1 parent 7da6ebf
commit 0fda8fb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/tests/deprecations_ignore.yml b/tests/deprecations_ignore.yml
@@ -37,12 +37,9 @@
 
 
 # Core
-- tests/models/test_cleartasks.py::TestClearTasks::test_dags_clear
 - tests/models/test_dagbag.py::TestDagBag::test_load_subdags
 - tests/models/test_mappedoperator.py::test_expand_mapped_task_instance_with_named_index
-- tests/models/test_skipmixin.py::TestSkipMixin::test_skip_none_dagrun
 - tests/models/test_xcom.py::TestXCom::test_set_serialize_call_old_signature
-- tests/ti_deps/deps/test_prev_dagrun_dep.py::TestPrevDagrunDep::test_first_task_run_of_new_task
 
 
 # Serialization

diff --git a/tests/models/test_cleartasks.py b/tests/models/test_cleartasks.py
@@ -642,6 +642,7 @@ def test_dags_clear(self):
                 state=State.RUNNING,
                 run_type=DagRunType.SCHEDULED,
                 session=session,
+                data_interval=(DEFAULT_DATE, DEFAULT_DATE),
             )
             ti = dr.task_instances[0]
             ti.task = task

diff --git a/tests/ti_deps/deps/test_prev_dagrun_dep.py b/tests/ti_deps/deps/test_prev_dagrun_dep.py
@@ -56,6 +56,7 @@ def test_first_task_run_of_new_task(self):
             state=TaskInstanceState.SUCCESS,
             execution_date=old_task.start_date,
             run_type=DagRunType.SCHEDULED,
+            data_interval=(old_task.start_date, old_task.start_date),
         )
 
         new_task = BaseOperator(
@@ -67,11 +68,13 @@ def test_first_task_run_of_new_task(self):
         )
 
         # New DAG run will include 1st TaskInstance of new_task
+        execution_date = convert_to_utc(datetime(2016, 1, 2))
         dr = dag.create_dagrun(
             run_id="new_run",
             state=DagRunState.RUNNING,
-            execution_date=convert_to_utc(datetime(2016, 1, 2)),
+            execution_date=execution_date,
             run_type=DagRunType.SCHEDULED,
+            data_interval=(execution_date, execution_date),
         )
 
         ti = dr.get_task_instance(new_task.task_id)