mirror of
https://github.com/himanshudongre/smriti.git
synced 2026-08-28 05:14:59 +00:00
123 lines
3.8 KiB
Python
123 lines
3.8 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_list_renders_probe_data(mock_client):
|
|
mock_client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
|
|
mock_client.list_worktrees.return_value = [
|
|
_worktree_dict(probe={"dirty_files": 3, "ahead": 0, "behind": 2})
|
|
]
|
|
|
|
result = mcp_server.smriti_worktree_list(space="my-project")
|
|
|
|
assert "3" in result
|
|
assert "-2" in result
|
|
|
|
|
|
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)
|
|
|
|
|
|
def test_mcp_claim_passes_worktree_id(mock_client):
|
|
mock_client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
|
|
mock_client.get_head.return_value = {"commit_id": "checkpoint-uuid"}
|
|
mock_client.create_claim.return_value = {
|
|
"id": "claim-uuid",
|
|
"agent": "codex-local",
|
|
"scope": "Implement V2",
|
|
"branch_name": "worktree-v2-binding-and-enrichment",
|
|
"intent_type": "implement",
|
|
}
|
|
|
|
result = mcp_server.smriti_claim(
|
|
space="my-project",
|
|
scope="Implement V2",
|
|
agent="codex-local",
|
|
branch="worktree-v2-binding-and-enrichment",
|
|
task_id="v2-plan",
|
|
worktree_id="wt-uuid",
|
|
)
|
|
|
|
assert "Claimed" in result
|
|
mock_client.create_claim.assert_called_once_with(
|
|
space_id="space-uuid",
|
|
agent="codex-local",
|
|
scope="Implement V2",
|
|
branch_name="worktree-v2-binding-and-enrichment",
|
|
base_commit_id="checkpoint-uuid",
|
|
task_id="v2-plan",
|
|
worktree_id="wt-uuid",
|
|
intent_type="implement",
|
|
ttl_hours=4.0,
|
|
)
|