You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@airflow.apache.org by po...@apache.org on 2022/01/24 17:35:46 UTC
[airflow] 02/04: Fix running airflow dags test results in error when run twice (#21031)
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 <al...@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,