Something went wrong. Try again.
The agent that grows with you
Something went wrong. Try again.
3.7 kB · 109 lines
Python
at bb/theme
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110from pathlib import Pathfrom unittest.mock import patch
from cli import ( HermesCLI, _collect_query_images, _format_image_attachment_badges, _termux_example_image_path,)
def _make_cli(): cli_obj = HermesCLI.__new__(HermesCLI) cli_obj._attached_images = [] return cli_obj
def _make_image(path: Path) -> Path: path.parent.mkdir(parents=True, exist_ok=True) path.write_bytes(b"\x89PNG\r\n\x1a\n") return path
class TestImageCommand: def test_handle_image_command_attaches_local_image(self, tmp_path): img = _make_image(tmp_path / "photo.png") cli_obj = _make_cli()
with patch("cli._cprint"): cli_obj._handle_image_command(f"/image {img}")
assert cli_obj._attached_images == [img]
def test_handle_image_command_supports_quoted_path_with_spaces(self, tmp_path): img = _make_image(tmp_path / "my photo.png") cli_obj = _make_cli()
with patch("cli._cprint"): cli_obj._handle_image_command(f'/image "{img}"')
assert cli_obj._attached_images == [img]
def test_handle_image_command_rejects_non_image_file(self, tmp_path): file_path = tmp_path / "notes.txt" file_path.write_text("hello\n", encoding="utf-8") cli_obj = _make_cli()
with patch("cli._cprint") as mock_print: cli_obj._handle_image_command(f"/image {file_path}")
assert cli_obj._attached_images == [] rendered = " ".join(str(arg) for call in mock_print.call_args_list for arg in call.args) assert "Not a supported image file" in rendered
class TestCollectQueryImages: def test_collect_query_images_accepts_explicit_image_arg(self, tmp_path): img = _make_image(tmp_path / "diagram.png")
message, images = _collect_query_images("describe this", str(img))
assert message == "describe this" assert images == [img]
def test_collect_query_images_extracts_leading_path(self, tmp_path): img = _make_image(tmp_path / "camera.png")
message, images = _collect_query_images(f"{img} what do you see?")
assert message == "what do you see?" assert images == [img]
def test_collect_query_images_supports_tilde_paths(self, tmp_path, monkeypatch): home = tmp_path / "home" img = _make_image(home / "storage" / "shared" / "Pictures" / "cat.png") monkeypatch.setenv("HOME", str(home))
message, images = _collect_query_images("describe this", "~/storage/shared/Pictures/cat.png")
assert message == "describe this" assert images == [img]
class TestTermuxImageHints: def test_termux_example_image_path_prefers_real_shared_storage_root(self, monkeypatch): existing = {"/sdcard", "/storage/emulated/0"} monkeypatch.setattr("cli.os.path.isdir", lambda path: path in existing)
hint = _termux_example_image_path()
assert hint == "/sdcard/Pictures/cat.png"
class TestImageBadgeFormatting: def test_compact_badges_use_filename_on_narrow_terminals(self, tmp_path): img = _make_image(tmp_path / "Screenshot 2026-04-09 at 11.22.33 AM.png")
badges = _format_image_attachment_badges([img], image_counter=1, width=40)
assert badges.startswith("[📎 ") assert "Image #1" not in badges
def test_compact_badges_summarize_multiple_images(self, tmp_path): img1 = _make_image(tmp_path / "one.png") img2 = _make_image(tmp_path / "two.png")
badges = _format_image_attachment_badges([img1, img2], image_counter=2, width=45)
assert badges == "[📎 2 images attached]"