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

Remove asynctest dependency

Browse source
This commit is contained in:
Alex Root Junior 2022-10-25 23:28:05 +03:00
parent 8370d57a24
commit ddf7b5f412
No known key found for this signature in database
GPG key ID: 074C1D455EBEA4AC
10 changed files with 34 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/test_utils/test_chat_action.py
View file

@ -1,6 +1,7 @@
import asyncio
import time
from datetime import datetime
from unittest.mock import AsyncMock, patch
import pytest
@ -10,13 +11,7 @@ from aiogram.types import Chat, Message, User
from aiogram.utils.chat_action import ChatActionMiddleware, ChatActionSender
from tests.mocked_bot import MockedBot
try:
from asynctest import CoroutineMock, patch
except ImportError:
from unittest.mock import AsyncMock as CoroutineMock # type: ignore
from unittest.mock import patch
pytestmarm = pytest.mark.asyncio
pytestmark = pytest.mark.asyncio
class TestChatActionSender:
@ -55,7 +50,7 @@ class TestChatActionSender:
async def test_worker(self, bot: Bot):
with patch(
"aiogram.client.bot.Bot.send_chat_action",
new_callable=CoroutineMock,
new_callable=AsyncMock,
) as mocked_send_chat_action:
async with ChatActionSender.typing(
bot=bot, chat_id=42, interval=0.01, initial_sleep=0
@ -101,10 +96,10 @@ class TestChatActionMiddleware:
middleware = ChatActionMiddleware()
with patch(
"aiogram.utils.chat_action.ChatActionSender._run",
new_callable=CoroutineMock,
new_callable=AsyncMock,
) as mocked_run, patch(
"aiogram.utils.chat_action.ChatActionSender._stop",
new_callable=CoroutineMock,
new_callable=AsyncMock,
) as mocked_stop:
data = {"handler": HandlerObject(callback=handler1), "bot": bot}
message = Message(