ForcePilot/backend/test/unit/channels/plugins/feishu/conftest.py

66 lines
2.5 KiB
Python
Raw Normal View History

"""channels plugins/feishu 层单元测试共享 fixture。
提供 ``fake_feishu_client`` 模拟 ``FeishuClient``
``yuxi.channels.channels.plugins.feishu.feishu_client.FeishuClient``
飞书适配器测试通过此桩注入客户端依赖不发起真实 HTTP 请求
桩使用 ``AsyncMock()``预设主要方法返回值 dict / bytes / token
元组调用方可按需覆盖方法签名以 ``FeishuClient`` 实现为准单文件
独有的 helper 应保留在对应测试文件内 testing-guidelines.md
"""
from __future__ import annotations
from unittest.mock import AsyncMock
import pytest
@pytest.fixture
def fake_feishu_client() -> AsyncMock:
"""FeishuClient 桩。
``AsyncMock()`` 模拟 ``FeishuClient``预设主要方法返回值
- ``get_tenant_access_token`` 返回 ``("fake-token", 7200)``
- ``send_message`` / ``update_message`` / ``recall_message`` /
``get_message`` / ``reply_message`` IM API 返回 ``{}``
- ``download_resource`` 返回 ``b""``
- ``upload_image`` / ``upload_file`` 返回 ``{}``
- 通讯录 / OAuth API 返回 ``{}``
"""
client = AsyncMock()
# Token 管理
client.get_tenant_access_token.return_value = ("fake-token", 7200)
# 通用请求
client.request.return_value = {}
# IM API
client.send_message.return_value = {}
client.update_message.return_value = {}
client.recall_message.return_value = {}
client.get_message.return_value = {}
client.reply_message.return_value = {}
client.batch_get_messages.return_value = {}
client.update_card.return_value = {}
client.streaming_start.return_value = {}
client.streaming_update.return_value = {}
client.streaming_stop.return_value = {}
client.add_reaction.return_value = {}
client.delete_reaction.return_value = {}
client.pin_message.return_value = {}
client.unpin_message.return_value = {}
# 资源上传/下载
client.upload_image.return_value = {}
client.upload_file.return_value = {}
client.download_resource.return_value = b""
# 通讯录 API
client.get_user.return_value = {}
client.search_users.return_value = {}
client.get_group_list.return_value = {}
client.get_group_members.return_value = {}
client.get_group_detail.return_value = {}
# OAuth API
client.get_qr_login_url.return_value = {}
client.poll_qr_login_status.return_value = {}
client.revoke_token.return_value = {}
return client