Skip to content

Commit f3e4857

Browse files
fix typo
1 parent 469c228 commit f3e4857

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

services/director-v2/tests/unit/with_dbs/test_modules_comp_scheduler_dask_scheduler.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -737,7 +737,9 @@ async def _return_1st_task_running(job_ids: list[str]) -> list[DaskClientTaskSta
737737
mocked_dask_client.get_tasks_status.reset_mock()
738738
mocked_dask_client.get_task_result.assert_not_called()
739739
messages = await _assert_message_received(
740-
instrumentation_rabbit_client_parser, 1, InstrumentationRabbitMessage.model_validate_json
740+
instrumentation_rabbit_client_parser,
741+
1,
742+
InstrumentationRabbitMessage.model_validate_json,
741743
)
742744
assert messages[0].metrics == "service_started"
743745
assert messages[0].service_uuid == exp_started_task.node_id
@@ -780,7 +782,9 @@ async def _return_random_task_result(job_id) -> TaskOutputData:
780782
expected_progress=1,
781783
)
782784
messages = await _assert_message_received(
783-
instrumentation_rabbit_client_parser, 1, InstrumentationRabbitMessage.model_validate_jsonidate_jsonidate_json
785+
instrumentation_rabbit_client_parser,
786+
1,
787+
InstrumentationRabbitMessage.model_validate_json,
784788
)
785789
assert messages[0].metrics == "service_stopped"
786790
assert messages[0].service_uuid == exp_started_task.node_id
@@ -882,7 +886,9 @@ async def _return_2nd_task_running(job_ids: list[str]) -> list[DaskClientTaskSta
882886
mocked_dask_client.get_tasks_status.reset_mock()
883887
mocked_dask_client.get_task_result.assert_not_called()
884888
messages = await _assert_message_received(
885-
instrumentation_rabbit_client_parser, 1, InstrumentationRabbitMessage.model_validate_json
889+
instrumentation_rabbit_client_parser,
890+
1,
891+
InstrumentationRabbitMessage.model_validate_json,
886892
)
887893
assert messages[0].metrics == "service_started"
888894
assert messages[0].service_uuid == exp_started_task.node_id
@@ -926,7 +932,9 @@ async def _return_2nd_task_failed(job_ids: list[str]) -> list[DaskClientTaskStat
926932
mocked_parse_output_data_fct.assert_not_called()
927933
expected_pending_tasks.remove(exp_started_task)
928934
messages = await _assert_message_received(
929-
instrumentation_rabbit_client_parser, 1, InstrumentationRabbitMessage.model_validate_json
935+
instrumentation_rabbit_client_parser,
936+
1,
937+
InstrumentationRabbitMessage.model_validate_json,
930938
)
931939
assert messages[0].metrics == "service_stopped"
932940
assert messages[0].service_uuid == exp_started_task.node_id
@@ -970,7 +978,9 @@ async def _return_3rd_task_success(job_ids: list[str]) -> list[DaskClientTaskSta
970978
)
971979
mocked_dask_client.get_task_result.assert_called_once_with(exp_started_task.job_id)
972980
messages = await _assert_message_received(
973-
instrumentation_rabbit_client_parser, 2, InstrumentationRabbitMessage.model_validate_json
981+
instrumentation_rabbit_client_parser,
982+
2,
983+
InstrumentationRabbitMessage.model_validate_json,
974984
)
975985
# NOTE: the service was fast and went directly to success
976986
assert messages[0].metrics == "service_started"

0 commit comments

Comments
 (0)