Skip to content

Commit

Permalink
fix test_long_stalled_task_is_killed_by_listener_overtime_if_ol_timeo…
Browse files Browse the repository at this point in the history
…ut_long_enough (apache#44537)

Signed-off-by: Maciej Obuchowski <[email protected]>
  • Loading branch information
mobuchowski authored Dec 1, 2024
1 parent 25e173a commit 29ac934
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions providers/tests/openlineage/plugins/test_execution.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,6 @@ def test_mid_stalled_task_is_killed_by_ol_timeout(self):
assert has_value_in_events(events, ["inputs", "name"], "on-start")
assert not has_value_in_events(events, ["inputs", "name"], "on-complete")

@pytest.mark.quarantined
@conf_vars(
{
("openlineage", "transport"): f'{{"type": "file", "log_file_path": "{listener_path}"}}',
Expand All @@ -176,7 +175,9 @@ def test_mid_stalled_task_is_killed_by_ol_timeout(self):
}
)
@pytest.mark.db_test
def test_long_stalled_task_is_killed_by_listener_overtime_if_ol_timeout_long_enough(self):
def test_success_overtime_kills_tasks(self):
# This test checks whether LocalTaskJobRunner kills OL listener which take
# longer time than permitted by core.task_success_overtime setting
dirpath = Path(tmp_dir)
if dirpath.exists():
shutil.rmtree(dirpath)
Expand All @@ -203,7 +204,7 @@ def test_long_stalled_task_is_killed_by_listener_overtime_if_ol_timeout_long_eno
task=task,
run_id="test_long_stalled_task_is_killed_by_listener_overtime_if_ol_timeout_long_enough",
)
job = Job(id="1", dag_id=ti.dag_id)
job = Job(dag_id=ti.dag_id)
job_runner = LocalTaskJobRunner(job=job, task_instance=ti, ignore_ti_state=True)
job_runner._execute()

Expand Down

0 comments on commit 29ac934

Please sign in to comment.