a11904b7c5
- 为 pam_deploy_graph 生产代码补充中文模块、类、函数/方法文档字符串 - 将原有英文说明和主要英文异常提示改为中文 - 新增当前整体逻辑结构流程图文档，覆盖模块结构、执行链路、action 路由、人工确认和 checkpoint 续跑 - 新增 Linux 自带运行环境打包脚本，使用 PyInstaller 生成解压即用目录和 tar.gz - 新增 Linux 打包说明，包含构建命令、运行方式、依赖说明和包大小评估 - 同步 README，补充流程图、打包方式、产物路径和大小预估 - 更新相关测试断言以匹配中文错误提示
71 lines
3.0 KiB
Python
71 lines
3.0 KiB
Python
"""供本地测试和预演使用的 fake action runner。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any
|
||
|
||
from .models import ActionResult
|
||
|
||
|
||
class FakeActionRunner:
|
||
"""返回确定性 action 结果，避免测试触碰真实 PAM 环境。"""
|
||
|
||
def __init__(self, fixtures: dict[str, dict[str, Any]] | None = None) -> None:
|
||
"""保存可覆盖默认行为的测试 fixture，并记录调用历史。"""
|
||
self.fixtures = fixtures or {}
|
||
self.calls: list[tuple[str, dict[str, Any]]] = []
|
||
|
||
def run(self, action: str, *, params: dict[str, Any], **kwargs: Any) -> ActionResult:
|
||
"""执行 fake action，优先使用 fixture，否则使用内置默认结果。"""
|
||
self.calls.append((action, kwargs))
|
||
values = self._fixture_for(action, kwargs)
|
||
if not values:
|
||
values = self._default_values(action, kwargs)
|
||
ok = not values.pop("_fail", False)
|
||
return ActionResult(
|
||
action=action,
|
||
backend="fake",
|
||
tool_name=f"fake:{action}",
|
||
ok=ok,
|
||
values=values,
|
||
exit_code=0 if ok else 1,
|
||
raw_output=str(values),
|
||
error_summary="" if ok else str(values.get("MESSAGE", "fake action 执行失败")),
|
||
)
|
||
|
||
def _default_values(self, action: str, kwargs: dict[str, Any]) -> dict[str, Any]:
|
||
"""为常见部署 action 构造稳定的默认返回值。"""
|
||
if action == "get-token":
|
||
return {"ACTION": action, "TOKEN": "***"}
|
||
if action == "upload-package":
|
||
return {"ACTION": action, "HASH_CODE": "fake-hash"}
|
||
if action == "get-node-url":
|
||
return {"ACTION": action, "NODE_URL": "https://fake-node.local"}
|
||
if action == "get-online-ips":
|
||
return {"ACTION": action, "COUNT": "2", "IP": ["192.168.1.10", "192.168.1.11"]}
|
||
if action == "upgrade-ip":
|
||
return {"ACTION": action, "IP": kwargs.get("ip", ""), "RESULT": "TASK_CREATED"}
|
||
if action == "poll-upgrade-progress":
|
||
return {
|
||
"ACTION": action,
|
||
"IP": kwargs.get("ip", ""),
|
||
"STEP": "DONE",
|
||
"RATE_OF_PROGRESS": "100",
|
||
"MESSAGE": "success",
|
||
}
|
||
if action == "start-ip":
|
||
return {"ACTION": action, "IP": kwargs.get("ip", ""), "RESULT": "OK"}
|
||
if action == "verify-ip":
|
||
return {"ACTION": action, "IP": kwargs.get("ip", ""), "SUCCESS": "true", "MESSAGE": "ok"}
|
||
if action == "download-log":
|
||
return {"ACTION": action, "IP": kwargs.get("ip", ""), "LOG_FILE": "logs/fake.zip"}
|
||
return {"ACTION": action, "RESULT": "OK"}
|
||
|
||
def _fixture_for(self, action: str, kwargs: dict[str, Any]) -> dict[str, Any]:
|
||
"""按 action 或 action:ip 查找测试 fixture。"""
|
||
ip = kwargs.get("ip")
|
||
ip_key = f"{action}:{ip}" if ip else ""
|
||
if ip_key and ip_key in self.fixtures:
|
||
return self.fixtures[ip_key].copy()
|
||
return self.fixtures.get(action, {}).copy()
|