async def mocked_connect_robust(loop=None, host=None): conn_mock = AsyncMock(spec=aio_pika.robust_connection.RobustConnection) conn_mock.loop = loop conn_mock.is_closed = closed chan_mock = CoroutineMock() exchange_mock = CoroutineMock() queue_mock = CoroutineMock() queue_mock.bind = CoroutineMock() queue_mock.consume = CoroutineMock() exchange_mock.publish = CoroutineMock() chan_mock.declare_exchange = CoroutineMock(return_value=exchange_mock) chan_mock.declare_queue = CoroutineMock(return_value=queue_mock) conn_mock.close = CoroutineMock() conn_mock.channel = CoroutineMock(return_value=chan_mock) return conn_mock