Function TestInitializeReusesExistingSession

server/mcp/handler_test.go:108–156

(t *testing.T)

}

func TestInitializeReusesExistingSession(t *testing.T) {
	gin.SetMode(gin.TestMode)
	srv := newTestServer(nil)
	currentSession := srv.createSession(1)
	currentSession.initialized = true

	r := gin.New()
	r.POST("/mcp", func(c *gin.Context) {
		common.GinAppendValues(c, conf.UserKey, &model.User{ID: 1, Role: model.ADMIN})
		srv.handlePost(c)
	})

	req := httptest.NewRequest(http.MethodPost, "http://example.com/mcp", strings.NewReader(`{
		"jsonrpc":"2.0",
		"id":1,
		"method":"initialize",
		"params":{
			"protocolVersion":"2025-06-18",
			"capabilities":{},
			"clientInfo":{"name":"test-client","version":"1.0.0"}
		}
	}`))
	req.Header.Set("Accept", "application/json, text/event-stream")
	req.Header.Set("Origin", "http://example.com")
	req.Header.Set(SessionHeader, currentSession.id)

	w := httptest.NewRecorder()
	r.ServeHTTP(w, req)

	if w.Code != http.StatusOK {
		t.Fatalf("unexpected status: got %d want %d", w.Code, http.StatusOK)
	}
	if got := w.Header().Get(SessionHeader); got != currentSession.id {
		t.Fatalf("unexpected session header: got %q want %q", got, currentSession.id)
	}
	if len(srv.sessions) != 1 {
		t.Fatalf("unexpected session count: got %d want %d", len(srv.sessions), 1)
	}
	reusedSession, ok := srv.getSession(currentSession.id)
	if !ok {
		t.Fatal("expected existing session to be reused")
	}
	if reusedSession.initialized {
		t.Fatal("expected reused session to require initialized notification again")
	}
	if reusedSession.protocolVersion != "2025-06-18" {
		t.Fatalf("unexpected protocol version: got %q want %q", reusedSession.protocolVersion, "2025-06-18")
	}
}

func TestInitializeNegotiatesUnsupportedProtocolVersion(t *testing.T) {
	gin.SetMode(gin.TestMode)