Fixed aiofiles tests

This commit is contained in:
Alex Root Junior 2022-10-25 23:35:31 +03:00
parent ddf7b5f412
commit 26daadd589
No known key found for this signature in database
GPG key ID: 074C1D455EBEA4AC

View file

@ -1,7 +1,7 @@
import io
import os
from tempfile import mkstemp
from unittest.mock import AsyncMock, patch
from unittest.mock import AsyncMock, MagicMock, patch
import aiofiles
import pytest
@ -73,11 +73,11 @@ class TestBot:
)
# https://github.com/Tinche/aiofiles#writing-tests-for-aiofiles
aiofiles.threadpool.wrap.register(AsyncMock)(
aiofiles.threadpool.wrap.register(MagicMock)(
lambda *args, **kwargs: aiofiles.threadpool.binary.AsyncBufferedIOBase(*args, **kwargs)
)
mock_file = AsyncMock()
mock_file = MagicMock()
bot = Bot("42:TEST")
with patch("aiofiles.threadpool.sync_open", return_value=mock_file):