mirror of
https://github.com/headroomlabs-ai/headroom.git
synced 2026-08-27 14:17:10 -04:00
107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib
|
|
import types
|
|
|
|
import pytest
|
|
|
|
import headroom.providers as providers
|
|
from headroom.install.models import ManagedMutation
|
|
from headroom.providers import install_registry
|
|
|
|
|
|
def test_providers_package_resolves_exports_lazily_and_caches_them(monkeypatch) -> None:
|
|
module = importlib.reload(providers)
|
|
sentinel = object()
|
|
import_calls: list[str] = []
|
|
|
|
def fake_import_module(name: str):
|
|
import_calls.append(name)
|
|
return types.SimpleNamespace(OpenAIProvider=sentinel)
|
|
|
|
monkeypatch.setattr(module, "import_module", fake_import_module)
|
|
|
|
assert module.OpenAIProvider is sentinel
|
|
assert module.OpenAIProvider is sentinel
|
|
assert import_calls == ["headroom.providers.openai"]
|
|
assert "OpenAIProvider" in module.__dir__()
|
|
|
|
|
|
def test_providers_package_rejects_missing_and_dunder_path_attributes() -> None:
|
|
module = importlib.reload(providers)
|
|
|
|
with pytest.raises(AttributeError, match="__path__"):
|
|
module.__getattr__("__path__")
|
|
|
|
with pytest.raises(AttributeError, match="does_not_exist"):
|
|
module.__getattribute__("does_not_exist")
|
|
|
|
|
|
def test_install_registry_build_install_target_envs_uses_known_targets_only(monkeypatch) -> None:
|
|
monkeypatch.setattr(
|
|
install_registry,
|
|
"_ENV_BUILDERS",
|
|
{
|
|
"claude": lambda *, port, backend: {"CLAUDE_PORT": f"{port}:{backend}"},
|
|
"cursor": lambda *, port, backend: {"CURSOR_PORT": f"{port}:{backend}"},
|
|
},
|
|
)
|
|
|
|
envs = install_registry.build_install_target_envs(
|
|
port=8787,
|
|
backend="anthropic",
|
|
targets=["claude", "unknown", "cursor"],
|
|
)
|
|
|
|
assert envs == {
|
|
"claude": {"CLAUDE_PORT": "8787:anthropic"},
|
|
"cursor": {"CURSOR_PORT": "8787:anthropic"},
|
|
}
|
|
|
|
|
|
def test_install_registry_apply_provider_scope_mutations_skips_missing_and_none(
|
|
monkeypatch,
|
|
) -> None:
|
|
manifest = types.SimpleNamespace(targets=["claude", "codex", "unknown"])
|
|
codex_mutation = ManagedMutation(target="codex", kind="toml-block")
|
|
|
|
monkeypatch.setattr(
|
|
install_registry,
|
|
"_PROVIDER_SCOPE_HANDLERS",
|
|
{
|
|
"claude": (lambda _manifest: None, lambda mutation, manifest: None),
|
|
"codex": (lambda _manifest: codex_mutation, lambda mutation, manifest: None),
|
|
},
|
|
)
|
|
|
|
mutations = install_registry.apply_provider_scope_mutations(manifest)
|
|
|
|
assert mutations == [codex_mutation]
|
|
|
|
|
|
def test_install_registry_revert_provider_scope_mutation_dispatches_known_targets(
|
|
monkeypatch,
|
|
) -> None:
|
|
manifest = types.SimpleNamespace()
|
|
mutation = ManagedMutation(target="codex", kind="toml-block")
|
|
recorded: list[tuple[ManagedMutation, object]] = []
|
|
|
|
monkeypatch.setattr(
|
|
install_registry,
|
|
"_PROVIDER_SCOPE_HANDLERS",
|
|
{
|
|
"codex": (
|
|
lambda _manifest: None,
|
|
lambda incoming_mutation, incoming_manifest: recorded.append(
|
|
(incoming_mutation, incoming_manifest)
|
|
),
|
|
)
|
|
},
|
|
)
|
|
|
|
install_registry.revert_provider_scope_mutation(manifest, mutation)
|
|
install_registry.revert_provider_scope_mutation(
|
|
manifest, ManagedMutation(target="unknown", kind="noop")
|
|
)
|
|
|
|
assert recorded == [(mutation, manifest)]
|