Объект Unittest asyncio patch с вложенными сопрограммами ⇐ Python

[Anonymous]

Я хочу имитировать клиент асинхронной базы данных в unittest.
Вызов функции клиента выглядит следующим образом:

Код: Выделить всё

test.client.json().set(argumentsforquery)

Тест выглядит следующим образом:

Код: Выделить всё

@pytest.mark.asyncio
async def test_client(self):
self.test.client = AsyncMock()

with patch.object(self.test.client.json, 'set', return_value='FooBar') as mock:
print(await mock())
# FooBar
print(await self.test.client.json.set())
# FooBar
print(await self.test.client.json().set())
# AttributeError: 'coroutine' object has no attribute 'set'

Моя проблема в том, что имитируемый объект self.test.client.json не возвращает набор атрибутов, как в третьем показанном операторе печати. Я попытался исправить self.test.client с помощью json и добавить атрибут set как return_value как тип AsyncMock() с тем же результатом.
Любые идеи как получить FooBar, вызвав await self.test.client.json().set()?

Подробнее здесь: https://stackoverflow.com/questions/792 ... sted-coros