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 #660 prepare parse mode for input_message_content (#661)

Browse source 
* fix #660 prepare parse mode for input_message_content of InlineQueryResult

* remove redundant get

* black lint

* add towncrier patch note
This commit is contained in:
darksidecat 2021-08-17 00:08:36 +03:00 • committed by GitHub
parent 7cb0ac1ee2
commit 71eb5fc44e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 29 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
CHANGES/660.bugfix Normal file
View file

@ -0,0 +1 @@
Prepare parse mode for InputMessageContent in AnswerInlineQuery method

9
aiogram/methods/answer_inline_query.py
View file

@ -37,5 +37,12 @@ class AnswerInlineQuery(TelegramMethod[bool]):
def build_request(self, bot: Bot) -> Request: def build_request(self, bot: Bot) -> Request:
data: Dict[str, Any] = self.dict() data: Dict[str, Any] = self.dict()
prepare_parse_mode(bot, data["results"])
input_message_contents = []
for result in data["results"]:
input_message_content = result.get("input_message_content", None)
if input_message_content is not None:
input_message_contents.append(input_message_content)
prepare_parse_mode(bot, data["results"] + input_message_contents)
return Request(method="answerInlineQuery", data=data) return Request(method="answerInlineQuery", data=data)

21
tests/test_api/test_methods/test_answer_inline_query.py
View file

@ -2,7 +2,7 @@ import pytest
from aiogram import Bot from aiogram import Bot
from aiogram.methods import AnswerInlineQuery, Request from aiogram.methods import AnswerInlineQuery, Request
from aiogram.types import InlineQueryResult, InlineQueryResultPhoto from aiogram.types import InlineQueryResult, InlineQueryResultPhoto, InputTextMessageContent
from tests.mocked_bot import MockedBot from tests.mocked_bot import MockedBot
pytestmark = pytest.mark.asyncio pytestmark = pytest.mark.asyncio
@ -40,3 +40,22 @@ class TestAnswerInlineQuery:
new_bot = Bot(token="42:TEST", parse_mode="HTML") new_bot = Bot(token="42:TEST", parse_mode="HTML")
request = query.build_request(new_bot) request = query.build_request(new_bot)
assert request.data["results"][0]["parse_mode"] == "HTML" assert request.data["results"][0]["parse_mode"] == "HTML"
def test_parse_mode_input_message_content(self, bot: MockedBot):
query = AnswerInlineQuery(
inline_query_id="query id",
results=[
InlineQueryResultPhoto(
id="result id",
photo_url="photo",
thumb_url="thumb",
input_message_content=InputTextMessageContent(message_text="test"),
)
],
)
request = query.build_request(bot)
assert request.data["results"][0]["input_message_content"]["parse_mode"] is None
new_bot = Bot(token="42:TEST", parse_mode="HTML")
request = query.build_request(new_bot)
assert request.data["results"][0]["input_message_content"]["parse_mode"] == "HTML"