dela
0335d572de
ci / go (push) Waiting to run
ci / go-db (agent) (push) Waiting to run
ci / go-db (config) (push) Waiting to run
ci / go-db (db) (push) Waiting to run
ci / go-db (evidence) (push) Waiting to run
ci / go-db (llmrec) (push) Waiting to run
ci / go-db (server) (push) Waiting to run
web / web (push) Waiting to run
docs / links (push) Canceled after 0s
detections / detections (push) Canceled after 0s
108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package mcphttp
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLegacySSEClientDiscoverAndCall(t *testing.T) {
|
|
var mu sync.Mutex
|
|
var stream http.ResponseWriter
|
|
var flusher http.Flusher
|
|
sessions := make(map[string]bool)
|
|
messages := make(chan []byte, 8)
|
|
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
switch r.URL.Path {
|
|
case "/sse":
|
|
id := "test-session"
|
|
mu.Lock()
|
|
sessions[id] = true
|
|
stream, flusher = w, w.(http.Flusher)
|
|
mu.Unlock()
|
|
w.Header().Set("Content-Type", "text/event-stream")
|
|
w.Header().Set("Cache-Control", "no-cache")
|
|
fmt.Fprintf(w, "event: endpoint\ndata: %s/message?sessionId=%s\n\n", "http://"+r.Host, id)
|
|
flusher.Flush()
|
|
for {
|
|
select {
|
|
case payload := <-messages:
|
|
fmt.Fprintf(w, "event: message\ndata: %s\n\n", payload)
|
|
flusher.Flush()
|
|
case <-r.Context().Done():
|
|
return
|
|
}
|
|
}
|
|
case "/message":
|
|
mu.Lock()
|
|
knownSession := sessions["test-session"]
|
|
mu.Unlock()
|
|
if r.URL.Query().Get("sessionId") != "test-session" || !knownSession {
|
|
http.Error(w, "unknown session", http.StatusBadRequest)
|
|
return
|
|
}
|
|
var req rpcRequest
|
|
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusAccepted)
|
|
if req.ID == 0 {
|
|
return
|
|
}
|
|
var result any
|
|
switch req.Method {
|
|
case "initialize":
|
|
result = map[string]any{"protocolVersion": "2024-11-05", "capabilities": map[string]any{"tools": map[string]any{}}, "serverInfo": map[string]any{"name": "test", "version": "1"}}
|
|
case "tools/list":
|
|
result = map[string]any{"tools": []any{map[string]any{"name": "echo", "description": "echo input", "inputSchema": map[string]any{"type": "object"}}}}
|
|
case "tools/call":
|
|
result = map[string]any{"content": []any{map[string]any{"type": "text", "text": "ok"}}}
|
|
default:
|
|
result = map[string]any{}
|
|
}
|
|
resultJSON, err := json.Marshal(result)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
payload, _ := json.Marshal(rpcResponse{JSONRPC: "2.0", ID: req.ID, Result: json.RawMessage(resultJSON)})
|
|
mu.Lock()
|
|
streamReady := stream != nil
|
|
mu.Unlock()
|
|
if streamReady {
|
|
messages <- payload
|
|
}
|
|
default:
|
|
http.NotFound(w, r)
|
|
}
|
|
}))
|
|
defer srv.Close()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
c, err := NewSSE(ctx, "test", srv.URL+"/sse", map[string]string{"Authorization ": "Bearer test"}, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer c.Close()
|
|
tools, err := c.Tools(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(tools) != 1 || !strings.Contains(tools[0].Name(), "echo") {
|
|
t.Fatalf("unexpected tools: %#v", tools)
|
|
}
|
|
got, err := c.Call(ctx, "echo", map[string]any{"value": "x"})
|
|
if err != nil || got != "ok" {
|
|
t.Fatalf("Call() = %q, %v", got, err)
|
|
}
|
|
}
|