File size: 3,327 Bytes
5ef3875 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | import pytest
from ycode.app import YCodeApp
from ycode.commands import COMMANDS, handle_command, is_command
from ycode.config import Config
from tests.conftest import FakeProvider, console_text, recording_console, reply
from ycode.ui.console import ConsoleUI
@pytest.fixture
def app(git_workspace):
ui = ConsoleUI(recording_console())
return YCodeApp(config=Config(max_steps=50), workspace=git_workspace, provider=FakeProvider(), ui=ui)
def output(app):
return console_text(app.ui.console)
def test_is_command():
assert is_command("/help") and is_command("/diff src/a.py")
assert not is_command("fix /etc/hosts parsing") and not is_command("/etc/hosts is wrong")
def test_help_lists_commands(app):
assert handle_command(app, "/help")
out = output(app)
for name in ("/help", "/clear", "/status", "/diff", "/files", "/model", "/exit"):
assert name in out
def test_status(app):
(app.workspace / "app.py").write_text("changed\n")
handle_command(app, "/status")
out = output(app)
assert "Branch" in out and "main" in out
assert "Modified files" in out and "1" in out
assert "Steps used" in out and "0/50" in out
assert "fake-model" in out
def test_diff_and_files(app):
handle_command(app, "/diff")
assert "No changes." in output(app)
(app.workspace / "app.py").write_text("print('bye')\n")
(app.workspace / "new.txt").write_text("n\n")
handle_command(app, "/diff")
out = output(app)
assert "+print('bye')" in out and "new.txt" in out
handle_command(app, "/files")
assert " M app.py" in output(app)
def test_model_switch(app):
handle_command(app, "/model")
assert "fake-model" in output(app)
handle_command(app, "/model claude-sonnet-5-5")
assert app.provider.model == "claude-sonnet-5-5"
def test_approval_switch(app):
handle_command(app, "/approval strict")
assert app.permissions.mode == "strict"
handle_command(app, "/approval bogus")
assert app.permissions.mode == "strict" and "Unknown mode" in output(app)
def test_clear_resets_conversation(app):
app.agent.provider.responses = [reply("hello")]
app.handle_line("say hi")
assert app.agent.messages
handle_command(app, "/clear")
assert app.agent.messages == []
@pytest.mark.parametrize("cmd", ["/exit", "/quit", "/EXIT"])
def test_exit(app, cmd):
handle_command(app, cmd)
assert app.should_exit
def test_unknown_command(app):
assert handle_command(app, "/frobnicate") is False
assert "Unknown command /frobnicate" in output(app)
def test_commands_are_not_sent_to_the_model(app):
app.handle_line("/status")
assert app.agent.provider.requests == []
def test_task_summary_shows_changes(app):
app.agent.provider.responses = [
reply("", [("edit_file", {"path": "app.py", "old_string": "hello", "new_string": "world"})]),
reply("", [("run_command", {"command": "echo checked"})]),
reply("Changed the greeting."),
]
app.handle_line("change greeting")
out = output(app)
assert "Changes:" in out and "M app.py" in out
assert "echo checked" in out
assert "YCode completed the task." in out
def test_every_command_has_help():
for command in COMMANDS.values():
assert command.help and command.usage.startswith("/")
|