Skip to content

Commit c810815

Browse files
committed
fixed events
1 parent ca1bc5e commit c810815

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

services/director-v2/tests/unit/with_dbs/comp_scheduler/test_scheduler_dask.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from typing import Any, cast
1717
from unittest import mock
1818

19+
import arrow
1920
import pytest
2021
from _helpers import (
2122
PublishedProject,
@@ -401,11 +402,10 @@ async def _trigger_progress_event(
401402
),
402403
)
403404
await cast(DaskScheduler, scheduler)._task_progress_change_handler( # noqa: SLF001
404-
event.model_dump_json()
405+
(arrow.utcnow().timestamp(), event.model_dump_json())
405406
)
406407

407408

408-
@pytest.mark.acceptance_test()
409409
async def test_proper_pipeline_is_scheduled( # noqa: PLR0915
410410
with_disabled_auto_scheduling: mock.Mock,
411411
with_disabled_scheduler_publisher: mock.Mock,
@@ -1191,7 +1191,9 @@ async def test_task_progress_triggers(
11911191
)
11921192
await cast( # noqa: SLF001
11931193
DaskScheduler, scheduler_api
1194-
)._task_progress_change_handler(progress_event.model_dump_json())
1194+
)._task_progress_change_handler(
1195+
(arrow.utcnow().timestamp(), progress_event.model_dump_json())
1196+
)
11951197
# NOTE: not sure whether it should switch to STARTED.. it would make sense
11961198
await assert_comp_tasks(
11971199
sqlalchemy_async_engine,

0 commit comments

Comments
 (0)