mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 20:17:13 +08:00
124 lines
4.4 KiB
Python
124 lines
4.4 KiB
Python
"""钉钉自定义机器人通知渠道测试。"""
|
|
|
|
import base64
|
|
import hashlib
|
|
import hmac
|
|
from types import SimpleNamespace
|
|
from unittest.mock import Mock
|
|
from urllib.parse import parse_qs, urlsplit
|
|
|
|
from app.modules.dingtalk import DingTalkModule
|
|
from app.modules.dingtalk.dingtalk import DingTalk
|
|
from app.schemas.message import Message
|
|
from app.schemas.notification import ChannelCapability, ChannelCapabilityManager
|
|
from app.schemas.types import MessageType, NotificationChannel
|
|
|
|
|
|
def _response(payload: dict, status_code: int = 200) -> Mock:
|
|
"""构造带关闭能力的 requests.Response 测试替身。"""
|
|
response = Mock(status_code=status_code)
|
|
response.json.return_value = payload
|
|
return response
|
|
|
|
|
|
def test_signed_webhook_preserves_access_token_and_uses_official_signature() -> None:
|
|
"""签名应使用毫秒时间戳、换行分隔串和 HMAC-SHA256。"""
|
|
client = DingTalk(
|
|
DINGTALK_WEBHOOK="https://oapi.dingtalk.com/robot/send?access_token=token",
|
|
DINGTALK_SECRET="SEC-test",
|
|
)
|
|
|
|
signed_url = client.build_request_url(timestamp=1720000000123)
|
|
query = parse_qs(urlsplit(signed_url).query)
|
|
expected = base64.b64encode(
|
|
hmac.new(
|
|
b"SEC-test",
|
|
b"1720000000123\nSEC-test",
|
|
digestmod=hashlib.sha256,
|
|
).digest()
|
|
).decode("utf-8")
|
|
|
|
assert query == {
|
|
"access_token": ["token"],
|
|
"timestamp": ["1720000000123"],
|
|
"sign": [expected],
|
|
}
|
|
|
|
|
|
def test_send_msg_posts_markdown_and_accepts_dingtalk_success() -> None:
|
|
"""普通通知应保留标题、正文、图片和详情链接并检查 errcode。"""
|
|
client = DingTalk(
|
|
DINGTALK_WEBHOOK="https://oapi.dingtalk.com/robot/send?access_token=token"
|
|
)
|
|
response = _response({"errcode": 0, "errmsg": "ok"})
|
|
client._req = Mock()
|
|
client._req.post_res.return_value = response
|
|
|
|
assert client.send_msg(
|
|
title="整理完成",
|
|
text="电影已入库",
|
|
image="https://example.com/poster.jpg",
|
|
link="https://example.com/history",
|
|
) is True
|
|
|
|
request = client._req.post_res.call_args.kwargs
|
|
assert request["url"].endswith("access_token=token")
|
|
assert request["json"]["msgtype"] == "markdown"
|
|
assert request["json"]["markdown"]["title"] == "整理完成"
|
|
markdown = request["json"]["markdown"]["text"]
|
|
assert "### 整理完成" in markdown
|
|
assert "电影已入库" in markdown
|
|
assert "" in markdown
|
|
assert "[查看详情](https://example.com/history)" in markdown
|
|
response.close.assert_called_once_with()
|
|
|
|
|
|
def test_send_msg_rejects_http_and_business_failures() -> None:
|
|
"""HTTP 失败或钉钉业务错误都不能被误判为发送成功。"""
|
|
client = DingTalk(
|
|
DINGTALK_WEBHOOK="https://oapi.dingtalk.com/robot/send?access_token=token"
|
|
)
|
|
client._req = Mock()
|
|
client._req.post_res.side_effect = [
|
|
_response({}, status_code=500),
|
|
_response({"errcode": 310000, "errmsg": "keywords not in content"}),
|
|
]
|
|
|
|
assert client.send_msg(title="测试") is False
|
|
assert client.send_msg(title="测试") is False
|
|
|
|
|
|
def test_module_routes_matching_notifications_to_dingtalk_client(monkeypatch) -> None:
|
|
"""模块应沿通知来源和消息类型契约向对应钉钉实例发送。"""
|
|
module = DingTalkModule()
|
|
module._channel = NotificationChannel.DingTalk
|
|
client = Mock()
|
|
config = SimpleNamespace(name="家庭群", switchs=[MessageType.Organize.value])
|
|
monkeypatch.setattr(module, "get_configs", lambda: {config.name: config})
|
|
monkeypatch.setattr(module, "get_config", lambda name=None: config)
|
|
monkeypatch.setattr(module, "get_instance", lambda name=None: client)
|
|
|
|
module.post_message(
|
|
Message(
|
|
channel=NotificationChannel.DingTalk,
|
|
source="家庭群",
|
|
mtype=MessageType.Organize,
|
|
title="整理完成",
|
|
text="测试内容",
|
|
)
|
|
)
|
|
|
|
client.send_msg.assert_called_once()
|
|
|
|
|
|
def test_dingtalk_channel_declares_markdown_image_and_link_capabilities() -> None:
|
|
"""能力表应允许通用消息层保留钉钉支持的富文本字段。"""
|
|
for capability in (
|
|
ChannelCapability.MARKDOWN,
|
|
ChannelCapability.IMAGES,
|
|
ChannelCapability.LINKS,
|
|
):
|
|
assert ChannelCapabilityManager.supports_capability(
|
|
NotificationChannel.DingTalk, capability
|
|
)
|