fixup! fixup! more test fixes

Author: ephraimbuddy

airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_task_instances.py

@@ -112,7 +112,7 @@ def create_task_instances(
         run_id = "TEST_DAG_RUN_ID"
         logical_date = self.ti_init.pop("logical_date", self.default_time)
         dr = None
-        dag_version = DagVersion.get_latest_version(dag.dag_id)
+        dag_version = DagVersion.get_latest_version(dag.dag_id, session=session)
         tis = []
         for i in range(counter):
             if task_instances is None:
@@ -3926,7 +3926,9 @@ def test_should_respond_403(self, unauthorized_test_client):
     def test_should_not_update_mapped_task_instance(self, test_client, session):
         map_index = 1
         tis = self.create_task_instances(session)
-        ti = TaskInstance(task=tis[0].task, run_id=tis[0].run_id, map_index=map_index)
+        ti = TaskInstance(
+            task=tis[0].task, run_id=tis[0].run_id, map_index=map_index, dag_version_id=tis[0].dag_version_id
+        )
         ti.rendered_task_instance_fields = RTIF(ti, render_templates=False)
         session.add(ti)
         session.commit()
@@ -3957,6 +3959,7 @@ def test_should_not_update_mapped_task_instance_summary(self, test_client, sessi
                 run_id=tis[0].run_id,
                 map_index=map_index,
                 state="running",
+                dag_version_id=tis[0].dag_version_id,
             )
             ti.rendered_task_instance_fields = RTIF(ti, render_templates=False)
             session.add(ti)

airflow-core/tests/unit/callbacks/test_callback_requests.py

@@ -16,8 +16,8 @@
 # under the License.
 from __future__ import annotations
 
+import uuid
 from datetime import datetime
-from unittest import mock
 
 import pytest
 
@@ -66,7 +66,7 @@ def test_from_json(self, input, request_class):
                 ),
                 run_id="fake_run",
                 state=State.RUNNING,
-                dag_version_id=mock.MagicMock(),
+                dag_version_id=uuid.uuid4(),
             )
             ti.start_date = timezone.utcnow()