This is an automated email from the ASF dual-hosted git repository. potiuk pushed a commit to branch v2-2-test in repository https://gitbox.apache.org/repos/asf/airflow.git
commit 17b7da5f2b340b7722a6d32bb314dc5fb1cdcdad Author: Chenglong Yan <[email protected]> AuthorDate: Mon Jan 24 07:59:10 2022 +0800 Fix running airflow dags test <dag_id> <execution_dt> results in error when run twice (#21031) related: #21023 (cherry picked from commit 515ea84335fc440fe022db2a0e3b158e0d7702da) --- 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 b94d6cf..e04bc73 100644 --- a/airflow/cli/commands/dag_command.py +++ b/airflow/cli/commands/dag_command.py @@ -406,7 +406,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 0044761..a937676 100644 --- a/tests/cli/commands/test_dag_command.py +++ b/tests/cli/commands/test_dag_command.py @@ -517,7 +517,7 @@ class TestCliDags(unittest.TestCase): 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, @@ -548,7 +548,7 @@ class TestCliDags(unittest.TestCase): 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,
