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>
85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package milestone
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"maps"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"sync"
|
|
"testing"
|
|
|
|
"gitea.com/gitea/gitea-mcp/pkg/flag"
|
|
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
)
|
|
|
|
func Test_milestoneWriteFn_dueOn(t *testing.T) {
|
|
const (
|
|
owner = "octo"
|
|
repo = "demo"
|
|
id = 42
|
|
due = "2026-05-18T23:59:59Z"
|
|
)
|
|
|
|
var (
|
|
mu sync.Mutex
|
|
bodies = map[string]map[string]any{}
|
|
)
|
|
|
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
switch r.URL.Path {
|
|
case "/api/v1/version":
|
|
_, _ = w.Write([]byte(`{"version":"1.12.0"}`))
|
|
case fmt.Sprintf("/api/v1/repos/%s/%s/milestones", owner, repo),
|
|
fmt.Sprintf("/api/v1/repos/%s/%s/milestones/%d", owner, repo, id):
|
|
mu.Lock()
|
|
var body map[string]any
|
|
_ = json.NewDecoder(r.Body).Decode(&body)
|
|
bodies[r.Method] = body
|
|
mu.Unlock()
|
|
_, _ = w.Write(fmt.Appendf(nil, `{"id":%d,"title":"v1","due_on":%q}`, id, due))
|
|
default:
|
|
http.NotFound(w, r)
|
|
}
|
|
})
|
|
|
|
server := httptest.NewServer(handler)
|
|
defer server.Close()
|
|
|
|
origHost, origToken, origVersion := flag.Host, flag.Token, flag.Version
|
|
flag.Host, flag.Token, flag.Version = server.URL, "", "test"
|
|
defer func() { flag.Host, flag.Token, flag.Version = origHost, origToken, origVersion }()
|
|
|
|
args := map[string]any{"owner": owner, "repo": repo, "due_on": due}
|
|
|
|
cases := []struct {
|
|
name string
|
|
fn func(context.Context, map[string]any) (*mcp.CallToolResult, error)
|
|
method string
|
|
extra map[string]any
|
|
}{
|
|
{"create", createMilestoneFn, http.MethodPost, map[string]any{"title": "v1"}},
|
|
{"edit", editMilestoneFn, http.MethodPatch, map[string]any{"id": float64(id)}},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
a := map[string]any{}
|
|
maps.Copy(a, args)
|
|
maps.Copy(a, tc.extra)
|
|
res, err := tc.fn(context.Background(), a)
|
|
if err != nil || res.IsError {
|
|
t.Fatalf("%s err=%v result=%v", tc.name, err, res)
|
|
}
|
|
mu.Lock()
|
|
body := bodies[tc.method]
|
|
mu.Unlock()
|
|
if got, _ := body["due_on"].(string); got != due {
|
|
t.Fatalf("%s: expected due_on=%q, got %v (body: %v)", tc.name, due, got, body)
|
|
}
|
|
})
|
|
}
|
|
}
|