Fixup tests

Author: gopidesupavan

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

@@ -69,6 +69,7 @@
     ConnectionResult,
     CreateHITLDetailPayload,
     DagRunStateResult,
+    DagStateResult,
     DeferTask,
     DeleteVariable,
     DeleteXCom,
@@ -80,6 +81,7 @@
     GetAssetEventByAssetAlias,
     GetConnection,
     GetDagRunState,
+    GetDagState,
     GetDRCount,
     GetHITLDetailResponse,
     GetPreviousDagRun,
@@ -2288,6 +2290,16 @@ class RequestTestCase:
         ),
         test_id="skip_downstream_tasks",
     ),
+    RequestTestCase(
+        message=GetDagState(dag_id="test_dag"),
+        expected_body={"is_paused": False, "type": "DagStateResult"},
+        client_mock=ClientMock(
+            method_path="dags.get_state",
+            args=("test_dag",),
+            response=DagStateResult(is_paused=False),
+        ),
+        test_id="get_dag_run_state",
+    ),
 ]