80 lines
3.1 KiB
Python
80 lines
3.1 KiB
Python
"""Test-env stubs for workspace-container deps.
|
|
|
|
`claude_agent_sdk` and `a2a` are installed inside the claude-code workspace
|
|
image, not in this package's test environment. Rather than pulling them as
|
|
dev deps (they're heavy + change frequently), we stub them before any test
|
|
module imports `molecule_runtime.*` — the stubs just need to be importable
|
|
so the package source doesn't crash at collect time.
|
|
|
|
Every known submodule + symbol used anywhere in `molecule_runtime/` is
|
|
stubbed here, so test_imports.py (walking every module) + test_session_
|
|
resume_gate.py (ClaudeSDKExecutor methods) + future tests all share one
|
|
consistent stub set.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import types
|
|
|
|
|
|
def _ensure_package(dotted: str) -> types.ModuleType:
|
|
"""Create a package-shaped module at `dotted` and all its parents.
|
|
|
|
Key detail: the __path__ attribute makes Python treat the stub as a
|
|
PACKAGE, so `from parent.child import X` resolves correctly. Without
|
|
__path__ you get `'parent' is not a package` which is the exact bug
|
|
that broke our first-attempt stubs.
|
|
"""
|
|
parts = dotted.split(".")
|
|
for i in range(1, len(parts) + 1):
|
|
name = ".".join(parts[:i])
|
|
if name not in sys.modules:
|
|
mod = types.ModuleType(name)
|
|
mod.__path__ = [] # type: ignore[attr-defined]
|
|
sys.modules[name] = mod
|
|
if i > 1:
|
|
parent = sys.modules[".".join(parts[:i-1])]
|
|
child_name = parts[i-1]
|
|
if not hasattr(parent, child_name):
|
|
setattr(parent, child_name, sys.modules[name])
|
|
return sys.modules[dotted]
|
|
|
|
|
|
def _set_attrs(mod: types.ModuleType, attrs: list[str]) -> None:
|
|
"""Attach placeholder classes/functions. AgentExecutor is a class
|
|
so subclasses can inherit; everything else is a no-op callable."""
|
|
for attr in attrs:
|
|
if hasattr(mod, attr):
|
|
continue
|
|
if attr in ("AgentExecutor",):
|
|
setattr(mod, attr, type(attr, (), {}))
|
|
else:
|
|
setattr(mod, attr, lambda *a, **kw: None)
|
|
|
|
|
|
# claude_agent_sdk — used by claude_sdk_executor.py
|
|
if "claude_agent_sdk" not in sys.modules:
|
|
sdk_stub = types.ModuleType("claude_agent_sdk")
|
|
sdk_stub.ClaudeAgentOptions = lambda **kwargs: types.SimpleNamespace(**kwargs)
|
|
sdk_stub.query = lambda **kwargs: iter([])
|
|
sys.modules["claude_agent_sdk"] = sdk_stub
|
|
|
|
# a2a + submodules — every import path in the package gets stubbed.
|
|
_A2A_MODULES: dict[str, list[str]] = {
|
|
"a2a": [],
|
|
"a2a.server": [],
|
|
"a2a.server.agent_execution": ["AgentExecutor", "RequestContext"],
|
|
"a2a.server.events": ["EventQueue"],
|
|
"a2a.server.tasks": ["TaskUpdater", "InMemoryTaskStore"],
|
|
"a2a.server.apps": ["A2AStarletteApplication"],
|
|
"a2a.server.request_handlers": ["DefaultRequestHandler"],
|
|
"a2a.types": [
|
|
"Part", "TextPart", "AgentCard", "AgentCapabilities", "AgentSkill",
|
|
"TaskStatus", "TaskState", "TaskStatusUpdateEvent",
|
|
],
|
|
"a2a.utils": ["new_agent_text_message"],
|
|
}
|
|
for dotted, attrs in _A2A_MODULES.items():
|
|
mod = _ensure_package(dotted)
|
|
_set_attrs(mod, attrs)
|