python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Try to fix tests

Browse source
This commit is contained in:
Gabben 2020-03-16 23:29:51 +05:00
parent 1037b6d927
commit 613821af44
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_api/test_client/test_base_bot.py
View file

@ -12,7 +12,7 @@ from aiogram.api.methods import GetMe
try:
from asynctest import CoroutineMock, patch
except ImportError:
from unittest.mock import AsyncMock as CoroutineMock, MagicMock, patch # type: ignore
from unittest.mock import AsyncMock as CoroutineMock, patch # type: ignore
class TestBaseBot:
@ -76,11 +76,11 @@ class TestBaseBot:
)
# https://github.com/Tinche/aiofiles#writing-tests-for-aiofiles
aiofiles.threadpool.wrap.register(MagicMock)(
aiofiles.threadpool.wrap.register(CoroutineMock)(
lambda *args, **kwargs: threadpool.AsyncBufferedIOBase(*args, **kwargs)
)
mock_file = MagicMock()
mock_file = CoroutineMock()
base_bot = BaseBot("42:TEST")
with patch("aiofiles.threadpool.sync_open", return_value=mock_file):