feat: MCP server with persistence — reattach to running servers on restart

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

tests/test_server.py

@@ -0,0 +1,105 @@
+import asyncio
+import json
+import os
+from pathlib import Path
+from unittest.mock import patch, AsyncMock, MagicMock
+
+import pytest
+
+from workbench.server import WorkbenchServer
+
+
+@pytest.fixture
+def server(tmp_workbench):
+    return WorkbenchServer(workbench_dir=tmp_workbench)
+
+
+@pytest.mark.asyncio
+async def test_scaffold_creates_project(server, tmp_workbench):
+    with patch.object(server, "_start_http_server", new_callable=AsyncMock, return_value=8070):
+        result = await server.workbench_scaffold("test-proj", "Test Project")
+    data = json.loads(result)
+    assert "url" in data
+    assert (tmp_workbench / "test-proj" / "index.html").exists()
+
+
+@pytest.mark.asyncio
+async def test_scaffold_reattaches_to_running_server(server, tmp_workbench):
+    from workbench.project import create_project, write_server_info
+    create_project("test-proj", "Test", workbench_dir=tmp_workbench)
+    write_server_info("test-proj", pid=os.getpid(), port=9999, workbench_dir=tmp_workbench)
+    with patch.object(server, "_is_server_alive", return_value=True):
+        with patch.object(server, "_start_http_server", new_callable=AsyncMock) as mock_start:
+            result = await server.workbench_scaffold("test-proj", "Test")
+            mock_start.assert_not_called()
+    data = json.loads(result)
+    assert "9999" in data["url"]
+
+
+@pytest.mark.asyncio
+async def test_scaffold_replaces_dead_server(server, tmp_workbench):
+    from workbench.project import create_project, write_server_info
+    create_project("test-proj", "Test", workbench_dir=tmp_workbench)
+    write_server_info("test-proj", pid=99999, port=9999, workbench_dir=tmp_workbench)
+    with patch.object(server, "_is_server_alive", return_value=False):
+        with patch.object(server, "_start_http_server", new_callable=AsyncMock, return_value=8070):
+            result = await server.workbench_scaffold("test-proj", "Test")
+    data = json.loads(result)
+    assert "8070" in data["url"]
+
+
+@pytest.mark.asyncio
+async def test_list_empty(server):
+    result = await server.workbench_list()
+    data = json.loads(result)
+    assert data["projects"] == []
+
+
+@pytest.mark.asyncio
+async def test_log_writes_to_disk(server, tmp_workbench):
+    with patch.object(server, "_start_http_server", new_callable=AsyncMock, return_value=8070):
+        await server.workbench_scaffold("test-proj", "Test")
+    result = await server.workbench_log("test-proj", "R412 measured 1.05M")
+    data = json.loads(result)
+    assert data["ok"] is True
+    jsonl = (tmp_workbench / "test-proj" / "session.jsonl").read_text().strip()
+    assert "R412 measured 1.05M" in jsonl
+
+
+@pytest.mark.asyncio
+async def test_state_saves_to_disk(server, tmp_workbench):
+    with patch.object(server, "_start_http_server", new_callable=AsyncMock, return_value=8070):
+        await server.workbench_scaffold("test-proj", "Test")
+    state = json.dumps({"template": "<h1>Hello</h1>"})
+    result = await server.workbench_state("test-proj", state)
+    data = json.loads(result)
+    assert data["ok"] is True
+    saved = json.loads((tmp_workbench / "test-proj" / "state.json").read_text())
+    assert saved["template"] == "<h1>Hello</h1>"
+
+
+@pytest.mark.asyncio
+async def test_stop_cleans_up(server, tmp_workbench):
+    with patch.object(server, "_start_http_server", new_callable=AsyncMock, return_value=8070):
+        await server.workbench_scaffold("test-proj", "Test")
+    server._runners["test-proj"] = AsyncMock()
+    result = await server.workbench_stop("test-proj")
+    data = json.loads(result)
+    assert data["ok"] is True
+    assert not (tmp_workbench / "test-proj" / ".server.json").exists()
+
+
+@pytest.mark.asyncio
+async def test_reconnect_on_startup(tmp_workbench):
+    from workbench.project import create_project, write_server_info
+    create_project("live-proj", "Live", workbench_dir=tmp_workbench)
+    write_server_info("live-proj", pid=os.getpid(), port=8070, workbench_dir=tmp_workbench)
+    create_project("dead-proj", "Dead", workbench_dir=tmp_workbench)
+    write_server_info("dead-proj", pid=99999, port=8071, workbench_dir=tmp_workbench)
+    srv = WorkbenchServer(workbench_dir=tmp_workbench)
+    with patch.object(srv, "_is_server_alive", side_effect=lambda port: port == 8070):
+        await srv.reconnect_existing_servers()
+    assert "live-proj" in srv._active
+    assert srv._active["live-proj"]["port"] == 8070
+    assert "dead-proj" not in srv._active
+    assert not (tmp_workbench / "dead-proj" / ".server.json").exists()