Skip to content

Commit 7b73bcf

Browse files
fix deprecated
1 parent c091c4b commit 7b73bcf

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

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

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)