modify: 测试

test/verify_example_plugin.py

@@ -0,0 +1,110 @@
+"""验证 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  # noqa: E402
+
+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 + Plugin 基类）==========
+import plugins.example as plug  # noqa: E402
+
+check("plugins.example 导入成功", True)
+check("create_workers 存在", callable(getattr(plug, "create_workers", None)))
+check("cleanup 存在", callable(getattr(plug, "cleanup", None)))
+check("plugin 实例存在", getattr(plug, "plugin", None) is not None)
+check("plugin._initialized 默认 False", plug.plugin._initialized is False)
+
+# ========== 3. create_workers 幂等性（mock session/config）==========
+import asyncio  # noqa: E402
+
+# 用桩替代 Plugin 基类的 worker 创建实现，避免真实 zenoh 会话
+calls = []
+
+
+def _stub_create_workers(session, config, count=1, worker_manager=None):
+    calls.append((session, config, count, worker_manager))
+    return ["task-stub"]
+
+
+plug.plugin._create_workers_impl = _stub_create_workers  # type: ignore[method-assign]
+
+
+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("plugin._initialized 置 True", plug.plugin._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 后 plugin._initialized 复位 False", plug.plugin._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("✅ 全部通过")