| from tests.conftest import git |
| from ycode.tools import ToolRegistry, default_tools |
| from ycode.tools.git import current_branch, diff_text, is_git_repo, status_entries |
|
|
|
|
| def test_not_a_repo(workspace, make_ctx): |
| assert not is_git_repo(workspace) |
| result = ToolRegistry(default_tools()).dispatch("git_status", {}, make_ctx(workspace)) |
| assert "Not a git repository" in result.content |
|
|
|
|
| def test_status_and_branch(git_workspace, make_ctx): |
| assert is_git_repo(git_workspace) |
| assert current_branch(git_workspace) == "main" |
| assert status_entries(git_workspace) == [] |
| (git_workspace / "app.py").write_text("print('changed')\n") |
| (git_workspace / "new.py").write_text("x = 1\n") |
| entries = {e.path: e.short for e in status_entries(git_workspace)} |
| assert entries == {"app.py": "M", "new.py": "A"} |
|
|
| result = ToolRegistry(default_tools()).dispatch("git_status", {}, make_ctx(git_workspace)) |
| assert "## main" in result.content and "app.py" in result.content |
| assert result.summary == "2 changed files" |
|
|
|
|
| def test_diff_tool(git_workspace, make_ctx): |
| (git_workspace / "app.py").write_text("print('changed')\n") |
| (git_workspace / "new.py").write_text("x = 1\n") |
| assert "+print('changed')" in diff_text(git_workspace) |
| registry = ToolRegistry(default_tools()) |
| result = registry.dispatch("git_diff", {}, make_ctx(git_workspace)) |
| assert "-print('hello')" in result.content |
| assert "Untracked files:" in result.content and "new.py" in result.content |
| stat = registry.dispatch("git_diff", {"stat": True}, make_ctx(git_workspace)) |
| assert "app.py" in stat.content and "1 file changed" in stat.content |
|
|
|
|
| def test_log_and_branch_tools(git_workspace, make_ctx): |
| registry = ToolRegistry(default_tools()) |
| log = registry.dispatch("git_log", {"limit": 5}, make_ctx(git_workspace)) |
| assert "init" in log.content |
| branches = registry.dispatch("git_branch", {}, make_ctx(git_workspace)) |
| assert "main" in branches.content |
|
|
|
|
| def test_fresh_repo_without_commits(workspace): |
| git(workspace, "init", "-q", "-b", "trunk") |
| assert current_branch(workspace) == "trunk" |
|
|
