banteg
1759 lines
50 KiB
Python
1759 lines
50 KiB
Python
from dataclasses import replace
|
|
from pathlib import Path
|
|
from typing import Any, cast
|
|
|
|
import anyio
|
|
import pytest
|
|
|
|
from takopi import commands, plugins
|
|
import takopi.telegram.bridge as bridge
|
|
import takopi.telegram.loop as telegram_loop
|
|
import takopi.telegram.commands as telegram_commands
|
|
import takopi.telegram.topics as telegram_topics
|
|
from takopi.directives import parse_directives
|
|
from takopi.telegram.bridge import (
|
|
TelegramBridgeConfig,
|
|
TelegramFilesConfig,
|
|
TelegramPresenter,
|
|
TelegramTransport,
|
|
build_bot_commands,
|
|
handle_callback_cancel,
|
|
handle_cancel,
|
|
is_cancel_command,
|
|
send_with_resume,
|
|
run_main_loop,
|
|
)
|
|
from takopi.telegram.client import BotClient
|
|
from takopi.telegram.topic_state import TopicStateStore, resolve_state_path
|
|
from takopi.context import RunContext
|
|
from takopi.config import ProjectConfig, ProjectsConfig
|
|
from takopi.runner_bridge import ExecBridgeConfig, RunningTask
|
|
from takopi.markdown import MarkdownPresenter
|
|
from takopi.model import EngineId, ResumeToken
|
|
from takopi.progress import ProgressTracker
|
|
from takopi.router import AutoRouter, RunnerEntry
|
|
from takopi.transport_runtime import TransportRuntime
|
|
from takopi.runners.mock import Return, ScriptRunner, Sleep, Wait
|
|
from takopi.telegram.types import (
|
|
TelegramCallbackQuery,
|
|
TelegramDocument,
|
|
TelegramIncomingMessage,
|
|
)
|
|
from takopi.transport import MessageRef, RenderedMessage, SendOptions
|
|
from tests.plugin_fixtures import FakeEntryPoint, install_entrypoints
|
|
|
|
CODEX_ENGINE = EngineId("codex")
|
|
|
|
|
|
def _empty_projects() -> ProjectsConfig:
|
|
return ProjectsConfig(projects={}, default_project=None)
|
|
|
|
|
|
def _make_router(runner) -> AutoRouter:
|
|
return AutoRouter(
|
|
entries=[RunnerEntry(engine=runner.engine, runner=runner)],
|
|
default_engine=runner.engine,
|
|
)
|
|
|
|
|
|
class _FakeTransport:
|
|
def __init__(self, progress_ready: anyio.Event | None = None) -> None:
|
|
self._next_id = 1
|
|
self.send_calls: list[dict] = []
|
|
self.edit_calls: list[dict] = []
|
|
self.delete_calls: list[MessageRef] = []
|
|
self.progress_ready = progress_ready
|
|
self.progress_ref: MessageRef | None = None
|
|
|
|
async def send(
|
|
self,
|
|
*,
|
|
channel_id: int | str,
|
|
message: RenderedMessage,
|
|
options: SendOptions | None = None,
|
|
) -> MessageRef:
|
|
ref = MessageRef(channel_id=channel_id, message_id=self._next_id)
|
|
self._next_id += 1
|
|
self.send_calls.append(
|
|
{
|
|
"ref": ref,
|
|
"channel_id": channel_id,
|
|
"message": message,
|
|
"options": options,
|
|
}
|
|
)
|
|
if (
|
|
self.progress_ref is None
|
|
and options is not None
|
|
and options.reply_to is not None
|
|
and options.notify is False
|
|
):
|
|
self.progress_ref = ref
|
|
if self.progress_ready is not None:
|
|
self.progress_ready.set()
|
|
return ref
|
|
|
|
async def edit(
|
|
self, *, ref: MessageRef, message: RenderedMessage, wait: bool = True
|
|
) -> MessageRef:
|
|
self.edit_calls.append({"ref": ref, "message": message, "wait": wait})
|
|
return ref
|
|
|
|
async def delete(self, *, ref: MessageRef) -> bool:
|
|
self.delete_calls.append(ref)
|
|
return True
|
|
|
|
async def close(self) -> None:
|
|
return None
|
|
|
|
|
|
class _FakeBot(BotClient):
|
|
def __init__(self) -> None:
|
|
self.command_calls: list[dict] = []
|
|
self.callback_calls: list[dict] = []
|
|
self.send_calls: list[dict] = []
|
|
self.document_calls: list[dict] = []
|
|
self.edit_calls: list[dict] = []
|
|
self.edit_topic_calls: list[dict[str, Any]] = []
|
|
self.delete_calls: list[dict] = []
|
|
|
|
async def get_updates(
|
|
self,
|
|
offset: int | None,
|
|
timeout_s: int = 50,
|
|
allowed_updates: list[str] | None = None,
|
|
) -> list[dict[str, Any]] | None:
|
|
_ = offset
|
|
_ = timeout_s
|
|
_ = allowed_updates
|
|
return []
|
|
|
|
async def get_file(self, file_id: str) -> dict[str, Any] | None:
|
|
_ = file_id
|
|
return None
|
|
|
|
async def download_file(self, file_path: str) -> bytes | None:
|
|
_ = file_path
|
|
return None
|
|
|
|
async def send_message(
|
|
self,
|
|
chat_id: int,
|
|
text: str,
|
|
reply_to_message_id: int | None = None,
|
|
disable_notification: bool | None = False,
|
|
message_thread_id: int | None = None,
|
|
entities: list[dict[str, Any]] | None = None,
|
|
parse_mode: str | None = None,
|
|
reply_markup: dict | None = None,
|
|
*,
|
|
replace_message_id: int | None = None,
|
|
) -> dict[str, Any]:
|
|
self.send_calls.append(
|
|
{
|
|
"chat_id": chat_id,
|
|
"text": text,
|
|
"reply_to_message_id": reply_to_message_id,
|
|
"disable_notification": disable_notification,
|
|
"message_thread_id": message_thread_id,
|
|
"entities": entities,
|
|
"parse_mode": parse_mode,
|
|
"reply_markup": reply_markup,
|
|
"replace_message_id": replace_message_id,
|
|
}
|
|
)
|
|
return {"message_id": 1}
|
|
|
|
async def send_document(
|
|
self,
|
|
chat_id: int,
|
|
filename: str,
|
|
content: bytes,
|
|
reply_to_message_id: int | None = None,
|
|
message_thread_id: int | None = None,
|
|
disable_notification: bool | None = False,
|
|
caption: str | None = None,
|
|
) -> dict[str, Any]:
|
|
self.document_calls.append(
|
|
{
|
|
"chat_id": chat_id,
|
|
"filename": filename,
|
|
"content": content,
|
|
"reply_to_message_id": reply_to_message_id,
|
|
"message_thread_id": message_thread_id,
|
|
"disable_notification": disable_notification,
|
|
"caption": caption,
|
|
}
|
|
)
|
|
return {"message_id": 2}
|
|
|
|
async def edit_message_text(
|
|
self,
|
|
chat_id: int,
|
|
message_id: int,
|
|
text: str,
|
|
entities: list[dict[str, Any]] | None = None,
|
|
parse_mode: str | None = None,
|
|
reply_markup: dict | None = None,
|
|
*,
|
|
wait: bool = True,
|
|
) -> dict[str, Any]:
|
|
self.edit_calls.append(
|
|
{
|
|
"chat_id": chat_id,
|
|
"message_id": message_id,
|
|
"text": text,
|
|
"entities": entities,
|
|
"parse_mode": parse_mode,
|
|
"reply_markup": reply_markup,
|
|
"wait": wait,
|
|
}
|
|
)
|
|
return {"message_id": message_id}
|
|
|
|
async def delete_message(self, chat_id: int, message_id: int) -> bool:
|
|
self.delete_calls.append({"chat_id": chat_id, "message_id": message_id})
|
|
return True
|
|
|
|
async def set_my_commands(
|
|
self,
|
|
commands: list[dict[str, Any]],
|
|
*,
|
|
scope: dict[str, Any] | None = None,
|
|
language_code: str | None = None,
|
|
) -> bool:
|
|
self.command_calls.append(
|
|
{
|
|
"commands": commands,
|
|
"scope": scope,
|
|
"language_code": language_code,
|
|
}
|
|
)
|
|
return True
|
|
|
|
async def get_me(self) -> dict[str, Any] | None:
|
|
return {"id": 1}
|
|
|
|
async def get_chat(self, chat_id: int) -> dict[str, Any] | None:
|
|
_ = chat_id
|
|
return {"id": chat_id, "type": "supergroup", "is_forum": True}
|
|
|
|
async def get_chat_member(
|
|
self, chat_id: int, user_id: int
|
|
) -> dict[str, Any] | None:
|
|
_ = chat_id
|
|
_ = user_id
|
|
return {"status": "administrator", "can_manage_topics": True}
|
|
|
|
async def create_forum_topic(
|
|
self, chat_id: int, name: str
|
|
) -> dict[str, Any] | None:
|
|
_ = chat_id
|
|
_ = name
|
|
return {"message_thread_id": 1}
|
|
|
|
async def edit_forum_topic(
|
|
self, chat_id: int, message_thread_id: int, name: str
|
|
) -> bool:
|
|
self.edit_topic_calls.append(
|
|
{
|
|
"chat_id": chat_id,
|
|
"message_thread_id": message_thread_id,
|
|
"name": name,
|
|
}
|
|
)
|
|
return True
|
|
|
|
async def close(self) -> None:
|
|
return None
|
|
|
|
async def answer_callback_query(
|
|
self,
|
|
callback_query_id: str,
|
|
text: str | None = None,
|
|
show_alert: bool | None = None,
|
|
) -> bool:
|
|
self.callback_calls.append(
|
|
{
|
|
"callback_query_id": callback_query_id,
|
|
"text": text,
|
|
"show_alert": show_alert,
|
|
}
|
|
)
|
|
return True
|
|
|
|
|
|
def _make_cfg(
|
|
transport: _FakeTransport, runner: ScriptRunner | None = None
|
|
) -> TelegramBridgeConfig:
|
|
if runner is None:
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=_empty_projects(),
|
|
)
|
|
return TelegramBridgeConfig(
|
|
bot=_FakeBot(),
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
)
|
|
|
|
|
|
def test_parse_directives_inline_engine() -> None:
|
|
directives = parse_directives(
|
|
"/claude do it",
|
|
engine_ids=("codex", "claude"),
|
|
projects=_empty_projects(),
|
|
)
|
|
assert directives.engine == "claude"
|
|
assert directives.prompt == "do it"
|
|
|
|
|
|
def test_parse_directives_newline() -> None:
|
|
directives = parse_directives(
|
|
"/codex\nhello",
|
|
engine_ids=("codex", "claude"),
|
|
projects=_empty_projects(),
|
|
)
|
|
assert directives.engine == "codex"
|
|
assert directives.prompt == "hello"
|
|
|
|
|
|
def test_parse_directives_ignores_unknown() -> None:
|
|
directives = parse_directives(
|
|
"/unknown hi",
|
|
engine_ids=("codex", "claude"),
|
|
projects=_empty_projects(),
|
|
)
|
|
assert directives.engine is None
|
|
assert directives.prompt == "/unknown hi"
|
|
|
|
|
|
def test_parse_directives_bot_suffix() -> None:
|
|
directives = parse_directives(
|
|
"/claude@bunny_agent_bot hi",
|
|
engine_ids=("claude",),
|
|
projects=_empty_projects(),
|
|
)
|
|
assert directives.engine == "claude"
|
|
assert directives.prompt == "hi"
|
|
|
|
|
|
def test_parse_directives_only_first_non_empty_line() -> None:
|
|
directives = parse_directives(
|
|
"hello\n/claude hi",
|
|
engine_ids=("codex", "claude"),
|
|
projects=_empty_projects(),
|
|
)
|
|
assert directives.engine is None
|
|
assert directives.prompt == "hello\n/claude hi"
|
|
|
|
|
|
def test_build_bot_commands_includes_cancel_and_engine() -> None:
|
|
runner = ScriptRunner(
|
|
[Return(answer="ok")], engine=CODEX_ENGINE, resume_value="sid"
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=_empty_projects(),
|
|
)
|
|
commands = build_bot_commands(runtime)
|
|
|
|
assert {"command": "cancel", "description": "cancel run"} in commands
|
|
assert {"command": "file", "description": "upload or fetch files"} in commands
|
|
assert any(cmd["command"] == "codex" for cmd in commands)
|
|
|
|
|
|
def test_build_bot_commands_includes_projects() -> None:
|
|
runner = ScriptRunner(
|
|
[Return(answer="ok")], engine=CODEX_ENGINE, resume_value="sid"
|
|
)
|
|
router = _make_router(runner)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
"good": ProjectConfig(
|
|
alias="good",
|
|
path=Path("."),
|
|
worktrees_dir=Path(".worktrees"),
|
|
),
|
|
"bad-name": ProjectConfig(
|
|
alias="bad-name",
|
|
path=Path("."),
|
|
worktrees_dir=Path(".worktrees"),
|
|
),
|
|
},
|
|
default_project=None,
|
|
)
|
|
|
|
runtime = TransportRuntime(router=router, projects=projects)
|
|
commands = build_bot_commands(runtime)
|
|
|
|
assert any(cmd["command"] == "good" for cmd in commands)
|
|
assert not any(cmd["command"] == "bad-name" for cmd in commands)
|
|
|
|
|
|
def test_build_bot_commands_includes_command_plugins(monkeypatch) -> None:
|
|
class _Command:
|
|
id = "pingcmd"
|
|
description = "ping command"
|
|
|
|
async def handle(self, ctx):
|
|
_ = ctx
|
|
return None
|
|
|
|
entrypoints = [
|
|
FakeEntryPoint(
|
|
"pingcmd",
|
|
"takopi.commands.ping:BACKEND",
|
|
plugins.COMMAND_GROUP,
|
|
loader=_Command,
|
|
)
|
|
]
|
|
install_entrypoints(monkeypatch, entrypoints)
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=_empty_projects(),
|
|
)
|
|
|
|
commands_list = build_bot_commands(runtime)
|
|
|
|
assert {"command": "pingcmd", "description": "ping command"} in commands_list
|
|
|
|
|
|
def test_build_bot_commands_caps_total() -> None:
|
|
runner = ScriptRunner(
|
|
[Return(answer="ok")], engine=CODEX_ENGINE, resume_value="sid"
|
|
)
|
|
router = _make_router(runner)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
f"proj{i}": ProjectConfig(
|
|
alias=f"proj{i}",
|
|
path=Path("."),
|
|
worktrees_dir=Path(".worktrees"),
|
|
)
|
|
for i in range(150)
|
|
},
|
|
default_project=None,
|
|
)
|
|
|
|
runtime = TransportRuntime(router=router, projects=projects)
|
|
commands = build_bot_commands(runtime)
|
|
|
|
assert len(commands) == 100
|
|
assert any(cmd["command"] == "codex" for cmd in commands)
|
|
assert any(cmd["command"] == "cancel" for cmd in commands)
|
|
|
|
|
|
def test_telegram_presenter_progress_shows_cancel_button() -> None:
|
|
presenter = TelegramPresenter()
|
|
state = ProgressTracker(engine="codex").snapshot()
|
|
|
|
rendered = presenter.render_progress(state, elapsed_s=0.0)
|
|
|
|
reply_markup = rendered.extra["reply_markup"]
|
|
assert reply_markup["inline_keyboard"][0][0]["text"] == "cancel"
|
|
assert reply_markup["inline_keyboard"][0][0]["callback_data"] == "takopi:cancel"
|
|
|
|
|
|
def test_telegram_presenter_clears_button_on_cancelled() -> None:
|
|
presenter = TelegramPresenter()
|
|
state = ProgressTracker(engine="codex").snapshot()
|
|
|
|
rendered = presenter.render_progress(state, elapsed_s=0.0, label="`cancelled`")
|
|
|
|
assert rendered.extra["reply_markup"]["inline_keyboard"] == []
|
|
|
|
|
|
def test_telegram_presenter_final_clears_button() -> None:
|
|
presenter = TelegramPresenter()
|
|
state = ProgressTracker(engine="codex").snapshot()
|
|
|
|
rendered = presenter.render_final(state, elapsed_s=0.0, status="done", answer="ok")
|
|
|
|
assert rendered.extra["reply_markup"]["inline_keyboard"] == []
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_telegram_transport_passes_replace_and_wait() -> None:
|
|
bot = _FakeBot()
|
|
transport = TelegramTransport(bot)
|
|
reply = MessageRef(channel_id=123, message_id=10)
|
|
replace = MessageRef(channel_id=123, message_id=11)
|
|
|
|
await transport.send(
|
|
channel_id=123,
|
|
message=RenderedMessage(text="hello"),
|
|
options=SendOptions(reply_to=reply, notify=True, replace=replace),
|
|
)
|
|
assert bot.send_calls
|
|
assert bot.send_calls[0]["replace_message_id"] == 11
|
|
|
|
await transport.edit(
|
|
ref=replace,
|
|
message=RenderedMessage(text="edit"),
|
|
wait=False,
|
|
)
|
|
assert bot.edit_calls
|
|
assert bot.edit_calls[0]["wait"] is False
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_telegram_transport_passes_reply_markup() -> None:
|
|
bot = _FakeBot()
|
|
transport = TelegramTransport(bot)
|
|
markup = {"inline_keyboard": []}
|
|
|
|
await transport.send(
|
|
channel_id=123,
|
|
message=RenderedMessage(text="hello", extra={"reply_markup": markup}),
|
|
)
|
|
assert bot.send_calls
|
|
assert bot.send_calls[0]["reply_markup"] == markup
|
|
|
|
ref = MessageRef(channel_id=123, message_id=1)
|
|
await transport.edit(
|
|
ref=ref,
|
|
message=RenderedMessage(text="edit", extra={"reply_markup": markup}),
|
|
)
|
|
assert bot.edit_calls
|
|
assert bot.edit_calls[0]["reply_markup"] == markup
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_telegram_transport_edit_wait_false_returns_ref() -> None:
|
|
class _OutboxBot(BotClient):
|
|
def __init__(self) -> None:
|
|
self.edit_calls: list[dict[str, Any]] = []
|
|
|
|
async def get_updates(
|
|
self,
|
|
offset: int | None,
|
|
timeout_s: int = 50,
|
|
allowed_updates: list[str] | None = None,
|
|
) -> list[dict[str, Any]] | None:
|
|
return None
|
|
|
|
async def get_file(self, file_id: str) -> dict[str, Any] | None:
|
|
_ = file_id
|
|
return None
|
|
|
|
async def download_file(self, file_path: str) -> bytes | None:
|
|
_ = file_path
|
|
return None
|
|
|
|
async def send_message(
|
|
self,
|
|
chat_id: int,
|
|
text: str,
|
|
reply_to_message_id: int | None = None,
|
|
disable_notification: bool | None = False,
|
|
message_thread_id: int | None = None,
|
|
entities: list[dict[str, Any]] | None = None,
|
|
parse_mode: str | None = None,
|
|
reply_markup: dict | None = None,
|
|
*,
|
|
replace_message_id: int | None = None,
|
|
) -> dict | None:
|
|
_ = reply_markup
|
|
return None
|
|
|
|
async def send_document(
|
|
self,
|
|
chat_id: int,
|
|
filename: str,
|
|
content: bytes,
|
|
reply_to_message_id: int | None = None,
|
|
message_thread_id: int | None = None,
|
|
disable_notification: bool | None = False,
|
|
caption: str | None = None,
|
|
) -> dict | None:
|
|
_ = (
|
|
chat_id,
|
|
filename,
|
|
content,
|
|
reply_to_message_id,
|
|
message_thread_id,
|
|
disable_notification,
|
|
caption,
|
|
)
|
|
return None
|
|
|
|
async def edit_message_text(
|
|
self,
|
|
chat_id: int,
|
|
message_id: int,
|
|
text: str,
|
|
entities: list[dict[str, Any]] | None = None,
|
|
parse_mode: str | None = None,
|
|
reply_markup: dict | None = None,
|
|
*,
|
|
wait: bool = True,
|
|
) -> dict | None:
|
|
self.edit_calls.append(
|
|
{
|
|
"chat_id": chat_id,
|
|
"message_id": message_id,
|
|
"text": text,
|
|
"entities": entities,
|
|
"parse_mode": parse_mode,
|
|
"reply_markup": reply_markup,
|
|
"wait": wait,
|
|
}
|
|
)
|
|
if not wait:
|
|
return None
|
|
return {"message_id": message_id}
|
|
|
|
async def delete_message(
|
|
self,
|
|
chat_id: int,
|
|
message_id: int,
|
|
) -> bool:
|
|
return False
|
|
|
|
async def set_my_commands(
|
|
self,
|
|
commands: list[dict[str, Any]],
|
|
*,
|
|
scope: dict[str, Any] | None = None,
|
|
language_code: str | None = None,
|
|
) -> bool:
|
|
return False
|
|
|
|
async def get_me(self) -> dict[str, Any] | None:
|
|
return None
|
|
|
|
async def close(self) -> None:
|
|
return None
|
|
|
|
async def answer_callback_query(
|
|
self,
|
|
callback_query_id: str,
|
|
text: str | None = None,
|
|
show_alert: bool | None = None,
|
|
) -> bool:
|
|
_ = callback_query_id, text, show_alert
|
|
return True
|
|
|
|
bot = _OutboxBot()
|
|
transport = TelegramTransport(bot)
|
|
ref = MessageRef(channel_id=123, message_id=1)
|
|
|
|
result = await transport.edit(
|
|
ref=ref,
|
|
message=RenderedMessage(text="edit"),
|
|
wait=False,
|
|
)
|
|
|
|
assert result == ref
|
|
assert bot.edit_calls
|
|
assert bot.edit_calls[0]["wait"] is False
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_cancel_without_reply_prompts_user() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
msg = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=10,
|
|
text="/cancel",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
running_tasks: dict = {}
|
|
|
|
await handle_cancel(cfg, msg, running_tasks)
|
|
|
|
assert len(transport.send_calls) == 1
|
|
assert "reply to the progress message" in transport.send_calls[0]["message"].text
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_cancel_with_no_progress_message_says_nothing_running() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
msg = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=10,
|
|
text="/cancel",
|
|
reply_to_message_id=None,
|
|
reply_to_text="no message id",
|
|
sender_id=123,
|
|
)
|
|
running_tasks: dict = {}
|
|
|
|
await handle_cancel(cfg, msg, running_tasks)
|
|
|
|
assert len(transport.send_calls) == 1
|
|
assert "nothing is currently running" in transport.send_calls[0]["message"].text
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_cancel_with_finished_task_says_nothing_running() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
progress_id = 99
|
|
msg = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=10,
|
|
text="/cancel",
|
|
reply_to_message_id=progress_id,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
running_tasks: dict = {}
|
|
|
|
await handle_cancel(cfg, msg, running_tasks)
|
|
|
|
assert len(transport.send_calls) == 1
|
|
assert "nothing is currently running" in transport.send_calls[0]["message"].text
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_cancel_cancels_running_task() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
progress_id = 42
|
|
msg = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=10,
|
|
text="/cancel",
|
|
reply_to_message_id=progress_id,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
|
|
running_task = RunningTask()
|
|
running_tasks = {MessageRef(channel_id=123, message_id=progress_id): running_task}
|
|
await handle_cancel(cfg, msg, running_tasks)
|
|
|
|
assert running_task.cancel_requested.is_set() is True
|
|
assert len(transport.send_calls) == 0 # No error message sent
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_cancel_only_cancels_matching_progress_message() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
task_first = RunningTask()
|
|
task_second = RunningTask()
|
|
msg = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=10,
|
|
text="/cancel",
|
|
reply_to_message_id=1,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
running_tasks = {
|
|
MessageRef(channel_id=123, message_id=1): task_first,
|
|
MessageRef(channel_id=123, message_id=2): task_second,
|
|
}
|
|
|
|
await handle_cancel(cfg, msg, running_tasks)
|
|
|
|
assert task_first.cancel_requested.is_set() is True
|
|
assert task_second.cancel_requested.is_set() is False
|
|
assert len(transport.send_calls) == 0
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_file_put_writes_file(tmp_path: Path) -> None:
|
|
payload = b"hello"
|
|
|
|
class _FileBot(_FakeBot):
|
|
async def get_file(self, file_id: str) -> dict[str, Any] | None:
|
|
_ = file_id
|
|
return {"file_path": "files/hello.txt"}
|
|
|
|
async def download_file(self, file_path: str) -> bytes | None:
|
|
_ = file_path
|
|
return payload
|
|
|
|
transport = _FakeTransport()
|
|
bot = _FileBot()
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
"proj": ProjectConfig(
|
|
alias="proj",
|
|
path=tmp_path,
|
|
worktrees_dir=Path(".worktrees"),
|
|
)
|
|
},
|
|
default_project=None,
|
|
)
|
|
runtime = TransportRuntime(router=_make_router(runner), projects=projects)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
files=TelegramFilesConfig(enabled=True),
|
|
)
|
|
msg = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=10,
|
|
text="",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=321,
|
|
chat_type="private",
|
|
document=TelegramDocument(
|
|
file_id="doc-id",
|
|
file_name="hello.txt",
|
|
mime_type="text/plain",
|
|
file_size=len(payload),
|
|
raw={"file_id": "doc-id"},
|
|
),
|
|
)
|
|
|
|
await telegram_commands._handle_file_put(
|
|
cfg, msg, "/proj uploads/hello.txt", None, None
|
|
)
|
|
|
|
target = tmp_path / "uploads" / "hello.txt"
|
|
assert target.read_bytes() == payload
|
|
assert transport.send_calls
|
|
text = transport.send_calls[-1]["message"].text
|
|
assert "saved uploads/hello.txt" in text
|
|
assert "(5 b)" in text
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_file_get_sends_document_for_allowed_user(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
payload = b"fetch"
|
|
target = tmp_path / "hello.txt"
|
|
target.write_bytes(payload)
|
|
|
|
transport = _FakeTransport()
|
|
bot = _FakeBot()
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
"proj": ProjectConfig(
|
|
alias="proj",
|
|
path=tmp_path,
|
|
worktrees_dir=Path(".worktrees"),
|
|
)
|
|
},
|
|
default_project=None,
|
|
)
|
|
runtime = TransportRuntime(router=_make_router(runner), projects=projects)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
files=TelegramFilesConfig(
|
|
enabled=True,
|
|
allowed_user_ids=frozenset({42}),
|
|
),
|
|
)
|
|
msg = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=-100,
|
|
message_id=10,
|
|
text="",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=42,
|
|
chat_type="supergroup",
|
|
)
|
|
|
|
await telegram_commands._handle_file_get(cfg, msg, "/proj hello.txt", None, None)
|
|
|
|
assert bot.document_calls
|
|
assert bot.document_calls[0]["filename"] == "hello.txt"
|
|
assert bot.document_calls[0]["content"] == payload
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_callback_cancel_cancels_running_task() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
progress_id = 42
|
|
running_task = RunningTask()
|
|
running_tasks = {MessageRef(channel_id=123, message_id=progress_id): running_task}
|
|
query = TelegramCallbackQuery(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=progress_id,
|
|
callback_query_id="cbq-1",
|
|
data="takopi:cancel",
|
|
sender_id=123,
|
|
)
|
|
|
|
await handle_callback_cancel(cfg, query, running_tasks)
|
|
|
|
assert running_task.cancel_requested.is_set() is True
|
|
assert len(transport.send_calls) == 0
|
|
bot = cast(_FakeBot, cfg.bot)
|
|
assert bot.callback_calls
|
|
assert bot.callback_calls[-1]["text"] == "cancelling..."
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_handle_callback_cancel_without_task_acknowledges() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
query = TelegramCallbackQuery(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=99,
|
|
callback_query_id="cbq-2",
|
|
data="takopi:cancel",
|
|
sender_id=123,
|
|
)
|
|
|
|
await handle_callback_cancel(cfg, query, {})
|
|
|
|
assert len(transport.send_calls) == 0
|
|
bot = cast(_FakeBot, cfg.bot)
|
|
assert bot.callback_calls
|
|
assert "nothing is currently running" in bot.callback_calls[-1]["text"].lower()
|
|
|
|
|
|
def test_cancel_command_accepts_extra_text() -> None:
|
|
assert is_cancel_command("/cancel now") is True
|
|
assert is_cancel_command("/cancel@takopi please") is True
|
|
assert is_cancel_command("/cancelled") is False
|
|
|
|
|
|
def test_resolve_message_accepts_backticked_ctx_line() -> None:
|
|
runtime = TransportRuntime(
|
|
router=_make_router(ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)),
|
|
projects=ProjectsConfig(
|
|
projects={
|
|
"takopi": ProjectConfig(
|
|
alias="takopi",
|
|
path=Path("."),
|
|
worktrees_dir=Path(".worktrees"),
|
|
)
|
|
},
|
|
default_project=None,
|
|
),
|
|
)
|
|
resolved = runtime.resolve_message(
|
|
text="do it",
|
|
reply_text="`ctx: takopi @ feat/api`",
|
|
)
|
|
|
|
assert resolved.prompt == "do it"
|
|
assert resolved.resume_token is None
|
|
assert resolved.engine_override is None
|
|
assert resolved.context == RunContext(project="takopi", branch="feat/api")
|
|
|
|
|
|
def test_topic_title_matches_command_syntax() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
|
|
title = telegram_topics._topic_title(
|
|
runtime=cfg.runtime,
|
|
context=RunContext(project="takopi", branch="master"),
|
|
)
|
|
|
|
assert title == "takopi @master"
|
|
|
|
title = telegram_topics._topic_title(
|
|
runtime=cfg.runtime,
|
|
context=RunContext(project="takopi", branch=None),
|
|
)
|
|
|
|
assert title == "takopi"
|
|
|
|
title = telegram_topics._topic_title(
|
|
runtime=cfg.runtime,
|
|
context=RunContext(project=None, branch="main"),
|
|
)
|
|
|
|
assert title == "@main"
|
|
|
|
|
|
def test_topic_title_projects_scope_includes_project() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = replace(
|
|
_make_cfg(transport),
|
|
topics=bridge.TelegramTopicsConfig(
|
|
enabled=True,
|
|
scope="projects",
|
|
),
|
|
)
|
|
|
|
title = telegram_topics._topic_title(
|
|
runtime=cfg.runtime,
|
|
context=RunContext(project="takopi", branch="master"),
|
|
)
|
|
|
|
assert title == "takopi @master"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_maybe_rename_topic_updates_title(tmp_path: Path) -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
store = TopicStateStore(tmp_path / "telegram_topics_state.json")
|
|
|
|
await store.set_context(
|
|
123,
|
|
77,
|
|
RunContext(project="takopi", branch="old"),
|
|
topic_title="takopi @old",
|
|
)
|
|
|
|
await telegram_topics._maybe_rename_topic(
|
|
cfg,
|
|
store,
|
|
chat_id=123,
|
|
thread_id=77,
|
|
context=RunContext(project="takopi", branch="new"),
|
|
)
|
|
|
|
bot = cast(_FakeBot, cfg.bot)
|
|
assert bot.edit_topic_calls
|
|
assert bot.edit_topic_calls[-1]["name"] == "takopi @new"
|
|
snapshot = await store.get_thread(123, 77)
|
|
assert snapshot is not None
|
|
assert snapshot.topic_title == "takopi @new"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_maybe_rename_topic_skips_when_title_matches(tmp_path: Path) -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
store = TopicStateStore(tmp_path / "telegram_topics_state.json")
|
|
|
|
await store.set_context(
|
|
123,
|
|
77,
|
|
RunContext(project="takopi", branch="main"),
|
|
topic_title="takopi @main",
|
|
)
|
|
snapshot = await store.get_thread(123, 77)
|
|
|
|
await telegram_topics._maybe_rename_topic(
|
|
cfg,
|
|
store,
|
|
chat_id=123,
|
|
thread_id=77,
|
|
context=RunContext(project="takopi", branch="main"),
|
|
snapshot=snapshot,
|
|
)
|
|
|
|
bot = cast(_FakeBot, cfg.bot)
|
|
assert bot.edit_topic_calls == []
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_send_with_resume_waits_for_token() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
sent: list[tuple[int, int, str, ResumeToken, RunContext | None, int | None]] = []
|
|
|
|
async def enqueue(
|
|
chat_id: int,
|
|
user_msg_id: int,
|
|
text: str,
|
|
resume: ResumeToken,
|
|
context: RunContext | None,
|
|
thread_id: int | None,
|
|
) -> None:
|
|
sent.append((chat_id, user_msg_id, text, resume, context, thread_id))
|
|
|
|
running_task = RunningTask()
|
|
|
|
async def trigger_resume() -> None:
|
|
await anyio.sleep(0)
|
|
running_task.resume = ResumeToken(engine=CODEX_ENGINE, value="abc123")
|
|
running_task.resume_ready.set()
|
|
|
|
async with anyio.create_task_group() as tg:
|
|
tg.start_soon(trigger_resume)
|
|
await send_with_resume(
|
|
cfg,
|
|
enqueue,
|
|
running_task,
|
|
123,
|
|
10,
|
|
None,
|
|
"hello",
|
|
)
|
|
|
|
assert sent == [
|
|
(
|
|
123,
|
|
10,
|
|
"hello",
|
|
ResumeToken(engine=CODEX_ENGINE, value="abc123"),
|
|
None,
|
|
None,
|
|
)
|
|
]
|
|
assert transport.send_calls == []
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_send_with_resume_reports_when_missing() -> None:
|
|
transport = _FakeTransport()
|
|
cfg = _make_cfg(transport)
|
|
sent: list[tuple[int, int, str, ResumeToken, RunContext | None, int | None]] = []
|
|
|
|
async def enqueue(
|
|
chat_id: int,
|
|
user_msg_id: int,
|
|
text: str,
|
|
resume: ResumeToken,
|
|
context: RunContext | None,
|
|
thread_id: int | None,
|
|
) -> None:
|
|
sent.append((chat_id, user_msg_id, text, resume, context, thread_id))
|
|
|
|
running_task = RunningTask()
|
|
running_task.done.set()
|
|
|
|
await send_with_resume(
|
|
cfg,
|
|
enqueue,
|
|
running_task,
|
|
123,
|
|
10,
|
|
None,
|
|
"hello",
|
|
)
|
|
|
|
assert sent == []
|
|
assert transport.send_calls
|
|
assert "resume token" in transport.send_calls[-1]["message"].text.lower()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_routes_reply_to_running_resume() -> None:
|
|
progress_ready = anyio.Event()
|
|
stop_polling = anyio.Event()
|
|
reply_ready = anyio.Event()
|
|
hold = anyio.Event()
|
|
|
|
transport = _FakeTransport(progress_ready=progress_ready)
|
|
bot = _FakeBot()
|
|
resume_value = "abc123"
|
|
runner = ScriptRunner(
|
|
[Wait(hold), Sleep(0.05), Return(answer="ok")],
|
|
engine=CODEX_ENGINE,
|
|
resume_value=resume_value,
|
|
)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=_empty_projects(),
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
)
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=1,
|
|
text="first",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
await progress_ready.wait()
|
|
assert transport.progress_ref is not None
|
|
assert isinstance(transport.progress_ref.message_id, int)
|
|
reply_id = transport.progress_ref.message_id
|
|
reply_ready.set()
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=2,
|
|
text="followup",
|
|
reply_to_message_id=reply_id,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
await stop_polling.wait()
|
|
|
|
async with anyio.create_task_group() as tg:
|
|
tg.start_soon(run_main_loop, cfg, poller)
|
|
try:
|
|
with anyio.fail_after(2):
|
|
await reply_ready.wait()
|
|
await anyio.sleep(0)
|
|
hold.set()
|
|
with anyio.fail_after(2):
|
|
while len(runner.calls) < 2:
|
|
await anyio.sleep(0)
|
|
assert runner.calls[1][1] == ResumeToken(
|
|
engine=CODEX_ENGINE, value=resume_value
|
|
)
|
|
finally:
|
|
hold.set()
|
|
stop_polling.set()
|
|
tg.cancel_scope.cancel()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_persists_topic_sessions_in_project_scope(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
project_chat_id = -100
|
|
resume_value = "resume-123"
|
|
|
|
transport = _FakeTransport()
|
|
bot = _FakeBot()
|
|
runner = ScriptRunner(
|
|
[Return(answer="ok")],
|
|
engine=CODEX_ENGINE,
|
|
resume_value=resume_value,
|
|
)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
"takopi": ProjectConfig(
|
|
alias="takopi",
|
|
path=Path("."),
|
|
worktrees_dir=Path(".worktrees"),
|
|
chat_id=project_chat_id,
|
|
)
|
|
},
|
|
default_project=None,
|
|
chat_map={project_chat_id: "takopi"},
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=projects,
|
|
config_path=tmp_path / "takopi.toml",
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
topics=bridge.TelegramTopicsConfig(
|
|
enabled=True,
|
|
scope="projects",
|
|
),
|
|
)
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=project_chat_id,
|
|
message_id=1,
|
|
text="hello",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
thread_id=77,
|
|
)
|
|
|
|
with anyio.fail_after(2):
|
|
await run_main_loop(cfg, poller)
|
|
|
|
state_path = resolve_state_path(runtime.config_path or tmp_path / "takopi.toml")
|
|
store = TopicStateStore(state_path)
|
|
stored = await store.get_session_resume(project_chat_id, 77, CODEX_ENGINE)
|
|
assert stored == ResumeToken(engine=CODEX_ENGINE, value=resume_value)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_replies_in_same_thread() -> None:
|
|
transport = _FakeTransport()
|
|
bot = _FakeBot()
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=_empty_projects(),
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
)
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=1,
|
|
text="hello",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
thread_id=77,
|
|
)
|
|
|
|
await run_main_loop(cfg, poller)
|
|
|
|
reply_calls = [
|
|
call
|
|
for call in transport.send_calls
|
|
if call["options"] is not None and call["options"].reply_to is not None
|
|
]
|
|
assert reply_calls
|
|
assert all(call["options"].thread_id == 77 for call in reply_calls)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_batches_media_group_upload(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
payloads = {
|
|
"photos/file_1.jpg": b"one",
|
|
"photos/file_2.jpg": b"two",
|
|
}
|
|
file_map = {
|
|
"doc-1": "photos/file_1.jpg",
|
|
"doc-2": "photos/file_2.jpg",
|
|
}
|
|
|
|
class _MediaBot(_FakeBot):
|
|
async def get_file(self, file_id: str) -> dict[str, Any] | None:
|
|
file_path = file_map.get(file_id)
|
|
if file_path is None:
|
|
return None
|
|
return {"file_path": file_path}
|
|
|
|
async def download_file(self, file_path: str) -> bytes | None:
|
|
return payloads.get(file_path)
|
|
|
|
transport = _FakeTransport()
|
|
bot = _MediaBot()
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
"proj": ProjectConfig(
|
|
alias="proj",
|
|
path=tmp_path,
|
|
worktrees_dir=Path(".worktrees"),
|
|
)
|
|
},
|
|
default_project=None,
|
|
)
|
|
runtime = TransportRuntime(router=_make_router(runner), projects=projects)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
files=TelegramFilesConfig(enabled=True, auto_put=True),
|
|
)
|
|
msg1 = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=1,
|
|
text="/file put /proj incoming/test1",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=321,
|
|
chat_type="private",
|
|
media_group_id="grp-1",
|
|
document=TelegramDocument(
|
|
file_id="doc-1",
|
|
file_name=None,
|
|
mime_type="image/jpeg",
|
|
file_size=len(payloads["photos/file_1.jpg"]),
|
|
raw={"file_id": "doc-1"},
|
|
),
|
|
)
|
|
msg2 = TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=2,
|
|
text="",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=321,
|
|
chat_type="private",
|
|
media_group_id="grp-1",
|
|
document=TelegramDocument(
|
|
file_id="doc-2",
|
|
file_name=None,
|
|
mime_type="image/jpeg",
|
|
file_size=len(payloads["photos/file_2.jpg"]),
|
|
raw={"file_id": "doc-2"},
|
|
),
|
|
)
|
|
|
|
stop_polling = anyio.Event()
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield msg1
|
|
yield msg2
|
|
await stop_polling.wait()
|
|
|
|
async with anyio.create_task_group() as tg:
|
|
tg.start_soon(run_main_loop, cfg, poller)
|
|
try:
|
|
with anyio.fail_after(3):
|
|
while len(transport.send_calls) < 1:
|
|
await anyio.sleep(0.05)
|
|
assert len(transport.send_calls) == 1
|
|
text = transport.send_calls[0]["message"].text
|
|
assert "saved file_1.jpg, file_2.jpg" in text
|
|
assert "to incoming/test1/" in text
|
|
target_dir = tmp_path / "incoming" / "test1"
|
|
assert (target_dir / "file_1.jpg").read_bytes() == payloads[
|
|
"photos/file_1.jpg"
|
|
]
|
|
assert (target_dir / "file_2.jpg").read_bytes() == payloads[
|
|
"photos/file_2.jpg"
|
|
]
|
|
finally:
|
|
stop_polling.set()
|
|
tg.cancel_scope.cancel()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_handles_command_plugins(monkeypatch) -> None:
|
|
class _Command:
|
|
id = "echo_cmd"
|
|
description = "echo"
|
|
|
|
async def handle(self, ctx):
|
|
return commands.CommandResult(text=f"echo:{ctx.args_text}")
|
|
|
|
entrypoints = [
|
|
FakeEntryPoint(
|
|
"echo_cmd",
|
|
"takopi.commands.echo:BACKEND",
|
|
plugins.COMMAND_GROUP,
|
|
loader=_Command,
|
|
)
|
|
]
|
|
install_entrypoints(monkeypatch, entrypoints)
|
|
|
|
transport = _FakeTransport()
|
|
bot = _FakeBot()
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=_empty_projects(),
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
)
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=1,
|
|
text="/echo_cmd hello",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
|
|
await run_main_loop(cfg, poller)
|
|
|
|
assert runner.calls == []
|
|
assert transport.send_calls
|
|
assert transport.send_calls[-1]["message"].text == "echo:hello"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_command_uses_project_default_engine(
|
|
monkeypatch,
|
|
) -> None:
|
|
class _Command:
|
|
id = "use_project"
|
|
description = "use project default"
|
|
|
|
async def handle(self, ctx):
|
|
result = await ctx.executor.run_one(
|
|
commands.RunRequest(
|
|
prompt="hello",
|
|
context=RunContext(project="proj"),
|
|
),
|
|
mode="capture",
|
|
)
|
|
return commands.CommandResult(text=f"ran:{result.engine}")
|
|
|
|
entrypoints = [
|
|
FakeEntryPoint(
|
|
"use_project",
|
|
"takopi.commands.use_project:BACKEND",
|
|
plugins.COMMAND_GROUP,
|
|
loader=_Command,
|
|
)
|
|
]
|
|
install_entrypoints(monkeypatch, entrypoints)
|
|
|
|
transport = _FakeTransport()
|
|
bot = _FakeBot()
|
|
codex_runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
pi_runner = ScriptRunner([Return(answer="ok")], engine=EngineId("pi"))
|
|
router = AutoRouter(
|
|
entries=[
|
|
RunnerEntry(engine=codex_runner.engine, runner=codex_runner),
|
|
RunnerEntry(engine=pi_runner.engine, runner=pi_runner),
|
|
],
|
|
default_engine=codex_runner.engine,
|
|
)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
"proj": ProjectConfig(
|
|
alias="proj",
|
|
path=Path("."),
|
|
worktrees_dir=Path(".worktrees"),
|
|
default_engine=pi_runner.engine,
|
|
)
|
|
},
|
|
default_project=None,
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=router,
|
|
projects=projects,
|
|
)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
)
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=1,
|
|
text="/use_project",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
|
|
await run_main_loop(cfg, poller)
|
|
|
|
assert codex_runner.calls == []
|
|
assert len(pi_runner.calls) == 1
|
|
assert transport.send_calls[-1]["message"].text == "ran:pi"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_command_defaults_to_chat_project(
|
|
monkeypatch,
|
|
) -> None:
|
|
class _Command:
|
|
id = "auto_ctx"
|
|
description = "auto context"
|
|
|
|
async def handle(self, ctx):
|
|
result = await ctx.executor.run_one(
|
|
commands.RunRequest(prompt="hello"),
|
|
mode="capture",
|
|
)
|
|
return commands.CommandResult(text=f"ran:{result.engine}")
|
|
|
|
entrypoints = [
|
|
FakeEntryPoint(
|
|
"auto_ctx",
|
|
"takopi.commands.auto_ctx:BACKEND",
|
|
plugins.COMMAND_GROUP,
|
|
loader=_Command,
|
|
)
|
|
]
|
|
install_entrypoints(monkeypatch, entrypoints)
|
|
|
|
transport = _FakeTransport()
|
|
bot = _FakeBot()
|
|
codex_runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
pi_runner = ScriptRunner([Return(answer="ok")], engine=EngineId("pi"))
|
|
router = AutoRouter(
|
|
entries=[
|
|
RunnerEntry(engine=codex_runner.engine, runner=codex_runner),
|
|
RunnerEntry(engine=pi_runner.engine, runner=pi_runner),
|
|
],
|
|
default_engine=codex_runner.engine,
|
|
)
|
|
projects = ProjectsConfig(
|
|
projects={
|
|
"proj": ProjectConfig(
|
|
alias="proj",
|
|
path=Path("."),
|
|
worktrees_dir=Path(".worktrees"),
|
|
default_engine=pi_runner.engine,
|
|
chat_id=-42,
|
|
)
|
|
},
|
|
default_project=None,
|
|
chat_map={-42: "proj"},
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=router,
|
|
projects=projects,
|
|
)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
)
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=-42,
|
|
message_id=1,
|
|
text="/auto_ctx",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
|
|
await run_main_loop(cfg, poller)
|
|
|
|
assert codex_runner.calls == []
|
|
assert len(pi_runner.calls) == 1
|
|
assert transport.send_calls[-1]["message"].text == "ran:pi"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_main_loop_refreshes_command_ids(monkeypatch) -> None:
|
|
class _Command:
|
|
id = "late_cmd"
|
|
description = "late command"
|
|
|
|
async def handle(self, ctx):
|
|
return commands.CommandResult(text="late")
|
|
|
|
entrypoints = [
|
|
FakeEntryPoint(
|
|
"late_cmd",
|
|
"takopi.commands.late:BACKEND",
|
|
plugins.COMMAND_GROUP,
|
|
loader=_Command,
|
|
)
|
|
]
|
|
install_entrypoints(monkeypatch, entrypoints)
|
|
|
|
calls = {"count": 0}
|
|
|
|
def _list_command_ids(*, allowlist=None):
|
|
_ = allowlist
|
|
calls["count"] += 1
|
|
if calls["count"] == 1:
|
|
return []
|
|
return ["late_cmd"]
|
|
|
|
monkeypatch.setattr(telegram_loop, "list_command_ids", _list_command_ids)
|
|
|
|
transport = _FakeTransport()
|
|
bot = _FakeBot()
|
|
runner = ScriptRunner([Return(answer="ok")], engine=CODEX_ENGINE)
|
|
exec_cfg = ExecBridgeConfig(
|
|
transport=transport,
|
|
presenter=MarkdownPresenter(),
|
|
final_notify=True,
|
|
)
|
|
runtime = TransportRuntime(
|
|
router=_make_router(runner),
|
|
projects=_empty_projects(),
|
|
)
|
|
cfg = TelegramBridgeConfig(
|
|
bot=bot,
|
|
runtime=runtime,
|
|
chat_id=123,
|
|
startup_msg="",
|
|
exec_cfg=exec_cfg,
|
|
)
|
|
|
|
async def poller(_cfg: TelegramBridgeConfig):
|
|
yield TelegramIncomingMessage(
|
|
transport="telegram",
|
|
chat_id=123,
|
|
message_id=1,
|
|
text="/late_cmd hello",
|
|
reply_to_message_id=None,
|
|
reply_to_text=None,
|
|
sender_id=123,
|
|
)
|
|
|
|
await run_main_loop(cfg, poller)
|
|
|
|
assert calls["count"] >= 2
|
|
assert transport.send_calls[-1]["message"].text == "late"
|