test: add go mcp verification script
This commit is contained in:
zhaoyilun
219
scripts/verify-go-mcp.ps1
Normal file
219
scripts/verify-go-mcp.ps1
Normal file
@@ -0,0 +1,219 @@
|
|||||||
|
$ErrorActionPreference = "Stop"
|
||||||
|
Set-StrictMode -Version Latest
|
||||||
|
|
||||||
|
$repo = (Resolve-Path -LiteralPath (Join-Path $PSScriptRoot "..")).Path
|
||||||
|
$expectedTools = @(
|
||||||
|
"win_helper_version",
|
||||||
|
"win_helper_self_check",
|
||||||
|
"win_helper_scan_windows",
|
||||||
|
"win_helper_dump_uia"
|
||||||
|
)
|
||||||
|
|
||||||
|
$tempRoot = Join-Path ([System.IO.Path]::GetTempPath()) ("isphere-go-mcp-verify-" + [guid]::NewGuid().ToString("N"))
|
||||||
|
$goMcpExe = Join-Path $tempRoot "isphere-mcp.exe"
|
||||||
|
$harnessDir = Join-Path $repo ("runs\go-mcp-harness-" + [guid]::NewGuid().ToString("N"))
|
||||||
|
$harnessMain = Join-Path $harnessDir "main.go"
|
||||||
|
|
||||||
|
function Assert-True([bool]$Condition, [string]$Message) {
|
||||||
|
if (-not $Condition) {
|
||||||
|
throw $Message
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function Invoke-Step([string]$Name, [scriptblock]$Body) {
|
||||||
|
Write-Host "## $Name"
|
||||||
|
& $Body
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
New-Item -ItemType Directory -Force -Path $tempRoot | Out-Null
|
||||||
|
New-Item -ItemType Directory -Force -Path $harnessDir | Out-Null
|
||||||
|
|
||||||
|
Push-Location $repo
|
||||||
|
try {
|
||||||
|
Invoke-Step "build C# helper" {
|
||||||
|
& powershell -NoProfile -ExecutionPolicy Bypass -File (Join-Path $repo "scripts\build-win-helper.ps1") | Out-Host
|
||||||
|
if ($LASTEXITCODE -ne 0) {
|
||||||
|
throw "build-win-helper.ps1 failed with exit code $LASTEXITCODE"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Invoke-Step "build Go MCP binary" {
|
||||||
|
& go build -o $goMcpExe ./cmd/isphere-mcp
|
||||||
|
if ($LASTEXITCODE -ne 0) {
|
||||||
|
throw "go build ./cmd/isphere-mcp failed with exit code $LASTEXITCODE"
|
||||||
|
}
|
||||||
|
Assert-True (Test-Path -LiteralPath $goMcpExe) "Go MCP binary was not created: $goMcpExe"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
Pop-Location
|
||||||
|
}
|
||||||
|
|
||||||
|
Invoke-Step "write temporary SDK harness" {
|
||||||
|
@"
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"reflect"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||||
|
|
||||||
|
"isphere-ai-bridge/internal/mcpserver"
|
||||||
|
)
|
||||||
|
|
||||||
|
var expectedTools = []string{
|
||||||
|
"win_helper_version",
|
||||||
|
"win_helper_self_check",
|
||||||
|
"win_helper_scan_windows",
|
||||||
|
"win_helper_dump_uia",
|
||||||
|
}
|
||||||
|
|
||||||
|
var forbiddenTerms = []string{"send", "search", "file", "login", "conversation", "upload", "download", "receive", "autologin"}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
serverTransport, clientTransport := mcp.NewInMemoryTransports()
|
||||||
|
server := mcpserver.NewServer()
|
||||||
|
serverErr := make(chan error, 1)
|
||||||
|
go func() {
|
||||||
|
serverErr <- server.Run(ctx, serverTransport)
|
||||||
|
}()
|
||||||
|
|
||||||
|
client := mcp.NewClient(&mcp.Implementation{Name: "verify-go-mcp", Version: "0.1.0"}, nil)
|
||||||
|
session, err := client.Connect(ctx, clientTransport, nil)
|
||||||
|
if err != nil {
|
||||||
|
fail("initialize/connect", err)
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
_ = session.Close()
|
||||||
|
cancel()
|
||||||
|
select {
|
||||||
|
case <-serverErr:
|
||||||
|
case <-time.After(2 * time.Second):
|
||||||
|
fail("server shutdown", fmt.Errorf("server did not stop"))
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
listResult, err := session.ListTools(ctx, &mcp.ListToolsParams{})
|
||||||
|
if err != nil {
|
||||||
|
fail("tools/list", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
toolNames := make([]string, 0, len(listResult.Tools))
|
||||||
|
for _, tool := range listResult.Tools {
|
||||||
|
toolNames = append(toolNames, tool.Name)
|
||||||
|
lower := strings.ToLower(tool.Name)
|
||||||
|
for _, term := range forbiddenTerms {
|
||||||
|
if strings.Contains(lower, term) {
|
||||||
|
fail("tools/list", fmt.Errorf("forbidden action term %q found in tool %q", term, tool.Name))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sort.Strings(toolNames)
|
||||||
|
want := append([]string(nil), expectedTools...)
|
||||||
|
sort.Strings(want)
|
||||||
|
if len(toolNames) != 4 || !reflect.DeepEqual(toolNames, want) {
|
||||||
|
fail("tools/list", fmt.Errorf("tool names = %#v, want %#v", toolNames, want))
|
||||||
|
}
|
||||||
|
|
||||||
|
callResult, err := session.CallTool(ctx, &mcp.CallToolParams{Name: "win_helper_version", Arguments: map[string]any{}})
|
||||||
|
if err != nil {
|
||||||
|
fail("tools/call win_helper_version", err)
|
||||||
|
}
|
||||||
|
if callResult.IsError {
|
||||||
|
fail("tools/call win_helper_version", fmt.Errorf("tool returned isError=true: %#v", callResult.Content))
|
||||||
|
}
|
||||||
|
|
||||||
|
structured, ok := callResult.StructuredContent.(map[string]any)
|
||||||
|
if !ok {
|
||||||
|
payload, _ := json.Marshal(callResult.StructuredContent)
|
||||||
|
if err := json.Unmarshal(payload, &structured); err != nil {
|
||||||
|
fail("tools/call win_helper_version", fmt.Errorf("structuredContent was not object: %T", callResult.StructuredContent))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if structured["ok"] != true {
|
||||||
|
fail("tools/call win_helper_version", fmt.Errorf("ok = %#v, want true", structured["ok"]))
|
||||||
|
}
|
||||||
|
data, ok := structured["data"].(map[string]any)
|
||||||
|
if !ok {
|
||||||
|
fail("tools/call win_helper_version", fmt.Errorf("data missing or not object: %#v", structured["data"]))
|
||||||
|
}
|
||||||
|
helperName, _ := data["helper_name"].(string)
|
||||||
|
if helperName != "ISphereWinHelper" {
|
||||||
|
fail("tools/call win_helper_version", fmt.Errorf("helper_name = %q, want ISphereWinHelper", helperName))
|
||||||
|
}
|
||||||
|
|
||||||
|
result := map[string]any{
|
||||||
|
"ok": true,
|
||||||
|
"transport": "sdk_in_memory",
|
||||||
|
"initialized": true,
|
||||||
|
"tool_count": len(toolNames),
|
||||||
|
"tools": toolNames,
|
||||||
|
"helper_name": helperName,
|
||||||
|
"real_isphere_login_required": false,
|
||||||
|
"python_runtime_required": false,
|
||||||
|
"action_tools_present": false,
|
||||||
|
}
|
||||||
|
encoded, _ := json.Marshal(result)
|
||||||
|
fmt.Println(string(encoded))
|
||||||
|
}
|
||||||
|
|
||||||
|
func fail(step string, err error) {
|
||||||
|
payload, _ := json.Marshal(map[string]any{"ok": false, "step": step, "error": err.Error()})
|
||||||
|
fmt.Fprintln(os.Stderr, string(payload))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
"@ | Set-Content -LiteralPath $harnessMain -Encoding UTF8
|
||||||
|
}
|
||||||
|
|
||||||
|
$harnessJson = $null
|
||||||
|
Invoke-Step "run MCP SDK harness initialize/tools/list/tools/call" {
|
||||||
|
Push-Location $harnessDir
|
||||||
|
try {
|
||||||
|
$output = & go run .
|
||||||
|
if ($LASTEXITCODE -ne 0) {
|
||||||
|
throw "go run SDK harness failed with exit code $LASTEXITCODE"
|
||||||
|
}
|
||||||
|
$output | ForEach-Object { Write-Host $_ }
|
||||||
|
$lastLine = @($output | Where-Object { -not [string]::IsNullOrWhiteSpace($_) }) | Select-Object -Last 1
|
||||||
|
$script:harnessJson = $lastLine | ConvertFrom-Json
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
Pop-Location
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Assert-True ($script:harnessJson.ok -eq $true) "SDK harness did not return ok=true"
|
||||||
|
Assert-True ($script:harnessJson.helper_name -eq "ISphereWinHelper") "SDK harness helper_name mismatch: $($script:harnessJson.helper_name)"
|
||||||
|
Assert-True ($script:harnessJson.tool_count -eq 4) "SDK harness tool_count mismatch: $($script:harnessJson.tool_count)"
|
||||||
|
|
||||||
|
[ordered]@{
|
||||||
|
ok = $true
|
||||||
|
verification = "go_mcp_sdk_harness"
|
||||||
|
go_mcp_binary_built = (Test-Path -LiteralPath $goMcpExe)
|
||||||
|
helper_name = $script:harnessJson.helper_name
|
||||||
|
tool_count = $script:harnessJson.tool_count
|
||||||
|
tools = $script:harnessJson.tools
|
||||||
|
python_runtime_required = $false
|
||||||
|
real_isphere_login_required = $false
|
||||||
|
action_tools_present = $false
|
||||||
|
} | ConvertTo-Json -Depth 8 -Compress
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
if (Test-Path -LiteralPath $tempRoot) {
|
||||||
|
Remove-Item -LiteralPath $tempRoot -Recurse -Force
|
||||||
|
}
|
||||||
|
if (Test-Path -LiteralPath $harnessDir) {
|
||||||
|
Remove-Item -LiteralPath $harnessDir -Recurse -Force
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user