File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
services/director-v2/tests/unit/with_dbs/comp_scheduler Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 1616from typing import Any , cast
1717from unittest import mock
1818
19+ import arrow
1920import pytest
2021from _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 ()
409409async 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 ,
You can’t perform that action at this time.
0 commit comments