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

Fix tests so they are able to run

Browse source
This commit is contained in:
evgfilim1 2021-07-24 17:02:19 +05:00
parent 427c10ea3f
commit 769ae8124c
No known key found for this signature in database
GPG key ID: 16AEE4D0BB188AEC
5 changed files with 20 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -3,7 +3,7 @@ from typing import List
import pytest
from aiogram.methods import GetChatAdministrators, Request
from aiogram.types import ChatMember, User
from aiogram.types import ChatMember, User, ChatMemberOwner
from tests.mocked_bot import MockedBot
@ -14,7 +14,7 @@ class TestGetChatAdministrators:
GetChatAdministrators,
ok=True,
result=[
ChatMember(user=User(id=42, is_bot=False, first_name="User"), status="creator")
ChatMemberOwner(user=User(id=42, is_bot=False, first_name="User"), is_anonymous=False)
],
)
@ -29,7 +29,7 @@ class TestGetChatAdministrators:
GetChatAdministrators,
ok=True,
result=[
ChatMember(user=User(id=42, is_bot=False, first_name="User"), status="creator")
ChatMemberOwner(user=User(id=42, is_bot=False, first_name="User"), is_anonymous=False)
],
)
response: List[ChatMember] = await bot.get_chat_administrators(chat_id=-42)

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

@ -1,7 +1,7 @@
import pytest
from aiogram.methods import GetChatMember, Request
from aiogram.types import ChatMember, User
from aiogram.types import ChatMember, User, ChatMemberOwner
from tests.mocked_bot import MockedBot
@ -11,7 +11,7 @@ class TestGetChatMember:
prepare_result = bot.add_result_for(
GetChatMember,
ok=True,
result=ChatMember(user=User(id=42, is_bot=False, first_name="User"), status="creator"),
result=ChatMemberOwner(user=User(id=42, is_bot=False, first_name="User"), is_anonymous=False),
)
response: ChatMember = await GetChatMember(chat_id=-42, user_id=42)
@ -24,7 +24,7 @@ class TestGetChatMember:
prepare_result = bot.add_result_for(
GetChatMember,
ok=True,
result=ChatMember(user=User(id=42, is_bot=False, first_name="User"), status="creator"),
result=ChatMemberOwner(user=User(id=42, is_bot=False, first_name="User"), is_anonymous=False),
)
response: ChatMember = await bot.get_chat_member(chat_id=-42, user_id=42)

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

@ -1,29 +0,0 @@
import pytest
from aiogram.types import ChatMember, User
user = User(id=42, is_bot=False, first_name="User", last_name=None)
class TestChatMember:
@pytest.mark.parametrize(
"status,result", [["administrator", True], ["creator", True], ["member", False]]
)
def test_is_chat_admin(self, status: str, result: bool):
chat_member = ChatMember(user=user, status=status)
assert chat_member.is_chat_admin == result
@pytest.mark.parametrize(
"status,result",
[
["administrator", True],
["creator", True],
["member", True],
["restricted", True],
["kicked", False],
["left", False],
],
)
def test_is_chat_member(self, status: str, result: bool):
chat_member = ChatMember(user=user, status=status)
assert chat_member.is_chat_member == result