mirror of
https://github.com/himanshudongre/smriti.git
synced 2026-08-28 05:14:59 +00:00
b5ccdce758
Adds WorkTree schema and migration, /api/v5/worktrees CRUD, CLI and MCP worktree surfaces, targeted regression tests, health capability, and minimal docs. Live Postgres migration/manual localhost verification intentionally remain pending until the full backend provider-config gate is resolved.
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from smriti_cli import mcp_server
|
|
|
|
|
|
def _worktree_dict(**overrides):
|
|
base = {
|
|
"id": "11111111-2222-3333-4444-555555555555",
|
|
"repo_id": "space-uuid",
|
|
"agent": "claude-1",
|
|
"path": "/tmp/worktree",
|
|
"branch_name": "smriti/claude-1/abc12345",
|
|
"base_commit_sha": "abc123",
|
|
"status": "active",
|
|
"created_at": "2026-05-04T00:00:00Z",
|
|
"closed_at": None,
|
|
}
|
|
base.update(overrides)
|
|
return base
|
|
|
|
|
|
def test_mcp_worktree_open_calls_client(mock_client):
|
|
mock_client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
|
|
mock_client.create_worktree.return_value = _worktree_dict(path="/tmp/wt")
|
|
|
|
result = mcp_server.smriti_worktree_open(
|
|
space="my-project",
|
|
agent="claude-1",
|
|
branch="feature/wt",
|
|
base_commit="abc123",
|
|
base_path="/tmp/wt",
|
|
)
|
|
|
|
assert "/tmp/wt" in result
|
|
mock_client.resolve_space.assert_called_once_with("my-project")
|
|
mock_client.create_worktree.assert_called_once_with(
|
|
space_id="space-uuid",
|
|
agent="claude-1",
|
|
branch_name="feature/wt",
|
|
base_commit_sha="abc123",
|
|
base_path="/tmp/wt",
|
|
)
|
|
|
|
|
|
def test_mcp_worktree_list_calls_client(mock_client):
|
|
mock_client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
|
|
mock_client.list_worktrees.return_value = [_worktree_dict()]
|
|
|
|
result = mcp_server.smriti_worktree_list(
|
|
space="my-project",
|
|
include_closed=True,
|
|
)
|
|
|
|
assert "claude-1" in result
|
|
mock_client.resolve_space.assert_called_once_with("my-project")
|
|
mock_client.list_worktrees.assert_called_once_with(
|
|
"space-uuid",
|
|
include_closed=True,
|
|
)
|
|
|
|
|
|
def test_mcp_worktree_show_calls_client(mock_client):
|
|
mock_client.get_worktree.return_value = _worktree_dict()
|
|
|
|
result = mcp_server.smriti_worktree_show("wt-uuid")
|
|
|
|
assert "smriti/claude-1/abc12345" in result
|
|
mock_client.get_worktree.assert_called_once_with("wt-uuid")
|
|
|
|
|
|
def test_mcp_worktree_close_calls_client(mock_client):
|
|
mock_client.close_worktree.return_value = _worktree_dict(status="closed")
|
|
|
|
result = mcp_server.smriti_worktree_close("wt-uuid", force=True)
|
|
|
|
assert "Closed worktree" in result
|
|
mock_client.close_worktree.assert_called_once_with("wt-uuid", force=True)
|