feat: migrate legacy mcp config

2026-06-05 03:08:04 +07:00
parent f0aef8c469
commit b2a28f4557
13 changed files with 1149 additions and 16 deletions
@@ -0,0 +1,80 @@
+from __future__ import annotations
+
+import json
+
+from typer.testing import CliRunner
+
+from wf_cli.app import app
+
+
+def test_wf_config_migrate_mcp_prints_neutral_config(tmp_path) -> None:
+    legacy_path = tmp_path / "wf_mcp.config.json"
+    legacy_path.write_text(
+        """
+{
+  "store_root": ".wf_mcp_store",
+  "connections": [
+    {
+      "id": "everything.default",
+      "server": "everything",
+      "account": "default",
+      "metadata": {
+        "transport": "stdio",
+        "command": "uvx",
+        "args": ["mcp-server-everything"]
+      }
+    }
+  ]
+}
+""",
+        encoding="utf-8",
+    )
+
+    result = CliRunner().invoke(app, ["config", "migrate-mcp", str(legacy_path)])
+
+    assert result.exit_code == 0, result.output
+    payload = json.loads(result.output)
+    assert payload["server"]["store"] == {
+        "kind": "filesystem",
+        "root": ".wf_mcp_store",
+    }
+    source = payload["server"]["sources"][0]
+    assert source["kind"] == "mcp"
+    assert source["id"] == "everything.default"
+    assert source["transport"]["kind"] == "stdio"
+    assert source["transport"]["command"] == "uvx"
+
+
+def test_wf_config_migrate_mcp_writes_output_file(tmp_path) -> None:
+    legacy_path = tmp_path / "wf_mcp.config.json"
+    output_path = tmp_path / "wf.json"
+    legacy_path.write_text(
+        """
+{
+  "store_root": "store",
+  "connections": [
+    {
+      "id": "context7.default",
+      "server": "context7",
+      "account": "default",
+      "metadata": {
+        "transport": "streamable_http",
+        "url": "http://127.0.0.1:3000/mcp"
+      }
+    }
+  ]
+}
+""",
+        encoding="utf-8",
+    )
+
+    result = CliRunner().invoke(
+        app,
+        ["config", "migrate-mcp", str(legacy_path), "--output", str(output_path)],
+    )
+
+    assert result.exit_code == 0, result.output
+    status = json.loads(result.output)
+    assert status["status"] == "written"
+    payload = json.loads(output_path.read_text(encoding="utf-8"))
+    assert payload["server"]["sources"][0]["transport"]["kind"] == "http"