@@ -129,7 +129,7 @@ async def test_publish_service_started_metrics(
129129 task = random .choice (tasks ), # noqa: S311
130130 )
131131 await _assert_message_received (
132- mocked_message_parser , 1 , InstrumentationRabbitMessage .parse_raw
132+ mocked_message_parser , 1 , InstrumentationRabbitMessage .model_validate_json
133133 )
134134
135135
@@ -154,7 +154,7 @@ async def test_publish_service_stopped_metrics(
154154 task_final_state = random .choice (list (RunningState )), # noqa: S311
155155 )
156156 await _assert_message_received (
157- mocked_message_parser , 1 , InstrumentationRabbitMessage .parse_raw
157+ mocked_message_parser , 1 , InstrumentationRabbitMessage .model_validate_json
158158 )
159159
160160
@@ -207,7 +207,7 @@ async def test_publish_service_resource_tracking_started(
207207 )
208208 after_publication_time = datetime .datetime .now (datetime .timezone .utc )
209209 received_messages = await _assert_message_received (
210- mocked_message_parser , 1 , RabbitResourceTrackingStartedMessage .parse_raw
210+ mocked_message_parser , 1 , RabbitResourceTrackingStartedMessage .model_validate_json
211211 )
212212 assert isinstance (received_messages [0 ], RabbitResourceTrackingStartedMessage )
213213 assert received_messages [0 ].service_run_id == random_service_run_id
@@ -241,7 +241,7 @@ async def test_publish_service_resource_tracking_stopped(
241241 )
242242 after_publication_time = datetime .datetime .now (datetime .timezone .utc )
243243 received_messages = await _assert_message_received (
244- mocked_message_parser , 1 , RabbitResourceTrackingStoppedMessage .parse_raw
244+ mocked_message_parser , 1 , RabbitResourceTrackingStoppedMessage .model_validate_json
245245 )
246246 assert isinstance (received_messages [0 ], RabbitResourceTrackingStoppedMessage )
247247 assert received_messages [0 ].service_run_id == random_service_run_id
@@ -272,7 +272,7 @@ async def test_publish_service_resource_tracking_heartbeat(
272272 )
273273 after_publication_time = datetime .datetime .now (datetime .timezone .utc )
274274 received_messages = await _assert_message_received (
275- mocked_message_parser , 1 , RabbitResourceTrackingHeartbeatMessage .parse_raw
275+ mocked_message_parser , 1 , RabbitResourceTrackingHeartbeatMessage .model_validate_json
276276 )
277277 assert isinstance (received_messages [0 ], RabbitResourceTrackingHeartbeatMessage )
278278 assert received_messages [0 ].service_run_id == random_service_run_id
0 commit comments