Fixed subtypes and union types generation, new enums added (#1213)

* Fixed subtypes and union types generation, new enums added

* Added changes description
This commit is contained in:
Alex Root Junior 2023-07-11 23:39:54 +03:00 committed by GitHub
parent a7b92bb050
commit 31c11c31e0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
115 changed files with 680 additions and 359 deletions

View file

@ -1,5 +1,5 @@
from aiogram.methods import AnswerWebAppQuery, Request
from aiogram.types import InlineQueryResult, SentWebAppMessage
from aiogram.methods import AnswerWebAppQuery
from aiogram.types import InlineQueryResultPhoto, SentWebAppMessage
from tests.mocked_bot import MockedBot
@ -9,7 +9,11 @@ class TestAnswerWebAppQuery:
response: SentWebAppMessage = await bot.answer_web_app_query(
web_app_query_id="test",
result=InlineQueryResult(),
result=InlineQueryResultPhoto(
id="test",
photo_url="test",
thumbnail_url="test",
),
)
request = bot.get_request()
assert response == prepare_result.result

View file

@ -1,4 +1,5 @@
from datetime import datetime
from typing import Optional
import pytest
@ -24,6 +25,11 @@ from aiogram.types import (
)
class ChatMemberCustom(ChatMember):
status: str
is_member: Optional[bool] = None
class TestMemberStatusMarker:
def test_str(self):
marker = _MemberStatusMarker("test")
@ -113,11 +119,11 @@ class TestMemberStatusMarker:
@pytest.mark.parametrize(
"name,is_member,member,result",
[
["test", None, ChatMember(status="member"), False],
["test", None, ChatMember(status="test"), True],
["test", True, ChatMember(status="test"), False],
["test", True, ChatMember(status="test", is_member=True), True],
["test", True, ChatMember(status="test", is_member=False), False],
["test", None, ChatMemberCustom(status="member"), False],
["test", None, ChatMemberCustom(status="test"), True],
["test", True, ChatMemberCustom(status="test"), False],
["test", True, ChatMemberCustom(status="test", is_member=True), True],
["test", True, ChatMemberCustom(status="test", is_member=False), False],
],
)
def test_check(self, name, is_member, member, result):
@ -244,29 +250,34 @@ class TestMemberStatusTransition:
@pytest.mark.parametrize(
"transition,old,new,result",
[
[JOIN_TRANSITION, ChatMember(status="left"), ChatMember(status="member"), True],
[
JOIN_TRANSITION,
ChatMember(status="restricted", is_member=True),
ChatMember(status="member"),
False,
],
[
JOIN_TRANSITION,
ChatMember(status="restricted", is_member=False),
ChatMember(status="member"),
ChatMemberCustom(status="left"),
ChatMemberCustom(status="member"),
True,
],
[
JOIN_TRANSITION,
ChatMember(status="member"),
ChatMember(status="restricted", is_member=False),
ChatMemberCustom(status="restricted", is_member=True),
ChatMemberCustom(status="member"),
False,
],
[
JOIN_TRANSITION,
ChatMemberCustom(status="restricted", is_member=False),
ChatMemberCustom(status="member"),
True,
],
[
JOIN_TRANSITION,
ChatMemberCustom(status="member"),
ChatMemberCustom(status="restricted", is_member=False),
False,
],
[
LEAVE_TRANSITION,
ChatMember(status="member"),
ChatMember(status="restricted", is_member=False),
ChatMemberCustom(status="member"),
ChatMemberCustom(status="restricted", is_member=False),
True,
],
],