mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-07-13 00:13:02 +08:00
fix: parse feishu post messages
This commit is contained in:
100
tests/test_feishu_post_message.py
Normal file
100
tests/test_feishu_post_message.py
Normal file
@@ -0,0 +1,100 @@
|
||||
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
|
||||
Reference in New Issue
Block a user