122 lines
4.4 KiB
Python
122 lines
4.4 KiB
Python
"""channels plugins/wechat_mp 层单元测试共享 fixture。
|
||
|
||
提供 ``fake_wechat_mp_client`` 桩，模拟 ``WeChatMpClient``
|
||
（``yuxi.channels.plugins.wechat_mp.wechat_mp_client.WeChatMpClient``）。
|
||
微信公众号适配器测试通过此桩注入客户端依赖，不发起真实 HTTP 请求。
|
||
|
||
桩使用 ``AsyncMock()``，预设主要方法返回值（errcode=0 响应 / media_id /
|
||
token / 用户信息等），调用方可按需覆盖。方法签名以 ``WeChatMpClient``
|
||
实现为准。单文件独有的 helper 应保留在对应测试文件内
|
||
（见 testing-guidelines.md）。
|
||
|
||
同时提供 ``fake_config_port`` / ``fake_cache_port`` / ``fake_logger_port`` /
|
||
``fake_persistence_port`` 端口桩，供适配器测试注入 DI 依赖。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from unittest.mock import AsyncMock, MagicMock
|
||
|
||
import pytest
|
||
|
||
|
||
@pytest.fixture
|
||
def fake_wechat_mp_client() -> AsyncMock:
|
||
"""WeChatMpClient 桩。
|
||
|
||
``AsyncMock()`` 模拟 ``WeChatMpClient``，预设主要方法返回值：
|
||
- ``send_custom_message`` 返回 ``{"errcode": 0, "msgid": "msg_123"}``；
|
||
- ``upload_media_cached`` / ``upload_media`` 返回 ``"media_id_abc"``；
|
||
- ``download_media`` 返回 ``b"fake_image_bytes"``；
|
||
- ``get_user_info`` 返回含 openid / nickname / headimgurl 的字典；
|
||
- ``get_user_list`` 返回含 total / count / data / next_openid 的字典；
|
||
- ``get_callback_ip`` 返回 ``{"ip_list": ["127.0.0.1"]}``；
|
||
- ``refresh_access_token`` / ``get_access_token`` 返回 token 字符串；
|
||
- ``check_customer_service_window`` / ``consume_customer_service_quota``
|
||
返回 ``True``；
|
||
- ``mark_user_message`` / ``mark_other_event`` / ``set_active`` /
|
||
``attach_http_client`` / ``detach_http_client`` 返回 ``None``；
|
||
- ``drain`` 返回 ``True``。
|
||
"""
|
||
client = AsyncMock()
|
||
# 客服消息发送
|
||
client.send_custom_message.return_value = {"errcode": 0, "msgid": "msg_123"}
|
||
# 媒体上传/下载
|
||
client.upload_media_cached.return_value = "media_id_abc"
|
||
client.upload_media.return_value = "media_id_abc"
|
||
client.download_media.return_value = b"fake_image_bytes"
|
||
# 用户信息查询
|
||
client.get_user_info.return_value = {
|
||
"openid": "o123",
|
||
"nickname": "TestUser",
|
||
"headimgurl": "http://example.com/avatar.jpg",
|
||
}
|
||
client.get_user_list.return_value = {
|
||
"total": 1,
|
||
"count": 1,
|
||
"data": {"openid": ["o123", "o456"]},
|
||
"next_openid": "o456",
|
||
}
|
||
# 通用 API
|
||
client.get_callback_ip.return_value = {"ip_list": ["127.0.0.1"]}
|
||
# access_token 中控
|
||
client.refresh_access_token.return_value = "fake_access_token"
|
||
client.get_access_token.return_value = "fake_access_token"
|
||
# 客服消息窗口与额度
|
||
client.check_customer_service_window.return_value = True
|
||
client.consume_customer_service_quota.return_value = True
|
||
client.mark_user_message.return_value = None
|
||
client.mark_other_event.return_value = None
|
||
# 生命周期管理
|
||
client.set_active.return_value = None
|
||
client.drain.return_value = True
|
||
client.attach_http_client.return_value = None
|
||
client.detach_http_client.return_value = None
|
||
return client
|
||
|
||
|
||
@pytest.fixture
|
||
def fake_config_port() -> AsyncMock:
|
||
"""ConfigPort 桩。
|
||
|
||
``ConfigPort`` 的 ``get`` 方法为异步，使用 ``AsyncMock()``。
|
||
默认返回 ``None``，调用方可按需覆盖。
|
||
"""
|
||
return AsyncMock()
|
||
|
||
|
||
@pytest.fixture
|
||
def fake_cache_port() -> AsyncMock:
|
||
"""CachePort 桩。
|
||
|
||
``CachePort`` 的 ``get`` / ``set`` / ``delete`` / ``acquireAdvisoryLock``
|
||
等方法为异步，使用 ``AsyncMock()``。默认返回 ``None``，调用方可按需覆盖。
|
||
"""
|
||
return AsyncMock()
|
||
|
||
|
||
@pytest.fixture
|
||
def fake_logger_port() -> MagicMock:
|
||
"""LoggerPort 桩。
|
||
|
||
``LoggerPort`` 的 ``info`` / ``warn`` / ``error`` / ``exception`` /
|
||
``debug`` 方法为异步，使用 ``MagicMock()`` + 按方法挂 ``AsyncMock()``
|
||
以精确模拟同步/异步混合的端口协议。
|
||
"""
|
||
logger = MagicMock()
|
||
logger.debug = AsyncMock()
|
||
logger.info = AsyncMock()
|
||
logger.warning = AsyncMock()
|
||
logger.error = AsyncMock()
|
||
logger.exception = AsyncMock()
|
||
return logger
|
||
|
||
|
||
@pytest.fixture
|
||
def fake_persistence_port() -> AsyncMock:
|
||
"""PersistencePort 桩。
|
||
|
||
``PersistencePort`` 的方法为异步，使用 ``AsyncMock()``。
|
||
"""
|
||
return AsyncMock()
|