modify amqtt broker启动问题

This commit is contained in:
2026-08-10 09:33:01 +08:00
parent 7f31591b22
commit 58a889dc6b
3 changed files with 120 additions and 2 deletions
+109
View File
@@ -0,0 +1,109 @@
"""验证 example 插件 global_var.py / __init__.py 改造效果"""
import sys
from pathlib import Path
SRC = Path(__file__).parent / "src"
sys.path.insert(0, str(SRC))
failures = []
def check(name, cond, detail=""):
status = "PASS" if cond else "FAIL"
print(f"[{status}] {name}" + (f" — {detail}" if detail else ""))
if not cond:
failures.append(name)
# ========== 1. global_var 白名单 / require / __all__ ==========
from plugins.example import global_var as gv
check("global_var.__all__ 包含 db", "db" in gv.__all__)
check("global_var.__all__ 包含 set_global_var", "set_global_var" in gv.__all__)
check("global_var.__all__ 包含 require", "require" in gv.__all__)
check("global_var 不再污染模型命名空间", not hasattr(gv, "AddGetRequest"))
# 拼错变量名应抛 KeyError(改造前会静默创建新变量)
try:
gv.set_global_var("db2", object())
check("set_global_var 未知变量应抛 KeyError", False, "未抛错,静默创建了 db2")
except KeyError:
check("set_global_var 未知变量应抛 KeyError", True, "db2 被拒绝")
# require 在未初始化时应抛清晰错误
try:
gv.require("db")
check("require 未初始化应抛 RuntimeError", False, "未抛错")
except RuntimeError:
check("require 未初始化应抛 RuntimeError", True)
# set 后 require 应返回同一个对象
marker = object()
gv.set_global_var("db", marker)
check("set 后 require 返回同一对象", gv.require("db") is marker)
check("模块属性 db 已被更新", gv.db is marker)
# ========== 2. __init__.py 导入链(router + auto_bind_models)==========
import plugins.example as plug
check("plugins.example 导入成功", True)
check("create_workers 存在", callable(getattr(plug, "create_workers", None)))
check("cleanup 存在", callable(getattr(plug, "cleanup", None)))
check("_initialized 默认 False", plug._initialized is False)
# ========== 3. create_workers 幂等性(mock session/config)==========
import asyncio
# 用桩替代 _create_workers,避免真实 zenoh 会话
calls = []
def _stub_create_workers(session, config, count=1, worker_manager=None):
calls.append((session, config, count, worker_manager))
return ["task-stub"]
plug._create_workers = _stub_create_workers # type: ignore[attr-defined]
async def main():
cfg = {"data_dir": "/tmp/example_data"}
# 第一次调用:应初始化 db
t1 = await plug.create_workers(session="s1", config=cfg, count=2)
check("首次 create_workers 调用桩", t1 == ["task-stub"])
check("_initialized 置 True", plug._initialized is True)
from plugins.example.global_var import db as gv_db
check("global_var.db 已初始化", gv_db is not None)
# 第二次调用:幂等跳过,不重复初始化(桩仍被调用来建 task,但 db 不重建)
t2 = await plug.create_workers(session="s2", config=cfg, count=2)
check("二次 create_workers 幂等返回", t2 == ["task-stub"])
check("桩被调用两次(建 task)", len(calls) == 2)
from plugins.example.global_var import db as gv_db2
check("db 对象未重建", gv_db2 is gv_db)
# cleanup 后复位
await plug.cleanup("worker/example/worker-i-1")
check("cleanup 后 _initialized 复位 False", plug._initialized is False)
from plugins.example.global_var import db as gv_db3
check("cleanup 后 db 为 None", gv_db3 is None)
# cleanup 后再启动:可重新初始化
t3 = await plug.create_workers(session="s3", config=cfg, count=1)
check("cleanup 后可再次 create_workers", t3 == ["task-stub"])
from plugins.example.global_var import db as gv_db4
check("再次初始化后 db 非 None", gv_db4 is not None)
check("再次初始化 db 为新对象", gv_db4 is not gv_db)
await plug.cleanup("worker/example/worker-i-1")
asyncio.run(main())
print("\n" + ("=" * 40))
if failures:
print(f"❌ {len(failures)} 项未通过: {failures}")
sys.exit(1)
print("✅ 全部通过")