mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/jxxghp/MoviePilot.git
synced 2026-09-20 08:03:34 +08:00
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
from urllib.parse import quote
|
|
|
|
import pytest
|
|
|
|
from app.api.endpoints.agent import run_agent_command
|
|
from app.schemas.agent import AgentCommandRunRequest
|
|
from app.schemas.types import NotificationChannel
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_agent_command_endpoint_decodes_internal_channel_headers() -> None:
|
|
"""命令端点应解析 ASCII 编码的渠道，并还原编码后的来源名称。"""
|
|
user = SimpleNamespace(id=1, is_superuser=True)
|
|
command_result = {
|
|
"message": "命令已触发",
|
|
"command": "/demo",
|
|
"command_desc": "示例命令",
|
|
}
|
|
|
|
with patch(
|
|
"app.api.endpoints.agent.web_agent_application.dispatch_command",
|
|
return_value=command_result,
|
|
) as dispatch_command:
|
|
response = await run_agent_command(
|
|
AgentCommandRunRequest(command="/demo"),
|
|
current_user=user,
|
|
agent_channel=quote(NotificationChannel.Wechat.value, safe=""),
|
|
agent_source="%E4%BC%81%E4%B8%9A%E5%BE%AE%E4%BF%A1",
|
|
)
|
|
|
|
assert response.success is True
|
|
dispatch_command.assert_called_once_with(
|
|
"/demo",
|
|
user_id="1",
|
|
channel=NotificationChannel.Wechat,
|
|
source="企业微信",
|
|
publish_event=dispatch_command.call_args.kwargs["publish_event"],
|
|
)
|