mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-27 11:22:03 +00:00
83aa84ae3b
Render the reactive pet pane in the classic CLI (steady redraw, right-aligned) and wire the /pet command to list and switch pets, plus an enable/disable toggle. Backed by hermes_cli/pets.py and the CLI commands mixin, registered in the central command registry. Covered by the CLI pet pane and toggle tests.
104 lines
2.9 KiB
Python
104 lines
2.9 KiB
Python
"""Tests for pet slash-command config helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from agent.pet import store
|
|
from agent.pet.constants import FRAME_H, FRAME_W
|
|
|
|
|
|
@pytest.fixture
|
|
def boba_installed(tmp_path, monkeypatch):
|
|
from PIL import Image
|
|
|
|
home = tmp_path / ".hermes"
|
|
home.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(home))
|
|
|
|
sheet = Image.new("RGBA", (FRAME_W * 8, FRAME_H * 9), (0, 0, 0, 0))
|
|
pet_dir = store.pets_dir() / "boba"
|
|
pet_dir.mkdir(parents=True, exist_ok=True)
|
|
sheet.save(pet_dir / "spritesheet.webp")
|
|
(pet_dir / "pet.json").write_text(
|
|
'{"id":"boba","displayName":"Boba","description":"d","spritesheetPath":"spritesheet.webp"}'
|
|
)
|
|
return home
|
|
|
|
|
|
def _write_config(home, *, enabled: bool, slug: str = "") -> None:
|
|
import yaml
|
|
|
|
cfg = {"display": {"pet": {"enabled": enabled, "slug": slug, "scale": 0.33}}}
|
|
(home / "config.yaml").write_text(yaml.dump(cfg), encoding="utf-8")
|
|
|
|
|
|
def test_toggle_pet_display_turns_off_when_enabled(boba_installed):
|
|
from hermes_cli.pets import _pet_config, toggle_pet_display
|
|
|
|
_write_config(boba_installed, enabled=True, slug="boba")
|
|
|
|
enabled, name, err = toggle_pet_display()
|
|
|
|
assert err is None
|
|
assert enabled is False
|
|
assert name == "Boba"
|
|
assert _pet_config()["enabled"] is False
|
|
|
|
|
|
def test_toggle_pet_display_turns_on_resolved_pet(boba_installed):
|
|
from hermes_cli.pets import _pet_config, toggle_pet_display
|
|
|
|
_write_config(boba_installed, enabled=False, slug="boba")
|
|
|
|
enabled, name, err = toggle_pet_display()
|
|
|
|
assert err is None
|
|
assert enabled is True
|
|
assert name == "Boba"
|
|
assert _pet_config()["enabled"] is True
|
|
|
|
|
|
def test_toggle_pet_display_errors_with_no_installed_pets(tmp_path, monkeypatch):
|
|
from hermes_cli.pets import toggle_pet_display
|
|
|
|
home = tmp_path / ".hermes"
|
|
home.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(home))
|
|
_write_config(home, enabled=False, slug="")
|
|
|
|
enabled, name, err = toggle_pet_display()
|
|
|
|
assert enabled is False
|
|
assert name is None
|
|
assert err is not None
|
|
|
|
|
|
@pytest.fixture
|
|
def empty_home(tmp_path, monkeypatch):
|
|
home = tmp_path / ".hermes"
|
|
home.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(home))
|
|
return home
|
|
|
|
|
|
def test_set_pet_scale_writes_clamped_value(empty_home):
|
|
from agent.pet.constants import MAX_SCALE, MIN_SCALE
|
|
from hermes_cli.pets import _pet_config, set_pet_scale
|
|
|
|
applied, err = set_pet_scale("0.5")
|
|
assert err is None
|
|
assert applied == 0.5
|
|
assert _pet_config()["scale"] == 0.5
|
|
|
|
# Out-of-range values clamp to the bounds rather than erroring.
|
|
assert set_pet_scale(99) == (MAX_SCALE, None)
|
|
assert set_pet_scale(0) == (MIN_SCALE, None)
|
|
|
|
|
|
def test_set_pet_scale_rejects_non_numbers(empty_home):
|
|
from hermes_cli.pets import set_pet_scale
|
|
|
|
applied, err = set_pet_scale("huge")
|
|
assert applied == 0.0
|
|
assert err is not None
|