mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-03 17:27:37 +08:00
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
|
"""Shared helper for loading platform-plugin ``adapter.py`` modules in tests.
|
||
|
|
|
||
|
|
Every platform plugin under ``plugins/platforms/<name>/`` ships its own
|
||
|
|
``adapter.py``. If two tests independently do::
|
||
|
|
|
||
|
|
sys.path.insert(0, "plugins/platforms/irc")
|
||
|
|
from adapter import IRCAdapter
|
||
|
|
|
||
|
|
sys.path.insert(0, "plugins/platforms/teams")
|
||
|
|
from adapter import TeamsAdapter
|
||
|
|
|
||
|
|
…then whichever collects first in an xdist worker wins
|
||
|
|
``sys.modules["adapter"]``, and the other raises ``ImportError`` at
|
||
|
|
collection time. The fallout cascades across unrelated tests sharing that
|
||
|
|
worker because ``sys.path`` is still polluted.
|
||
|
|
|
||
|
|
Use :func:`load_plugin_adapter` instead of ad-hoc ``sys.path`` tricks.
|
||
|
|
It loads the adapter from an explicit file path under a unique module
|
||
|
|
name (``plugin_adapter_<plugin_name>``), so it cannot collide with any
|
||
|
|
other plugin's adapter module.
|
||
|
|
|
||
|
|
The ``tests/gateway/conftest.py`` guard rejects the anti-pattern at
|
||
|
|
collection time so this can't regress when new plugin adapter tests are
|
||
|
|
added.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import importlib.util
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
from types import ModuleType
|
||
|
|
|
||
|
|
|
||
|
|
_REPO_ROOT = Path(__file__).resolve().parents[2]
|
||
|
|
_PLUGINS_DIR = _REPO_ROOT / "plugins" / "platforms"
|
||
|
|
|
||
|
|
|
||
|
|
def load_plugin_adapter(plugin_name: str) -> ModuleType:
|
||
|
|
"""Import ``plugins/platforms/<plugin_name>/adapter.py`` in isolation.
|
||
|
|
|
||
|
|
The module is registered under the unique name
|
||
|
|
``plugin_adapter_<plugin_name>`` in ``sys.modules``. No ``sys.path``
|
||
|
|
mutation. Safe to call multiple times — repeat calls return the
|
||
|
|
already-loaded module.
|
||
|
|
"""
|
||
|
|
module_name = f"plugin_adapter_{plugin_name}"
|
||
|
|
cached = sys.modules.get(module_name)
|
||
|
|
if cached is not None:
|
||
|
|
return cached
|
||
|
|
|
||
|
|
adapter_path = _PLUGINS_DIR / plugin_name / "adapter.py"
|
||
|
|
if not adapter_path.is_file():
|
||
|
|
raise FileNotFoundError(
|
||
|
|
f"Plugin adapter not found: {adapter_path}. "
|
||
|
|
f"Known plugins: {sorted(p.name for p in _PLUGINS_DIR.iterdir() if p.is_dir())}"
|
||
|
|
)
|
||
|
|
|
||
|
|
spec = importlib.util.spec_from_file_location(module_name, adapter_path)
|
||
|
|
if spec is None or spec.loader is None:
|
||
|
|
raise ImportError(f"Could not build import spec for {adapter_path}")
|
||
|
|
|
||
|
|
module = importlib.util.module_from_spec(spec)
|
||
|
|
# Register BEFORE exec so the module can find itself if needed (some
|
||
|
|
# modules do ``sys.modules[__name__]`` reflection during import).
|
||
|
|
sys.modules[module_name] = module
|
||
|
|
try:
|
||
|
|
spec.loader.exec_module(module)
|
||
|
|
except Exception:
|
||
|
|
sys.modules.pop(module_name, None)
|
||
|
|
raise
|
||
|
|
return module
|