3f8487f761
Co-authored-by: Alexis Tacnet <alexis@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: p.vezia <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Hiba Chaabnia <Hiba-Chaabnia@users.noreply.github.com> Co-authored-by: Nikhil Bhima <nikhilbhima@users.noreply.github.com> Co-authored-by: Nkipohcs <Nkipohcs@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
import stat
|
|
|
|
import pytest
|
|
|
|
from tests.mock.utils import collect_result
|
|
from vibe.core.tools.base import BaseToolState, ToolError
|
|
from vibe.core.tools.builtins.edit import Edit, EditArgs, EditConfig
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_edit_rewrites_with_detected_encoding(
|
|
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.chdir(tmp_path)
|
|
path = tmp_path / "utf16.txt"
|
|
original = "line one café\nline two été\n"
|
|
path.write_bytes(original.encode("utf-16"))
|
|
|
|
tool = Edit(config_getter=lambda: EditConfig(), state=BaseToolState())
|
|
await collect_result(
|
|
tool.run(
|
|
EditArgs(
|
|
file_path=str(path),
|
|
old_string="line one café",
|
|
new_string="LINE ONE CAFÉ",
|
|
)
|
|
)
|
|
)
|
|
|
|
assert path.read_bytes().startswith(b"\xff\xfe")
|
|
assert path.read_text(encoding="utf-16") == "LINE ONE CAFÉ\nline two été\n"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_edit_preserves_file_mode(
|
|
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.chdir(tmp_path)
|
|
path = tmp_path / "script.sh"
|
|
path.write_text("#!/bin/sh\necho old\n")
|
|
os.chmod(path, 0o755)
|
|
|
|
tool = Edit(config_getter=lambda: EditConfig(), state=BaseToolState())
|
|
await collect_result(
|
|
tool.run(EditArgs(file_path=str(path), old_string="old", new_string="new"))
|
|
)
|
|
|
|
assert stat.S_IMODE(path.stat().st_mode) == 0o755
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize("newline", ["\r\n", "\r", "\n"])
|
|
async def test_edit_preserves_line_endings(
|
|
newline: str, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.chdir(tmp_path)
|
|
path = tmp_path / "f.txt"
|
|
original = newline.join(["alpha", "beta", "gamma"])
|
|
path.write_bytes(original.encode("utf-8"))
|
|
|
|
tool = Edit(config_getter=lambda: EditConfig(), state=BaseToolState())
|
|
await collect_result(
|
|
tool.run(EditArgs(file_path=str(path), old_string="beta", new_string="BETA"))
|
|
)
|
|
|
|
assert path.read_bytes() == newline.join(["alpha", "BETA", "gamma"]).encode("utf-8")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_edit_binary_file_raises_tool_error(
|
|
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.chdir(tmp_path)
|
|
path = tmp_path / "blob.bin"
|
|
# PNG header + some non-text payload.
|
|
path.write_bytes(b"\x89PNG\r\n\x1a\n" + bytes(range(256)) * 4)
|
|
|
|
tool = Edit(config_getter=lambda: EditConfig(), state=BaseToolState())
|
|
with pytest.raises(ToolError, match="not valid text"):
|
|
await collect_result(
|
|
tool.run(EditArgs(file_path=str(path), old_string="PNG", new_string="JPG"))
|
|
)
|