From 05abe7efe0b64aabd0dd17518e06e9728ad1e2c7 Mon Sep 17 00:00:00 2001 From: hanwg Date: Fri, 20 Feb 2026 00:50:51 +0800 Subject: [PATCH] Add callback inline keyboard tests for Telegram bot (#163328) --- tests/components/telegram_bot/conftest.py | 27 +++++++++++ .../telegram_bot/test_telegram_bot.py | 45 ++++++++++++++++++- 2 files changed, 70 insertions(+), 2 deletions(-) diff --git a/tests/components/telegram_bot/conftest.py b/tests/components/telegram_bot/conftest.py index 905a6db390e1..7ceb3599700b 100644 --- a/tests/components/telegram_bot/conftest.py +++ b/tests/components/telegram_bot/conftest.py @@ -250,6 +250,33 @@ def update_callback_query(): } +@pytest.fixture +def update_callback_inline_keyboard(): + """Fixture for mocking an incoming update of type callback_query from inline keyboard button.""" + return { + "update_id": 1, + "callback_query": { + "id": "4382bfdwdsb323b2d9", + "from": { + "id": 12345678, + "type": "private", + "is_bot": False, + "last_name": "Test Lastname", + "first_name": "Test Firstname", + "username": "Testusername", + }, + "message": { + "message_id": 101, + "chat": {"id": 987654321, "type": "private"}, + "date": 1708181000, + "text": "command", + }, + "chat_instance": "aaa111", + "data": "/command arg1 arg2", + }, + } + + @pytest.fixture def mock_broadcast_config_entry() -> MockConfigEntry: """Return the default mocked config entry.""" diff --git a/tests/components/telegram_bot/test_telegram_bot.py b/tests/components/telegram_bot/test_telegram_bot.py index 7ade0ba3ffeb..7ac6ecd02a09 100644 --- a/tests/components/telegram_bot/test_telegram_bot.py +++ b/tests/components/telegram_bot/test_telegram_bot.py @@ -646,14 +646,14 @@ async def test_webhook_endpoint_generates_telegram_command_event( assert isinstance(events[0].context, Context) -async def test_webhook_endpoint_generates_telegram_callback_event( +async def test_webhook_callback_inline_query( hass: HomeAssistant, webhook_bot, hass_client: ClientSessionGenerator, update_callback_query, mock_generate_secret_token, ) -> None: - """POST to the configured webhook endpoint and assert fired `telegram_callback` event.""" + """Test callback query triggered by inline query.""" client = await hass_client() events = async_capture_events(hass, "telegram_callback") @@ -673,6 +673,47 @@ async def test_webhook_endpoint_generates_telegram_callback_event( assert isinstance(events[0].context, Context) +async def test_webhook_callback_inline_keyboard( + hass: HomeAssistant, + webhook_bot: None, + hass_client: ClientSessionGenerator, + update_callback_inline_keyboard, + mock_generate_secret_token, +) -> None: + """Test callback query triggered by inline keyboard button.""" + client = await hass_client() + events = async_capture_events(hass, "telegram_callback") + + response = await client.post( + f"{TELEGRAM_WEBHOOK_URL}_123456", + json=update_callback_inline_keyboard, + headers={"X-Telegram-Bot-Api-Secret-Token": mock_generate_secret_token}, + ) + assert response.status == 200 + assert (await response.read()).decode("utf-8") == "" + + # Make sure event has fired + await hass.async_block_till_done() + + assert len(events) == 1 + assert ( + events[0].data["chat_id"] + == update_callback_inline_keyboard["callback_query"]["message"]["chat"]["id"] + ) + expected_message = { + **update_callback_inline_keyboard["callback_query"]["message"], + "delete_chat_photo": False, + "group_chat_created": False, + "supergroup_chat_created": False, + "channel_chat_created": False, + } + assert events[0].data["message"] == expected_message + assert events[0].data["data"] == "/command arg1 arg2" + assert events[0].data["command"] == "/command" + assert events[0].data["args"] == ["arg1", "arg2"] + assert isinstance(events[0].context, Context) + + @pytest.mark.parametrize( ("attachment_type"), [