wehub-resource-sync
120 lines
4.1 KiB
Python
120 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from opensquilla.cli.chat_cmd import (
|
|
_parse_path_command,
|
|
_path_prompt_and_attachments,
|
|
_path_strategy_hint,
|
|
)
|
|
from opensquilla.cli.repl.commands import REGISTRY
|
|
from opensquilla.engine.commands import DEFAULT_REGISTRY, Surface
|
|
|
|
|
|
def test_path_command_parses_quoted_path_with_prompt(tmp_path: Path) -> None:
|
|
target = tmp_path / "file with spaces.log"
|
|
target.write_text("hello\n", encoding="utf-8")
|
|
|
|
prompt, attachments = _path_prompt_and_attachments(
|
|
f'/path "{target}" summarize errors'
|
|
)
|
|
|
|
assert str(target.resolve(strict=False)) in prompt
|
|
assert "summarize errors" in prompt
|
|
assert "attachments=[]" in prompt
|
|
assert attachments == []
|
|
|
|
|
|
def test_path_command_parses_unquoted_existing_path_with_spaces(tmp_path: Path) -> None:
|
|
target = tmp_path / "file with spaces.log"
|
|
target.write_text("hello\n", encoding="utf-8")
|
|
|
|
path, trailing_prompt = _parse_path_command(f"/path {target} inspect it")
|
|
|
|
assert path == target
|
|
assert trailing_prompt == "inspect it"
|
|
|
|
|
|
@pytest.mark.parametrize("unsafe", ["<", ">", "\n", "\r"])
|
|
def test_path_command_rejects_unsafe_path_token(unsafe: str) -> None:
|
|
with pytest.raises(ValueError, match="not allowed|Invalid"):
|
|
_path_prompt_and_attachments(f"/path bad{unsafe}name")
|
|
|
|
|
|
@pytest.mark.parametrize("command", ["/path", '/path "unterminated'])
|
|
def test_path_command_rejects_missing_or_unclosed_quote(command: str) -> None:
|
|
with pytest.raises(ValueError, match="Usage"):
|
|
_path_prompt_and_attachments(command)
|
|
|
|
|
|
def test_path_command_default_prompt_mentions_no_upload(tmp_path: Path) -> None:
|
|
target = tmp_path / "notes.md"
|
|
target.write_text("# Notes\n", encoding="utf-8")
|
|
|
|
prompt, attachments = _path_prompt_and_attachments(f"/path {target}")
|
|
|
|
assert "Analyze this local path" in prompt
|
|
assert "did not upload or attach file bytes" in prompt
|
|
assert "path string" in prompt
|
|
assert str(target.resolve(strict=False)) in prompt
|
|
assert attachments == []
|
|
|
|
|
|
def test_path_strategy_directory_uses_list_and_search(tmp_path: Path) -> None:
|
|
hint = _path_strategy_hint(tmp_path)
|
|
assert "list_dir" in hint
|
|
assert "glob_search" in hint
|
|
assert "grep_search" in hint
|
|
|
|
|
|
@pytest.mark.parametrize("suffix", [".csv", ".tsv", ".xlsx"])
|
|
def test_path_strategy_spreadsheet_uses_read_spreadsheet(
|
|
tmp_path: Path,
|
|
suffix: str,
|
|
) -> None:
|
|
target = tmp_path / f"data{suffix}"
|
|
target.write_bytes(b"a,b\n1,2\n")
|
|
assert "read_spreadsheet" in _path_strategy_hint(target)
|
|
|
|
|
|
@pytest.mark.parametrize("suffix", [".txt", ".log", ".md", ".json", ".html"])
|
|
def test_path_strategy_text_uses_read_file_or_grep(tmp_path: Path, suffix: str) -> None:
|
|
target = tmp_path / f"data{suffix}"
|
|
target.write_text("hello\n", encoding="utf-8")
|
|
hint = _path_strategy_hint(target)
|
|
assert "read_file" in hint
|
|
assert "grep_search" in hint
|
|
|
|
|
|
def test_path_strategy_pdf_rejects_with_file_guidance(tmp_path: Path) -> None:
|
|
target = tmp_path / "paper.pdf"
|
|
target.write_bytes(b"%PDF-1.4\n")
|
|
with pytest.raises(ValueError, match="/file"):
|
|
_path_strategy_hint(target)
|
|
|
|
|
|
@pytest.mark.parametrize("suffix", [".zip", ".exe"])
|
|
def test_path_strategy_obvious_binary_rejects(tmp_path: Path, suffix: str) -> None:
|
|
target = tmp_path / f"payload{suffix}"
|
|
target.write_bytes(b"binary")
|
|
with pytest.raises(ValueError, match="not suitable"):
|
|
_path_strategy_hint(target)
|
|
|
|
|
|
def test_path_strategy_nul_sample_rejects(tmp_path: Path) -> None:
|
|
target = tmp_path / "payload.txt"
|
|
target.write_bytes(b"abc\x00def")
|
|
with pytest.raises(ValueError, match="NUL|not suitable"):
|
|
_path_strategy_hint(target)
|
|
|
|
|
|
def test_tui_help_includes_path_command() -> None:
|
|
commands = DEFAULT_REGISTRY.for_surface(Surface.TUI)
|
|
path_command = next(cmd for cmd in commands if cmd.name == "/path")
|
|
|
|
assert path_command.usage == "/path <path> [prompt]"
|
|
assert "without uploading bytes" in path_command.description
|
|
assert any(command.usage == "/path <path> [prompt]" for command in REGISTRY)
|