87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
"""Shared fixtures for bedrock plugin tests.
|
|
|
|
Registers the bedrock plugin in the singleton registry before each test.
|
|
"""
|
|
import pytest
|
|
|
|
|
|
class _FullCtx:
|
|
"""Plugin context that wires up all registry hooks."""
|
|
|
|
def register_provider_services(self, name, services):
|
|
from agent.plugin_registries import registries
|
|
registries.register_provider_services(name, services)
|
|
|
|
def register_provider_resolver(self, name, resolver):
|
|
from agent.plugin_registries import registries
|
|
registries.register_provider_resolver(name, resolver)
|
|
|
|
def register_credential_pool_hook(self, name, hook):
|
|
from agent.plugin_registries import registries
|
|
registries.register_credential_pool_hook(name, hook)
|
|
|
|
def register_transport(self, api_mode, transport_cls):
|
|
from agent.plugin_registries import registries
|
|
registries._transports[api_mode] = transport_cls
|
|
|
|
def register_pricing_provider(self, name, entries):
|
|
from agent.plugin_registries import registries
|
|
registries.register_pricing_provider(name, entries)
|
|
|
|
def register_provider_overlay(self, entry):
|
|
from agent.plugin_registries import registries
|
|
registries.register_provider_overlay(entry)
|
|
|
|
def __getattr__(self, name):
|
|
if name.startswith("register_"):
|
|
return lambda *a, **kw: None
|
|
raise AttributeError(name)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _register_bedrock_plugin():
|
|
"""Register the real bedrock plugin for the duration of each test."""
|
|
from agent.plugin_registries import registries
|
|
from hermes_cli import providers as _prov
|
|
|
|
_prev_services = dict(registries._provider_services)
|
|
_prev_resolvers = dict(registries._provider_resolvers)
|
|
_prev_cph = dict(registries._credential_pool_hooks)
|
|
_prev_overlays = dict(registries._provider_overlays)
|
|
_prev_hermes_overlays = dict(_prov.HERMES_OVERLAYS)
|
|
_prev_aliases = dict(_prov.ALIASES)
|
|
_prev_merged = _prov._plugin_overlays_merged
|
|
|
|
ctx = _FullCtx()
|
|
try:
|
|
from hermes_agent_bedrock import register as _reg
|
|
_reg(ctx)
|
|
except ImportError:
|
|
pass
|
|
try:
|
|
from hermes_agent_anthropic import register as _ant_reg
|
|
_ant_reg(ctx)
|
|
except ImportError:
|
|
pass
|
|
|
|
# Force a re-merge so plugin-registered overlays and aliases
|
|
# appear in HERMES_OVERLAYS / ALIASES for the test.
|
|
_prov._plugin_overlays_merged = False
|
|
_prov._merge_plugin_overlays()
|
|
|
|
yield
|
|
|
|
for d, prev in [
|
|
(registries._provider_services, _prev_services),
|
|
(registries._provider_resolvers, _prev_resolvers),
|
|
(registries._credential_pool_hooks, _prev_cph),
|
|
(registries._provider_overlays, _prev_overlays),
|
|
]:
|
|
d.clear()
|
|
d.update(prev)
|
|
_prov.HERMES_OVERLAYS.clear()
|
|
_prov.HERMES_OVERLAYS.update(_prev_hermes_overlays)
|
|
_prov.ALIASES.clear()
|
|
_prov.ALIASES.update(_prev_aliases)
|
|
_prov._plugin_overlays_merged = _prev_merged
|