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 5

fix: invoke factory instead of direct ChatMember class

Browse source
This commit is contained in:
Egor 2020-07-03 16:30:28 +05:00
parent 8f3913b728
commit 36c70577bf
3 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_api/test_methods/test_get_chat_administrators.py
View file

@ -3,8 +3,8 @@ from typing import List
import pytest import pytest
from aiogram.api.methods import GetChatAdministrators, Request from aiogram.api.methods import GetChatAdministrators, Request
from aiogram.api.types import ChatMember, User from aiogram.api.types import ChatMember
from tests.factories.user import UserFactory from tests.factories.chat_member import ChatMemberFactory
from tests.mocked_bot import MockedBot from tests.mocked_bot import MockedBot
@ -12,7 +12,7 @@ class TestGetChatAdministrators:
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_method(self, bot: MockedBot): async def test_method(self, bot: MockedBot):
prepare_result = bot.add_result_for( prepare_result = bot.add_result_for(
GetChatAdministrators, ok=True, result=[ChatMember(status="creator")], GetChatAdministrators, ok=True, result=[ChatMemberFactory(status="creator")],
) )
response: List[ChatMember] = await GetChatAdministrators(chat_id=-42) response: List[ChatMember] = await GetChatAdministrators(chat_id=-42)
@ -23,7 +23,7 @@ class TestGetChatAdministrators:
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_bot_method(self, bot: MockedBot): async def test_bot_method(self, bot: MockedBot):
prepare_result = bot.add_result_for( prepare_result = bot.add_result_for(
GetChatAdministrators, ok=True, result=[ChatMember(status="creator")], GetChatAdministrators, ok=True, result=[ChatMemberFactory(status="creator")],
) )
response: List[ChatMember] = await bot.get_chat_administrators(chat_id=-42) response: List[ChatMember] = await bot.get_chat_administrators(chat_id=-42)
request: Request = bot.get_request() request: Request = bot.get_request()

5
tests/test_api/test_methods/test_get_chat_member.py
View file

@ -2,6 +2,7 @@ import pytest
from aiogram.api.methods import GetChatMember, Request from aiogram.api.methods import GetChatMember, Request
from aiogram.api.types import ChatMember from aiogram.api.types import ChatMember
from tests.factories.chat_member import ChatMemberFactory
from tests.factories.user import UserFactory from tests.factories.user import UserFactory
from tests.mocked_bot import MockedBot from tests.mocked_bot import MockedBot
@ -10,7 +11,7 @@ class TestGetChatMember:
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_method(self, bot: MockedBot): async def test_method(self, bot: MockedBot):
prepare_result = bot.add_result_for( prepare_result = bot.add_result_for(
GetChatMember, ok=True, result=ChatMember(status="creator"), GetChatMember, ok=True, result=ChatMemberFactory(status="creator"),
) )
response: ChatMember = await GetChatMember(chat_id=-42, user_id=42) response: ChatMember = await GetChatMember(chat_id=-42, user_id=42)
@ -21,7 +22,7 @@ class TestGetChatMember:
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_bot_method(self, bot: MockedBot): async def test_bot_method(self, bot: MockedBot):
prepare_result = bot.add_result_for( prepare_result = bot.add_result_for(
GetChatMember, ok=True, result=ChatMember(status="creator"), GetChatMember, ok=True, result=ChatMemberFactory(status="creator"),
) )
response: ChatMember = await bot.get_chat_member(chat_id=-42, user_id=42) response: ChatMember = await bot.get_chat_member(chat_id=-42, user_id=42)

7
tests/test_api/test_types/test_chat_member.py
View file

@ -1,7 +1,6 @@
import pytest import pytest
from aiogram.api.types import ChatMember from tests.factories.chat_member import ChatMemberFactory
from tests.factories.user import UserFactory
class TestChatMember: class TestChatMember:
@ -9,7 +8,7 @@ class TestChatMember:
"status,result", [["administrator", True], ["creator", True], ["member", False]] "status,result", [["administrator", True], ["creator", True], ["member", False]]
) )
def test_is_chat_admin(self, status: str, result: bool): def test_is_chat_admin(self, status: str, result: bool):
chat_member = ChatMember(status=status) chat_member = ChatMemberFactory(status=status)
assert chat_member.is_chat_admin == result assert chat_member.is_chat_admin == result
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -24,5 +23,5 @@ class TestChatMember:
], ],
) )
def test_is_chat_member(self, status: str, result: bool): def test_is_chat_member(self, status: str, result: bool):
chat_member = ChatMember(status=status) chat_member = ChatMemberFactory(status=status)
assert chat_member.is_chat_member == result assert chat_member.is_chat_member == result