From 515ea84335fc440fe022db2a0e3b158e0d7702da Mon Sep 17 00:00:00 2001 From: Chenglong Yan Date: Mon, 24 Jan 2022 07:59:10 +0800 Subject: [PATCH] Fix running airflow dags test results in error when run twice (#21031) related: #21023 --- airflow/cli/commands/dag_command.py | 2 +- tests/cli/commands/test_dag_command.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/airflow/cli/commands/dag_command.py b/airflow/cli/commands/dag_command.py index 2e537fb86ce6f..cb18e0c3faa26 100644 --- a/airflow/cli/commands/dag_command.py +++ b/airflow/cli/commands/dag_command.py @@ -426,7 +426,7 @@ def dag_list_dag_runs(args, dag=None): def dag_test(args, session=None): """Execute one single DagRun for a given DAG and execution date, using the DebugExecutor.""" dag = get_dag(subdir=args.subdir, dag_id=args.dag_id) - dag.clear(start_date=args.execution_date, end_date=args.execution_date, dag_run_state=State.NONE) + dag.clear(start_date=args.execution_date, end_date=args.execution_date, dag_run_state=False) try: dag.run( executor=DebugExecutor(), diff --git a/tests/cli/commands/test_dag_command.py b/tests/cli/commands/test_dag_command.py index 097a54dd2e2e5..cea282980e2f2 100644 --- a/tests/cli/commands/test_dag_command.py +++ b/tests/cli/commands/test_dag_command.py @@ -559,7 +559,7 @@ def test_dag_test(self, mock_get_dag, mock_executor): mock.call().clear( start_date=cli_args.execution_date, end_date=cli_args.execution_date, - dag_run_state=State.NONE, + dag_run_state=False, ), mock.call().run( executor=mock_executor.return_value, @@ -588,7 +588,7 @@ def test_dag_test_show_dag(self, mock_get_dag, mock_executor, mock_render_dag): mock.call().clear( start_date=cli_args.execution_date, end_date=cli_args.execution_date, - dag_run_state=State.NONE, + dag_run_state=False, ), mock.call().run( executor=mock_executor.return_value,