Spaces:
Paused
Paused
| from pathlib import Path | |
| from 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]" | |