Skip to content

Commit 90af8e7

Browse files
committed
ongoing
1 parent f278199 commit 90af8e7

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

services/dask-sidecar/tests/unit/test_tasks.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,11 +114,16 @@ def dask_subsystem_mock(
114114
return_value=False,
115115
)
116116
# mock dask rabbitmq plugin
117+
mock_dask_rabbitmq_plugin = mocker.patch(
118+
"simcore_service_dask_sidecar.rabbitmq_plugin.RabbitMQPlugin", autospec=True
119+
)
117120
mock_rabbitmq_client = create_rabbitmq_client("pytest_dask_sidecar_logs_publisher")
121+
mock_dask_rabbitmq_plugin.get_client.return_value = mock_rabbitmq_client
122+
118123
mocker.patch(
119124
"simcore_service_dask_sidecar.utils.dask.get_rabbitmq_client",
120125
autospec=True,
121-
return_value=mock_rabbitmq_client,
126+
return_value=mock_dask_rabbitmq_plugin,
122127
)
123128

124129
return {

0 commit comments

Comments
 (0)