wehub-resource-sync
152 lines
4.3 KiB
Python
152 lines
4.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
|
|
def test_input_bridge_announces_image_attachment_with_supplied_console(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
from opensquilla.cli.repl import input_bridge
|
|
|
|
captured: dict[str, str] = {}
|
|
prints: list[str] = []
|
|
|
|
class FakeConsole:
|
|
def print(self, message: str) -> None:
|
|
prints.append(message)
|
|
|
|
def fake_image_prompt_and_attachments(
|
|
command: str,
|
|
) -> tuple[str, list[dict[str, str]]]:
|
|
captured["command"] = command
|
|
return (
|
|
"describe it",
|
|
[{"type": "image/png", "data": "x" * 3072, "name": "screen.png"}],
|
|
)
|
|
|
|
monkeypatch.setattr(
|
|
input_bridge._input_assets,
|
|
"image_prompt_and_attachments",
|
|
fake_image_prompt_and_attachments,
|
|
)
|
|
|
|
prompt, attachments = input_bridge.image_prompt_and_attachments(
|
|
"/image screen.png describe it",
|
|
output_console=FakeConsole(),
|
|
)
|
|
|
|
assert captured["command"] == "/image screen.png describe it"
|
|
assert prompt == "describe it"
|
|
assert attachments == [
|
|
{"type": "image/png", "data": "x" * 3072, "name": "screen.png"}
|
|
]
|
|
assert prints == ["[dim]Sending image: screen.png (3KB base64)[/dim]"]
|
|
|
|
|
|
def test_input_bridge_announces_image_attachment_with_default_console(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
from opensquilla.cli.repl import input_bridge
|
|
|
|
prints: list[str] = []
|
|
|
|
class FakeConsole:
|
|
def print(self, message: str) -> None:
|
|
prints.append(message)
|
|
|
|
def fake_image_prompt_and_attachments(
|
|
command: str,
|
|
) -> tuple[str, list[dict[str, str]]]:
|
|
return (
|
|
"describe it",
|
|
[{"type": "image/png", "data": "x" * 2048, "name": "screen.png"}],
|
|
)
|
|
|
|
monkeypatch.setattr(
|
|
input_bridge._input_assets,
|
|
"image_prompt_and_attachments",
|
|
fake_image_prompt_and_attachments,
|
|
)
|
|
monkeypatch.setattr(input_bridge, "console", FakeConsole())
|
|
|
|
prompt, attachments = input_bridge.image_prompt_and_attachments(
|
|
"/image screen.png describe it"
|
|
)
|
|
|
|
assert prompt == "describe it"
|
|
assert attachments == [
|
|
{"type": "image/png", "data": "x" * 2048, "name": "screen.png"}
|
|
]
|
|
assert prints == ["[dim]Sending image: screen.png (2KB base64)[/dim]"]
|
|
|
|
|
|
def test_input_bridge_does_not_announce_when_image_has_no_attachments(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
from opensquilla.cli.repl import input_bridge
|
|
|
|
prints: list[str] = []
|
|
|
|
class FakeConsole:
|
|
def print(self, message: str) -> None:
|
|
prints.append(message)
|
|
|
|
def fake_image_prompt_and_attachments(
|
|
command: str,
|
|
) -> tuple[str, list[dict[str, str]]]:
|
|
return "describe it", []
|
|
|
|
monkeypatch.setattr(
|
|
input_bridge._input_assets,
|
|
"image_prompt_and_attachments",
|
|
fake_image_prompt_and_attachments,
|
|
)
|
|
|
|
prompt, attachments = input_bridge.image_prompt_and_attachments(
|
|
"/image missing.png describe it",
|
|
output_console=FakeConsole(),
|
|
)
|
|
|
|
assert prompt == "describe it"
|
|
assert attachments == []
|
|
assert prints == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_input_bridge_wraps_async_file_prompt_builder(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
from opensquilla.cli.repl import input_bridge
|
|
|
|
captured: dict[str, Any] = {}
|
|
|
|
async def fake_async_file_prompt_and_attachments(
|
|
command: str,
|
|
*,
|
|
upload_callable: Any | None = None,
|
|
) -> tuple[str, list[dict[str, Any]]]:
|
|
captured["command"] = command
|
|
captured["upload_callable"] = upload_callable
|
|
return "summarize", [{"type": "text/plain", "name": "note.txt"}]
|
|
|
|
monkeypatch.setattr(
|
|
input_bridge._input_assets,
|
|
"async_file_prompt_and_attachments",
|
|
fake_async_file_prompt_and_attachments,
|
|
)
|
|
|
|
def fake_upload(*args: Any) -> str:
|
|
return "u-file"
|
|
|
|
prompt, attachments = await input_bridge.async_file_prompt_and_attachments(
|
|
"/file note.txt summarize",
|
|
upload_callable=fake_upload,
|
|
)
|
|
|
|
assert captured["command"] == "/file note.txt summarize"
|
|
assert captured["upload_callable"] is fake_upload
|
|
assert prompt == "summarize"
|
|
assert attachments == [{"type": "text/plain", "name": "note.txt"}]
|