284dbaf537
Activation reaches plugin discovery before the application dependencies exist. Give PM its own locked Python project and runtime so it can install or repair the application without importing that dependency tree. Keep PM outside the application workspace. A shared uv workspace resolves the application graph and cannot provide this isolation. Route mutations through an isolated worker and preserve transaction callbacks, cancellation, custom package registrations, and correlated receipts. Use the same runtime builder for source installs and packaged payloads. Keep offline wheelhouse support in that builder. Nix builds the independent PM lock as a separate derivation. Refuse lazy-disabled bootstrap before installing tools or dependencies. Move first-party YAML readers and writers to ruamel. Keep the application lock's transitive PyYAML requirements for third-party packages. Verification: - Focused canonical Python suite: 177 passed, 1 host-gated skip. - Electron backend probes: 12 passed. Electron typecheck passed. - Both uv locks, scoped lint, Bash syntax, and whitespace checks passed. - Cold activation, corrupt-app repair, offline staging, and relocation ran. - Built and exercised the Nix PM runtime and standalone YAML merge script. Six broader caller test files retain the same 24 failing test IDs as an archive of HEAD. The existing real-home guard blocks those tests before they can exercise the affected paths. No full-suite pass is claimed. Native Windows signing and full Bionic package execution remain unverified.
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
"""Tests for agent/image_gen_registry.py — provider registration & active lookup."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from agent import image_gen_registry
|
|
from agent.image_gen_provider import ImageGenProvider
|
|
|
|
|
|
class _FakeProvider(ImageGenProvider):
|
|
def __init__(self, name: str, available: bool = True):
|
|
self._name = name
|
|
self._available = available
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return self._name
|
|
|
|
def is_available(self) -> bool:
|
|
return self._available
|
|
|
|
def generate(self, prompt, aspect_ratio="landscape", **kw):
|
|
return {"success": True, "image": f"{self._name}://{prompt}"}
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _reset_registry():
|
|
image_gen_registry._reset_for_tests()
|
|
yield
|
|
image_gen_registry._reset_for_tests()
|
|
|
|
|
|
class TestRegisterProvider:
|
|
|
|
|
|
def test_rejects_empty_name(self):
|
|
class Empty(ImageGenProvider):
|
|
@property
|
|
def name(self) -> str:
|
|
return ""
|
|
|
|
def generate(self, prompt, aspect_ratio="landscape", **kw):
|
|
return {}
|
|
|
|
with pytest.raises(ValueError):
|
|
image_gen_registry.register_provider(Empty())
|
|
|
|
|
|
def test_list_is_sorted(self):
|
|
image_gen_registry.register_provider(_FakeProvider("zeta"))
|
|
image_gen_registry.register_provider(_FakeProvider("alpha"))
|
|
names = [p.name for p in image_gen_registry.list_providers()]
|
|
assert names == ["alpha", "zeta"]
|
|
|
|
|
|
class TestGetActiveProvider:
|
|
|
|
|
|
def test_explicit_config_wins(self, tmp_path, monkeypatch):
|
|
import hermes_yaml as yaml
|
|
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
(tmp_path / "config.yaml").write_text(
|
|
yaml.safe_dump({"image_gen": {"provider": "openai"}})
|
|
)
|
|
image_gen_registry.register_provider(_FakeProvider("fal"))
|
|
image_gen_registry.register_provider(_FakeProvider("openai"))
|
|
active = image_gen_registry.get_active_provider()
|
|
assert active is not None and active.name == "openai"
|
|
|
|
|
|
def test_none_when_empty(self, tmp_path, monkeypatch):
|
|
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
|
assert image_gen_registry.get_active_provider() is None
|