diff --git a/tests/rabbitmq/test_user_management_service_client.py b/tests/rabbitmq/test_user_management_service_client.py index b0b67ce..8a4e7fd 100644 --- a/tests/rabbitmq/test_user_management_service_client.py +++ b/tests/rabbitmq/test_user_management_service_client.py @@ -40,8 +40,9 @@ class TestUserManagementServiceClient(unittest.IsolatedAsyncioTestCase): mock_connection = MagicMock() mock_connect.return_value = mock_connection - # Mock channel - use AsyncMock for async methods + # Mock channel - create an AsyncMock that returns the channel when awaited mock_channel = AsyncMock() + mock_connection.channel = AsyncMock() mock_connection.channel.return_value = mock_channel mock_channel.set_qos = AsyncMock()