[airflow] branch v2-2-test updated: Fix running airflow dags test results in error when run twice (#21031)

[potiuk]

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


The following commit(s) were added to refs/heads/v2-2-test by this push:
     new 8edcf10  Fix running airflow dags test <dag_id> <execution_dt> results 
in error when run twice (#21031)
8edcf10 is described below

commit 8edcf10a17d29690618ba98fa5a385fff2e90c9b
Author: Chenglong Yan <alanx....@gmail.com>
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,