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

#901 Fixed false-positive coercing of Union types in API methods

Browse source
This commit is contained in:
Alex Root Junior 2022-05-14 17:15:31 +03:00
parent 6ad242399b
commit dab4e26f1c
No known key found for this signature in database
GPG key ID: 074C1D455EBEA4AC
3 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
CHANGES/901.bugfix.rst Normal file
View file

@ -0,0 +1 @@
Fixed false-positive coercing of Union types in API methods

1
aiogram/methods/base.py
View file

@ -46,6 +46,7 @@ class TelegramMethod(abc.ABC, BaseModel, Generic[TelegramType]):
allow_population_by_field_name = True
arbitrary_types_allowed = True
orm_mode = True
smart_union = True # https://github.com/aiogram/aiogram/issues/901
@root_validator(pre=True)
def remove_unset(cls, values: Dict[str, Any]) -> Dict[str, Any]:

7
tests/test_api/test_methods/test_send_message.py
View file

@ -3,7 +3,7 @@ import datetime
import pytest
from aiogram.methods import Request, SendMessage
from aiogram.types import Chat, Message
from aiogram.types import Chat, ForceReply, Message
from tests.mocked_bot import MockedBot
pytestmark = pytest.mark.asyncio
@ -43,3 +43,8 @@ class TestSendMessage:
request: Request = bot.get_request()
assert request.method == "sendMessage"
assert response == prepare_result.result
async def test_force_reply(self):
# https://github.com/aiogram/aiogram/issues/901
method = SendMessage(text="test", chat_id=42, reply_markup=ForceReply())
assert isinstance(method.reply_markup, ForceReply)