Fix http exception when ti not found for extra links API (#51443)

amoghrajesh · web-flow · commit ce4e9e8ea25e · 2025-06-06T11:43:20.000+05:30
* Fix http exception when ti not found for extra links API

* no question of dagrun here, remove that test
diff --git a/airflow-core/src/airflow/api_fastapi/core_api/routes/public/extra_links.py b/airflow-core/src/airflow/api_fastapi/core_api/routes/public/extra_links.py
@@ -77,7 +77,7 @@ def get_extra_links(
     if not ti:
         raise HTTPException(
             status.HTTP_404_NOT_FOUND,
-            f"DAG Run with ID = {dag_run_id} not found",
+            "TaskInstance not found",
         )
 
     all_extra_link_pairs = (
diff --git a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_extra_links.py b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_extra_links.py
@@ -131,12 +131,6 @@ def _create_dag(self, dag_maker):
                 {"detail": "DAG with ID = INVALID not found"},
                 id="missing_dag",
             ),
-            pytest.param(
-                "/dags/TEST_DAG_ID/dagRuns/INVALID/taskInstances/TEST_SINGLE_LINK/links",
-                404,
-                {"detail": "DAG Run with ID = INVALID not found"},
-                id="missing_dag_run",
-            ),
             pytest.param(
                 "/dags/TEST_DAG_ID/dagRuns/TEST_DAG_RUN_ID/taskInstances/INVALID/links",
                 404,

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ def get_extra_links(`
`77`	`77`	`if not ti:`
`78`	`78`	`raise HTTPException(`
`79`	`79`	`status.HTTP_404_NOT_FOUND,`
`80`		`- f"DAG Run with ID = {dag_run_id} not found",`
	`80`	`+ "TaskInstance not found",`
`81`	`81`	`)`
`82`	`82`
`83`	`83`	`all_extra_link_pairs = (`