hermes-agent/tests/tui_gateway/test_pet_generate_rpc.py
Teknium 6b81590c55
test: prune low-value tests suite-wide (wave 1) — 46,820 → 28,106 test functions
Systematic prune per AGENTS.md test policy, one pass over every major
test tree (gateway, hermes_cli, tools, agent, run_agent, plugins, cli,
cron, tui_gateway, honcho/openviking, root-level):

- DELETE: source-reading tests (read_text/getsource on prod files),
  change-detector tests (exact catalog counts, model-name snapshots,
  config version literals), mock-echo tests (assert a mock returns what
  it was told), assertion-free/trivial tests, near-duplicate
  parametrizations (boundaries + one representative kept), async/sync
  twin duplicates, cosmetic within-file variations.
- KEEP (mandatory): security/redaction/approval guards, message-role
  alternation invariants, prompt-caching/deterministic-call-id
  invariants, issue-number regression tests (deduped), E2E tests.
- 6 test files deleted outright (script-style/no-assert or fully
  redundant); conftest.py, fakes/, fixtures/ untouched.
- tests/acp/conftest.py added: autouse fixture stubs the live
  models.dev/GitHub/Copilot/Anthropic inventory fetches that ACP server
  tests performed on every session create — test_server.py 147s → 3.4s,
  and the tests are now genuinely hermetic.
- Sleep-based slowness shrunk where safe (codex_ttfb_watchdog,
  compression_concurrent_fork, etc.); no wall-clock assertion tightened.

Verification: full hermetic suite via scripts/run_tests.sh —
2439 files, 31,130 tests passed, 0 failed, 0 flaky retries, 315s wall
(baseline: 583s wall, 13,564s subprocess CPU).
2026-07-29 13:10:23 -07:00

64 lines
1.9 KiB
Python

"""Gateway RPC tests for pet generation (pet.generate / pet.hatch).
Image generation is mocked, so these assert the RPC contract + staging behavior
(draft tokens, data-URI previews, expiry, activation) without any API calls.
"""
from __future__ import annotations
import pytest
pytest.importorskip("PIL")
from PIL import Image # noqa: E402
from tui_gateway import server # noqa: E402
def _png(path):
Image.new("RGBA", (64, 64), (200, 80, 80, 255)).save(path)
def test_pet_generate_requires_prompt():
resp = server._methods["pet.generate"]("r1", {"prompt": " "})
assert "error" in resp
def _fake_drafts_factory(tmp_path):
def fake_drafts(prompt, *, n=4, style="auto", reference_images=None, provider=None, on_draft=None, is_cancelled=None):
paths = []
for i in range(n):
p = tmp_path / f"d{i}.png"
_png(p)
paths.append(p)
if on_draft is not None:
on_draft(i, p)
return paths
return fake_drafts
def _fake_hatch_factory(captured):
"""A hatch that registers a real local pet (so the preview payload populates)."""
import agent.pet.generate as gen
from agent.pet import store
def fake_hatch(*, base_image, slug, display_name="", description="", concept="", style="auto", on_progress=None, provider=None, is_cancelled=None):
captured["base_image"] = str(base_image)
captured["slug"] = slug
pet = store.register_local_pet(
Image.new("RGBA", (192, 208), (10, 20, 30, 255)),
slug=slug,
display_name=display_name,
description=description,
)
return gen.HatchResult(
slug=pet.slug,
display_name=display_name or pet.display_name,
spritesheet=pet.spritesheet,
states=["idle", "wave"],
validation={"ok": True, "warnings": ["state 'jump' has no frames"]},
)
return fake_hatch