Fixup tests

Author: gopidesupavan

airflow-core/tests/unit/jobs/test_triggerer_job.py

@@ -1228,6 +1228,7 @@ def get_type_names(union_type):
             "ResendLoggingFD",
             "CreateHITLDetailPayload",
             "SetRenderedMapIndex",
+            "GetDagState",
         }
 
         in_task_but_not_in_trigger_runner = {
@@ -1247,6 +1248,7 @@ def get_type_names(union_type):
             "PreviousDagRunResult",
             "PreviousTIResult",
             "HITLDetailRequestResult",
+            "DagStateResult",
         }
 
         supervisor_diff = (

task-sdk/tests/task_sdk/execution_time/test_supervisor.py

@@ -1422,7 +1422,8 @@ class RequestTestCase:
         client_mock=ClientMock(
             method_path="connections.get",
             args=("test_conn",),
-            response=ConnectionResult(conn_id="test_conn", conn_type="mysql", schema="mysql"),  # type: ignore[call-arg]
+            response=ConnectionResult(conn_id="test_conn", conn_type="mysql", schema="mysql"),
+            # type: ignore[call-arg]
         ),
         expected_body={
             "conn_id": "test_conn",
@@ -2459,10 +2460,12 @@ class RequestTestCase:
         expected_body={"is_paused": False, "type": "DagStateResult"},
         client_mock=ClientMock(
             method_path="dags.get_state",
-            args=("test_dag",),
+            kwargs={
+                "dag_id": "test_dag",
+            },
             response=DagStateResult(is_paused=False),
         ),
-        test_id="get_dag_run_state",
+        test_id="get_dag_state",
     ),
 ]