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
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""Tests for Telegram document-size cap.
|
|
|
|
The public Telegram Bot API caps `getFile` at 20MB. A locally-hosted
|
|
`telegram-bot-api` server raises that ceiling to 2GB. We treat the presence
|
|
of `extra.base_url` as the explicit opt-in to the higher cap.
|
|
"""
|
|
|
|
import sys
|
|
from unittest.mock import MagicMock
|
|
|
|
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 test_max_doc_bytes_defaults_to_20mb_without_base_url():
|
|
adapter = TelegramAdapter(PlatformConfig(enabled=True, token="***", extra={}))
|
|
assert adapter._max_doc_bytes == 20 * 1024 * 1024
|
|
|
|
|
|
def test_max_doc_bytes_raised_to_2gb_when_base_url_set():
|
|
adapter = TelegramAdapter(
|
|
PlatformConfig(
|
|
enabled=True,
|
|
token="***",
|
|
extra={"base_url": "http://localhost:8081/bot"},
|
|
)
|
|
)
|
|
assert adapter._max_doc_bytes == 2 * 1024 * 1024 * 1024
|
|
|
|
|
|
def test_max_doc_bytes_empty_base_url_keeps_default():
|
|
"""An empty/falsy `base_url` should not flip the cap — only a real URL does."""
|
|
adapter = TelegramAdapter(
|
|
PlatformConfig(enabled=True, token="***", extra={"base_url": ""}),
|
|
)
|
|
assert adapter._max_doc_bytes == 20 * 1024 * 1024
|