mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-06-25 09:34:19 +08:00
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
import json
|
|
from types import SimpleNamespace
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from app.testing.bootstrap import ensure_optional_stub
|
|
|
|
ensure_optional_stub("psutil")
|
|
ensure_optional_stub("dateparser")
|
|
ensure_optional_stub("Pinyin2Hanzi", is_pinyin=lambda value: False)
|
|
|
|
from app.modules.feishu.feishu import Feishu
|
|
|
|
|
|
def _build_feishu_client() -> Feishu:
|
|
"""构造不会启动飞书长连接的测试客户端。"""
|
|
with (
|
|
patch.object(Feishu, "_build_api_client", return_value=MagicMock()),
|
|
patch.object(Feishu, "_start_ws_client"),
|
|
):
|
|
return Feishu(
|
|
FEISHU_APP_ID="test_app_id",
|
|
FEISHU_APP_SECRET="test_app_secret",
|
|
name="feishu-test",
|
|
)
|
|
|
|
|
|
def test_on_message_extracts_localized_post_text_and_images():
|
|
"""飞书富文本事件应提取标题、正文、链接和图片引用。"""
|
|
client = _build_feishu_client()
|
|
message = SimpleNamespace(
|
|
message_id="om_post_evt",
|
|
chat_id="oc_chat_evt",
|
|
chat_type="p2p",
|
|
message_type="post",
|
|
content=json.dumps(
|
|
{
|
|
"post": {
|
|
"zh_cn": {
|
|
"title": "搜索请求",
|
|
"content": [
|
|
[
|
|
{"tag": "text", "text": "/search "},
|
|
{
|
|
"tag": "a",
|
|
"text": "MoviePilot",
|
|
"href": "https://example.com/moviepilot",
|
|
},
|
|
],
|
|
[
|
|
{"tag": "at", "user_name": "管理员"},
|
|
{"tag": "text", "text": " 请处理"},
|
|
],
|
|
[{"tag": "img", "image_key": "img_v2_post"}],
|
|
],
|
|
}
|
|
}
|
|
},
|
|
ensure_ascii=False,
|
|
),
|
|
)
|
|
sender = SimpleNamespace(
|
|
sender_id=SimpleNamespace(open_id="ou_user_evt", user_id=None)
|
|
)
|
|
event = SimpleNamespace(sender=sender, message=message)
|
|
|
|
with patch.object(client, "_forward_to_message_chain") as forward:
|
|
client._on_message(SimpleNamespace(event=event))
|
|
|
|
payload = forward.call_args.args[0]
|
|
assert payload["text"] == (
|
|
"搜索请求\n"
|
|
"/search MoviePilot https://example.com/moviepilot\n"
|
|
"@管理员 请处理"
|
|
)
|
|
assert payload["images"][0]["ref"] == "feishu://image/om_post_evt/img_v2_post"
|
|
|
|
|
|
def test_parse_message_content_supports_direct_post_body():
|
|
"""飞书富文本直接正文结构应被转换为普通文本。"""
|
|
message = SimpleNamespace(
|
|
message_id="om_post_direct",
|
|
message_type="post",
|
|
content=json.dumps(
|
|
{
|
|
"title": "",
|
|
"content": [
|
|
[{"tag": "text", "text": "/help"}],
|
|
[{"tag": "text", "text": "第二行"}],
|
|
],
|
|
},
|
|
ensure_ascii=False,
|
|
),
|
|
)
|
|
|
|
text, images, audio_refs, files = Feishu._parse_message_content(message)
|
|
|
|
assert text == "/help\n第二行"
|
|
assert images is None
|
|
assert audio_refs is None
|
|
assert files is None
|