Function TestInvalidProtocolVersionHeader

internal/server/mcp_test.go:1245–1291

(t *testing.T)

}

func TestInvalidProtocolVersionHeader(t *testing.T) {
	mockTools := []testutils.MockTool{testutils.MockTool1, testutils.MockTool2, testutils.MockTool3, testutils.MockTool4, testutils.MockTool5}
	mockPrompts := []testutils.MockPrompt{testutils.MockPrompt1}
	toolsMap, toolsets, promptsMap, promptsets := testutils.SetUpResources(t, mockTools, mockPrompts)
	r, shutdown := setUpServer(t, "mcp", toolsMap, toolsets, promptsMap, promptsets)
	defer shutdown()
	ts := runServer(r, false)
	defer ts.Close()

	reqBody := jsonrpc.JSONRPCRequest{
		Jsonrpc: jsonrpcVersion,
		Id:      "tools-list",
		Request: jsonrpc.Request{
			Method: "tools/list",
		},
	}
	reqMarshal, err := json.Marshal(reqBody)
	if err != nil {
		t.Fatalf("unexpected error during marshaling of body")
	}
	header := map[string]string{}
	header["MCP-Protocol-Version"] = "foo"

	resp, body, err := runRequest(ts, http.MethodPost, "/", bytes.NewBuffer(reqMarshal), header)
	if resp.Status != "400 Bad Request" {
		t.Fatalf("unexpected status: %s; %s", resp.Status, body)
	}
	var got map[string]any
	if err := json.Unmarshal(body, &got); err != nil {
		t.Fatalf("unexpected error unmarshalling body: %s", err)
	}
	errMap, ok := got["error"].(map[string]any)
	if !ok {
		t.Fatalf("expected 'error' field to be a map, got %T", got["error"])
	}
	msg, ok := errMap["message"].(string)
	if !ok {
		t.Fatalf("expected 'message' field to be a string, got %T", errMap["message"])
	}
	want := "unsupported protocol version"
	if msg != want {
		t.Fatalf("unexpected error message: got %s, want %s", got["error"], want)
	}
	if err != nil {
		t.Fatalf("unexpected error during request: %s", err)
	}
}

func TestDeleteEndpoint(t *testing.T) {
	r, shutdown := setUpServer(t, "mcp", nil, nil, nil, nil)