feat(notification): add DingTalk robot channel

tests/test_dingtalk.py

@@ -0,0 +1,123 @@
+"""钉钉自定义机器人通知渠道测试。"""
+
+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 "![图片](https://example.com/poster.jpg)" 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
+        )