Я пытаюсь создать имитируемый AsyncOpenSearch и все его методы, чтобы в моем модульном тесте, когда я использую AsyncOS.ping(), он гарантировал мне, что AsyncOpenSearch.native_ping был вызван или ожидается
Как мне исправить весь AsyncOpenSearch и Native_ping?
assert_await или Assert_Call завершается с ошибкой, но печать из Native_ping печатается.
import pytest
from unittest.mock import AsyncMock, patch
import asyncio
class OSConfig:
pass
# Different module: ops.my_module.AsyncOpenSearch
# class AsyncOpenSearch:
# async def native_ping(self):
# print("I have been called!")
class AsyncOS:
def __init__(self, config: OSConfig):
self.client: AsyncOpenSearch = AsyncOpenSearch()
async def ping(self):
await self.client.native_ping()
@pytest.mark.asyncio
async def test_ping_called():
with patch("ops.my_module.AsyncOpenSearch", new_callable=AsyncMock) as os_mock:
async_os = AsyncOS(config=OSConfig())
await async_os.ping()
#os_mock.native_ping.assert_called() # doesn't work
os_mock.native_ping.assert_awaited() # doesn't work
Подробнее здесь: https://stackoverflow.com/questions/790 ... ert-called
