From a8df8ffd2fd3ffc49263e22e829bb642197d19b4 Mon Sep 17 00:00:00 2001 From: Karthikeyan Singaravelan Date: Mon, 25 Apr 2022 19:53:32 +0530 Subject: [PATCH] Fix extra_links endpoint. --- airflow/www/views.py | 2 +- tests/www/views/test_views.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/airflow/www/views.py b/airflow/www/views.py index 14a572ad3f2f4..f89dda378da58 100644 --- a/airflow/www/views.py +++ b/airflow/www/views.py @@ -3427,7 +3427,7 @@ def extra_links(self, session: "Session" = NEW_SESSION): map_index = request.args.get('map_index', -1, type=int) execution_date = request.args.get('execution_date') link_name = request.args.get('link_name') - dttm = timezone.parse(execution_date) + dttm = _safe_parse_datetime(execution_date) dag = current_app.dag_bag.get_dag(dag_id) if not dag or task_id not in dag.task_ids: diff --git a/tests/www/views/test_views.py b/tests/www/views/test_views.py index 300b23d0323bd..887bd4898a0a6 100644 --- a/tests/www/views/test_views.py +++ b/tests/www/views/test_views.py @@ -418,6 +418,10 @@ def test_get_task_stats_from_query(): 'dags/example_bash_operator/gantt?execution_date=invalid', "Invalid datetime: 'invalid'", ), + ( + 'extra_links?execution_date=invalid', + "Invalid datetime: 'invalid'", + ), ], ) def test_invalid_dates(app, admin_client, url, content):