mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-12 10:11:52 +00:00
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:
parent
a7b92bb050
commit
31c11c31e0
115 changed files with 680 additions and 359 deletions
|
|
@ -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,
|
||||
],
|
||||
],
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue