modified after local test

jx2lee · jx2lee · commit bdf647ff660d · 2025-05-15T22:08:13.000+09:00
diff --git a/airflow-ctl/src/airflowctl/api/operations.py b/airflow-ctl/src/airflowctl/api/operations.py
@@ -397,10 +397,10 @@ def patch(self, dag_id: str, dag_body: DAGPatchBody) -> DAGResponse | ServerResp
         except ServerResponseError as e:
             raise e
 
-    def delete(self, dag_id: str) -> DAGResponse | ServerResponseError:
+    def delete(self, dag_id: str) -> str | ServerResponseError:
         try:
             self.response = self.client.delete(f"dags/{dag_id}")
-            return DAGResponse.model_validate_json(self.response.content)
+            return dag_id
         except ServerResponseError as e:
             raise e
 
@@ -439,9 +439,9 @@ def list_version(self, dag_id: str) -> DAGVersionCollectionResponse | ServerResp
         except ServerResponseError as e:
             raise e
 
-    def list_warning(self, dag_id: str) -> DAGWarningCollectionResponse | ServerResponseError:
+    def list_warning(self) -> DAGWarningCollectionResponse | ServerResponseError:
         try:
-            self.response = self.client.get(f"dags/{dag_id}/dagWarnings")
+            self.response = self.client.get("dagWarnings")
             return DAGWarningCollectionResponse.model_validate_json(self.response.content)
         except ServerResponseError as e:
             raise e
diff --git a/airflow-ctl/tests/airflow_ctl/api/test_operations.py b/airflow-ctl/tests/airflow_ctl/api/test_operations.py
@@ -621,7 +621,7 @@ def handle_request(request: httpx.Request) -> httpx.Response:
 
         client = make_api_client(transport=httpx.MockTransport(handle_request))
         response = client.dags.delete(dag_id="dag_id")
-        assert response == self.dag_response
+        assert response == self.dag_id
 
     def test_get_import_error(self):
         def handle_request(request: httpx.Request) -> httpx.Response:
@@ -661,7 +661,7 @@ def handle_request(request: httpx.Request) -> httpx.Response:
         response = client.dags.get_version(dag_id="dag_id", version_number="0")
         assert response == self.dag_version_response
 
-    def list_version(self):
+    def test_list_version(self):
         def handle_request(request: httpx.Request) -> httpx.Response:
             assert request.url.path == "/api/v2/dags/dag_id/dagVersions"
             return httpx.Response(
@@ -672,15 +672,15 @@ def handle_request(request: httpx.Request) -> httpx.Response:
         response = client.dags.list_version(dag_id="dag_id")
         assert response == self.dag_version_collection_response
 
-    def list_warning(self):
+    def test_list_warning(self):
         def handle_request(request: httpx.Request) -> httpx.Response:
-            assert request.url.path == "/api/v2/dags/dag_id/dagWarnings"
+            assert request.url.path == "/api/v2/dagWarnings"
             return httpx.Response(
                 200, json=json.loads(self.dag_warning_collection_response.model_dump_json())
             )
 
         client = make_api_client(transport=httpx.MockTransport(handle_request))
-        response = client.dags.list_warning(dag_id="dag_id")
+        response = client.dags.list_warning()
         assert response == self.dag_warning_collection_response
 
 
