hermes-agent/tests/agent/test_compressor_media_stripping.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

67 lines
2.2 KiB
Python

"""Tests for MEDIA directive stripping in context compaction (#14665).
MEDIA directives in assistant messages must not leak into compaction
summaries — if they do, the downstream model re-emits them as active
directives on the next turn.
"""
import pytest
from unittest.mock import patch
from agent.context_compressor import ContextCompressor
@pytest.fixture()
def compressor():
with patch("agent.context_compressor.get_model_context_length", return_value=100000):
return ContextCompressor(
model="test/model",
threshold_percent=0.85,
protect_first_n=2,
protect_last_n=2,
quiet_mode=True,
)
class TestMediaDirectiveStripping:
"""MEDIA directives must be stripped before summarization (#14665)."""
def test_non_media_content_preserved(self, compressor):
turns = [
{"role": "assistant", "content": "The file path is /tmp/test.txt and it works."},
]
result = compressor._serialize_for_summary(turns)
assert "/tmp/test.txt" in result
def test_multiple_media_directives(self, compressor):
turns = [
{"role": "assistant", "content": "MEDIA:/a.ogg and MEDIA:/b.mp3"},
]
result = compressor._serialize_for_summary(turns)
assert "MEDIA:" not in result
assert result.count("[media attachment]") == 2
def test_multimodal_list_content_does_not_crash(self, compressor):
"""content as a list (multimodal) must be flattened to clean text.
Without flattening, str() coercion in redact_sensitive_text dumps
the raw part-dict repr — including base64 image data — into the
summarizer input, burning summary budget on noise.
"""
turns = [
{
"role": "user",
"content": [
{"type": "text", "text": "What is in this image?"},
{"type": "image_url", "image_url": {"url": "data:image/png;base64,abc123"}},
],
},
]
result = compressor._serialize_for_summary(turns)
assert "What is in this image?" in result
assert "[image]" in result
assert "base64" not in result