119 lines
4.0 KiB
Python
119 lines
4.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
企业技能对接外部系统（ERP / TMS / WMS / 船司 / 报关 / 邮件 / 浏览器 / LLM / RPA）时，
|
||
测试侧共用的 **profile / 禁止真实外呼** 工具。
|
||
|
||
- 仅标准库 + jiangchang_skill_core.config；不发起 HTTP；不启动浏览器。
|
||
- 真实 API / RPA 契约测试请放在 ``tests/integration/*.sample``，并显式设置 ``OPENCLAW_TEST_TARGET``。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
from typing import Any, Final
|
||
|
||
from jiangchang_skill_core import config
|
||
|
||
ALLOWED_TEST_TARGETS: Final[frozenset[str]] = frozenset(
|
||
{
|
||
"unit",
|
||
"mock",
|
||
"simulator_api",
|
||
"simulator_rpa",
|
||
"real_api",
|
||
"real_rpa",
|
||
}
|
||
)
|
||
|
||
|
||
def _raw_test_target_env() -> str:
|
||
return (config.get("OPENCLAW_TEST_TARGET") or "").strip()
|
||
|
||
|
||
def get_test_target() -> str:
|
||
"""
|
||
读取 ``OPENCLAW_TEST_TARGET``（经 ``config.get`` 三级合并）。
|
||
|
||
未设置时返回 ``unit``。非法取值抛出 ``ValueError``，便于尽早发现拼写错误。
|
||
"""
|
||
raw = _raw_test_target_env()
|
||
if not raw:
|
||
return "unit"
|
||
if raw not in ALLOWED_TEST_TARGETS:
|
||
raise ValueError(
|
||
f"Invalid test target {raw!r}. Allowed: {sorted(ALLOWED_TEST_TARGETS)}"
|
||
)
|
||
return raw
|
||
|
||
|
||
def real_access_allowed(profile: str) -> bool:
|
||
"""
|
||
判断当前 ``OPENCLAW_TEST_TARGET`` 是否允许运行 ``profile`` 档位测试。
|
||
"""
|
||
target = get_test_target()
|
||
if profile in ("mock", "simulator_api", "simulator_rpa"):
|
||
return True
|
||
if profile == "real_api":
|
||
return target == "real_api"
|
||
if profile == "real_rpa":
|
||
return target == "real_rpa"
|
||
raise ValueError(f"unknown profile for real_access_allowed: {profile!r}")
|
||
|
||
|
||
def should_skip_profile(profile: str, write: bool = False) -> tuple[bool, str]:
|
||
"""
|
||
返回 ``(should_skip, reason)``。
|
||
|
||
策略摘要：
|
||
- ``mock``：永不跳过（内存 / FakeAdapter，无外呼）。
|
||
- ``simulator_api`` / ``simulator_rpa``：仅当 ``get_test_target()`` 与 profile 一致时不跳过。
|
||
- ``real_api`` / ``real_rpa``：仅当 ``get_test_target()`` 与 profile 一致时不跳过。
|
||
- ``write`` 参数保留以兼容 integration 样例签名；档位一致即允许写操作测试。
|
||
"""
|
||
_ = write
|
||
target = get_test_target()
|
||
|
||
if profile == "mock":
|
||
return (False, "")
|
||
|
||
if profile in ("simulator_api", "simulator_rpa", "real_api", "real_rpa"):
|
||
if target != profile:
|
||
return (True, f"OPENCLAW_TEST_TARGET must be {profile} (got {target!r})")
|
||
return (False, "")
|
||
|
||
raise ValueError(f"unknown profile: {profile!r}")
|
||
|
||
|
||
class FakeAdapter:
|
||
"""
|
||
模拟外部系统 adapter，用于单元测试中的超时 / 鉴权 / 畸形响应等分支。
|
||
|
||
``mode``：``success`` | ``timeout`` | ``invalid_response`` | ``unauthorized``
|
||
"""
|
||
|
||
def __init__(self, mode: str = "success") -> None:
|
||
if mode not in ("success", "timeout", "invalid_response", "unauthorized"):
|
||
raise ValueError(f"unsupported FakeAdapter mode: {mode!r}")
|
||
self._mode = mode
|
||
|
||
def call(self, payload: Any) -> Any:
|
||
if self._mode == "success":
|
||
return {"success": True, "data": {"echo": payload}}
|
||
if self._mode == "timeout":
|
||
raise TimeoutError("simulated upstream timeout")
|
||
if self._mode == "invalid_response":
|
||
return object() # 非 dict，模拟无法解析的响应体
|
||
if self._mode == "unauthorized":
|
||
raise PermissionError("simulated 401/403")
|
||
raise RuntimeError("unreachable")
|
||
|
||
|
||
def assert_no_real_network_env() -> None:
|
||
"""
|
||
断言默认必跑套件未误设真实外联档位（``real_api`` / ``real_rpa``）。
|
||
|
||
若需在受控环境跑真实集成，请显式设置 ``OPENCLAW_TEST_TARGET`` 并使用 ``tests/integration/*.sample``。
|
||
"""
|
||
target = get_test_target()
|
||
assert target not in ("real_api", "real_rpa"), (
|
||
f"OPENCLAW_TEST_TARGET must not be {target!r} in default unit tests"
|
||
)
|