mirror of
https://gitea.com/gitea/gitea-mcp.git
synced 2026-08-03 15:49:23 +02:00
Bo-Yi Wu
80c8b25d6e
- Swap the mark3labs MCP dependency for modelcontextprotocol/go-sdk v1.7.0 - Add a declarative tool definition and JSON Schema builder to the tool package - Adapt registered tools to the official low-level handler, recovering panics and mapping failures to JSON-RPC errors - Narrow tool handlers to take a plain argument map instead of an SDK request type - Rewire stdio and HTTP transports onto the official server, moving Authorization parsing into receiving middleware - Add a golden contract test that locks the exposed tool schemas, plus SDK integration and helper tests - Add a test target and run it in the pull request workflow BREAKING CHANGE: The exported helpers change signature. Tool.RegisterRead and Tool.RegisterWrite now take tool.ServerTool instead of server.ServerTool, and the annotation constructors return *mcp.ToolAnnotations from the official SDK. Callers must build tool definitions with tool.NewDefinition and handlers with the func(context.Context, map[string]any) signature. The 30-second SSE heartbeat is removed because the official SDK has no equivalent. ServerOptions.KeepAlive is deliberately not used as a substitute, since it sends MCP ping requests and ping is removed in protocol 2026. HTTP stays stateless=false, so new clients negotiate at most 2025-11-25. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package wiki
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
mcpContext "gitea.com/gitea/gitea-mcp/pkg/context"
|
|
"gitea.com/gitea/gitea-mcp/pkg/flag"
|
|
)
|
|
|
|
func TestWikiWriteBase64Encoding(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
method string
|
|
content string
|
|
}{
|
|
{"create ascii", "create", "Hello, World!"},
|
|
{"create unicode", "create", "日本語テスト 🎉"},
|
|
{"create multiline", "create", "line1\nline2\nline3"},
|
|
{"update ascii", "update", "Updated content"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
var gotBody map[string]string
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
body, _ := io.ReadAll(r.Body)
|
|
_ = json.Unmarshal(body, &gotBody)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte(`{"title":"test"}`))
|
|
}))
|
|
defer srv.Close()
|
|
|
|
origHost := flag.Host
|
|
flag.Host = srv.URL
|
|
defer func() { flag.Host = origHost }()
|
|
|
|
ctx := context.WithValue(context.Background(), mcpContext.TokenContextKey, "test-token")
|
|
|
|
args := map[string]any{
|
|
"method": tt.method,
|
|
"owner": "org",
|
|
"repo": "repo",
|
|
"content": tt.content,
|
|
"pageName": "TestPage",
|
|
"title": "TestPage",
|
|
}
|
|
|
|
result, err := wikiWriteFn(ctx, args)
|
|
if err != nil {
|
|
t.Fatalf("wikiWriteFn() error: %v", err)
|
|
}
|
|
if result.IsError {
|
|
t.Fatalf("wikiWriteFn() returned error result")
|
|
}
|
|
|
|
got := gotBody["content_base64"]
|
|
want := base64.StdEncoding.EncodeToString([]byte(tt.content))
|
|
if got != want {
|
|
t.Errorf("content_base64 = %q, want %q", got, want)
|
|
}
|
|
})
|
|
}
|
|
}
|