diff --git a/airflow-core/tests/unit/jobs/test_triggerer_job.py b/airflow-core/tests/unit/jobs/test_triggerer_job.py index 2ec8733992e27..bd390c82cbb60 100644 --- a/airflow-core/tests/unit/jobs/test_triggerer_job.py +++ b/airflow-core/tests/unit/jobs/test_triggerer_job.py @@ -66,7 +66,9 @@ clear_db_connections, clear_db_dag_bundles, clear_db_dags, + clear_db_jobs, clear_db_runs, + clear_db_triggers, clear_db_variables, clear_db_xcom, ) @@ -86,6 +88,8 @@ def clean_database(): clear_db_dag_bundles() clear_db_xcom() clear_db_variables() + clear_db_triggers() + clear_db_jobs() yield # Test runs here clear_db_connections() clear_db_runs() @@ -93,6 +97,8 @@ def clean_database(): clear_db_dag_bundles() clear_db_xcom() clear_db_variables() + clear_db_triggers() + clear_db_jobs() def create_trigger_in_db(session, trigger, operator=None): @@ -270,7 +276,10 @@ def send_msg_spy(self, msg, *args, **kwargs): @pytest.mark.parametrize( "trigger, watcher_count, trigger_count", - [(TimeDeltaTrigger(datetime.timedelta(days=7)), 0, 1), (FileDeleteTrigger("/tmp/foo.txt"), 1, 0)], + [ + (TimeDeltaTrigger(datetime.timedelta(days=7)), 0, 1), + (FileDeleteTrigger("/tmp/foo.txt", poke_interval=1), 1, 0), + ], ) @patch("time.monotonic", side_effect=itertools.count(start=1, step=60)) def test_trigger_log(mock_monotonic, trigger, watcher_count, trigger_count, session, capsys): @@ -279,10 +288,10 @@ def test_trigger_log(mock_monotonic, trigger, watcher_count, trigger_count, sess """ create_trigger_in_db(session, trigger) - trigger_runner_supervisor = TriggerRunnerSupervisor.start(job=Job(id=12345), capacity=10) + trigger_runner_supervisor = TriggerRunnerSupervisor.start(job=Job(id=123456), capacity=10) trigger_runner_supervisor.load_triggers() - for _ in range(10): + for _ in range(30): trigger_runner_supervisor._service_subprocess(0.1) stdout = capsys.readouterr().out