68 lines
2.7 KiB
Python
68 lines
2.7 KiB
Python
"""Pytest 全局 fixture：把 actor 句柄、PostgresDatabase、loguru 等重副作用替换成可控 stub。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import types
|
|
from typing import Any, Dict
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# actor 注册表存根：测试期把名字 -> mock 实例塞进进程内 actor 运行时。
|
|
# 生产代码 ``get_actor(name).method(args)`` 会拿到 LocalActorHandle 转发到 mock。
|
|
# mock 方法用 AsyncMock 即可（handle 统一返回 awaitable）。
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
|
|
class _FakeActorRegistry:
|
|
"""薄封装：``register(name, instance)`` 直接登记到真实 actor 运行时。"""
|
|
|
|
def register(self, name: str, instance: Any) -> None:
|
|
from kilostar.utils import actor
|
|
|
|
actor.register_actor(name, instance)
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_actors(monkeypatch) -> _FakeActorRegistry:
|
|
"""清空并接管进程内 actor 运行时，测试结束后复位。
|
|
|
|
用法::
|
|
|
|
def test_xxx(fake_actors):
|
|
gsm = MagicMock()
|
|
gsm.get_tool_config = AsyncMock(return_value={"api_key": "k"})
|
|
fake_actors.register("global_state_machine", gsm)
|
|
"""
|
|
from kilostar.utils import actor
|
|
|
|
actor.clear_actors()
|
|
yield _FakeActorRegistry()
|
|
actor.clear_actors()
|
|
|
|
|
|
@pytest.fixture
|
|
def gsm_handle(fake_actors) -> MagicMock:
|
|
"""快捷 fixture：注册名为 ``global_state_machine`` 的 mock 实例并返回它。
|
|
|
|
方法默认为 ``AsyncMock``；生产侧 ``await get_actor(...).method(args)`` 正常工作。
|
|
"""
|
|
gsm = MagicMock()
|
|
fake_actors.register("global_state_machine", gsm)
|
|
return gsm
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# 工具：构造一个 ``MCP_AVAILABLE`` 关闭的 mcp_helper 状态
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
|
|
@pytest.fixture
|
|
def mcp_unavailable(monkeypatch):
|
|
from kilostar.utils import mcp_helper
|
|
|
|
monkeypatch.setattr(mcp_helper, "_MCP_AVAILABLE", False)
|
|
yield
|