Как я могу имитировать несколько менеджеров контекста и вызовы функций с помощью pytest в асинхронных контекстах? ⇐ Python

[Anonymous]

Я пишу тесты для приложения fastApi на Python, однако проблема не связана напрямую с самим fastAPI. У меня есть блок кода, который, похоже, не имитируется должным образом, и код-нарушитель:

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

async with await self._db.get_pool().getconn() as conn:
async with await conn.cursor() as cursor:
await cursor.execute(...)
result = await cursor.fetchone()
...
return Object(id=result[0]...)

ключ self._db имеет тип DB, определенный в другом месте. вызов выполнения — это просто запрос к базе данных, а тип объекта — например. тип возвращаемого значения функции заполняется списком значений, возвращенных при выполнении.
У меня есть такая макетная настройка:

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

mock_db = AsyncMock(spec=DB)
mock_pool = AsyncMock(spec=AsyncConnectionPool)
mock_conn = AsyncMock(spec=AsyncConnection)
mock_cursor = AsyncMock(spec=AsyncCursor)

mock_db.__aenter__.return_value = mock_db
mock_db.get_pool.return_value = mock_pool
mock_pool.getconn.return_value = mock_conn
mock_conn.__aenter__.return_value = mock_conn
mock_conn.cursor.return_value = mock_cursor
mock_cursor.__aenter__.return_value = mock_cursor
mock_cursor.fetchone.return_value = (uuid(), ....)
result = await test_func(...)

при запуске теста мы не получаем утверждений, поскольку все макеты испорчены.
Я знаю, что где-то ошибся, но где находится то, чего я сейчас сказать не могу.
Я пытался найти документацию специально для тестирования этого случая, но не смог найти четкого объяснения того, как для издевательства над вложенными менеджерами контекста и объяснение того, где мне нужно издеваться с помощью функции __aenter__.
Я также пытался получить каждое издеваемое значение, имеющее __aenter__ для вызова свойства return_value перед __aenter__

Подробнее здесь: https://stackoverflow.com/questions/791 ... t-in-async