wehub-resource-sync
b4fbd6fe9f
Deploy Site / deploy-vercel (push) Has been skipped
Deploy Site / deploy-docs (push) Has been skipped
Build Skills Index / build-index (push) Has been skipped
CI / Deny unrelated histories (push) Has been skipped
CI / Detect affected areas (push) Successful in 27m35s
CI / OSV scan (push) Failing after 4s
CI / Build&Test Docker image (push) Successful in 9s
CI / Supply-chain scan (push) Has been skipped
CI / Lint Docker scripts (push) Failing after 5m13s
CI / Check contributors (push) Failing after 12m8s
CI / Docs Site (push) Failing after 12m8s
CI / TypeScript (push) Failing after 12m8s
CI / Python lints (push) Failing after 12m9s
CI / Python tests (push) Failing after 12m9s
CI / Check uv.lock (push) Failing after 23m22s
CI / CI timing report (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
CI / All required checks pass (push) Has been cancelled
121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
"""Tests for the Telegram bot status indicator.
|
|
|
|
Telegram bots have no real online/offline presence dot (that's a user-account
|
|
feature). The closest Bot API surface is the bot's *short description* — the
|
|
line shown under the bot's name in its profile. When `extra.status_indicator`
|
|
is enabled, the adapter sets it to "Online" on connect and "Offline" on clean
|
|
disconnect so users can tell whether the gateway is up.
|
|
"""
|
|
|
|
import sys
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from gateway.config import PlatformConfig
|
|
|
|
|
|
def _ensure_telegram_mock():
|
|
if "telegram" in sys.modules and hasattr(sys.modules["telegram"], "__file__"):
|
|
return
|
|
|
|
telegram_mod = MagicMock()
|
|
telegram_mod.ext.ContextTypes.DEFAULT_TYPE = type(None)
|
|
telegram_mod.constants.ParseMode.MARKDOWN_V2 = "MarkdownV2"
|
|
telegram_mod.constants.ChatType.GROUP = "group"
|
|
telegram_mod.constants.ChatType.SUPERGROUP = "supergroup"
|
|
telegram_mod.constants.ChatType.CHANNEL = "channel"
|
|
telegram_mod.constants.ChatType.PRIVATE = "private"
|
|
|
|
for name in ("telegram", "telegram.ext", "telegram.constants", "telegram.request"):
|
|
sys.modules.setdefault(name, telegram_mod)
|
|
|
|
|
|
_ensure_telegram_mock()
|
|
|
|
from plugins.platforms.telegram.adapter import TelegramAdapter # noqa: E402
|
|
|
|
|
|
def _make_adapter(extra):
|
|
adapter = TelegramAdapter(PlatformConfig(enabled=True, token="***", extra=extra))
|
|
adapter._bot = MagicMock()
|
|
adapter._bot.set_my_short_description = AsyncMock()
|
|
return adapter
|
|
|
|
|
|
def test_disabled_by_default():
|
|
adapter = _make_adapter(extra={})
|
|
assert adapter._status_indicator_enabled is False
|
|
|
|
|
|
def test_enabled_via_extra():
|
|
adapter = _make_adapter(extra={"status_indicator": True})
|
|
assert adapter._status_indicator_enabled is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_disabled_is_noop():
|
|
adapter = _make_adapter(extra={"status_indicator": False})
|
|
await adapter._set_status_indicator(online=True)
|
|
adapter._bot.set_my_short_description.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_online_sets_default_text():
|
|
adapter = _make_adapter(extra={"status_indicator": True})
|
|
await adapter._set_status_indicator(online=True)
|
|
adapter._bot.set_my_short_description.assert_awaited_once_with(
|
|
short_description="Online"
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_offline_sets_default_text():
|
|
adapter = _make_adapter(extra={"status_indicator": True})
|
|
await adapter._set_status_indicator(online=False)
|
|
adapter._bot.set_my_short_description.assert_awaited_once_with(
|
|
short_description="Offline"
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_custom_status_strings():
|
|
adapter = _make_adapter(
|
|
extra={
|
|
"status_indicator": True,
|
|
"status_online": "🟢 Gateway up",
|
|
"status_offline": "🔴 Gateway down",
|
|
}
|
|
)
|
|
await adapter._set_status_indicator(online=True)
|
|
adapter._bot.set_my_short_description.assert_awaited_once_with(
|
|
short_description="🟢 Gateway up"
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_text_truncated_to_120_chars():
|
|
adapter = _make_adapter(
|
|
extra={"status_indicator": True, "status_online": "x" * 200}
|
|
)
|
|
await adapter._set_status_indicator(online=True)
|
|
_, kwargs = adapter._bot.set_my_short_description.call_args
|
|
assert len(kwargs["short_description"]) == 120
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_noop_when_bot_is_none():
|
|
adapter = _make_adapter(extra={"status_indicator": True})
|
|
adapter._bot = None
|
|
# Must not raise even though there's no bot to call.
|
|
await adapter._set_status_indicator(online=True)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_failure_is_swallowed():
|
|
adapter = _make_adapter(extra={"status_indicator": True})
|
|
adapter._bot.set_my_short_description.side_effect = RuntimeError("flood wait")
|
|
# Best-effort: a Bot API failure must never propagate out of the helper,
|
|
# so it can't block connect/disconnect.
|
|
await adapter._set_status_indicator(online=True)
|