@@ -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