Function testFrontend

modules/frontend/mcp_tools_test.go:560–617

()

}

func testFrontend() (*MCPServer, func(t *testing.T, req mcp.CallToolRequest, handler func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error), expected expectedResult)) {
	var lastRequest *http.Request

	// Mock search handler that returns a successful response and stores the request
	mockHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		lastRequest = r
		w.WriteHeader(http.StatusOK)
	})

	server := &MCPServer{
		frontend: &QueryFrontend{
			SearchHandler:              mockHandler,
			TraceByIDHandlerV2:         mockHandler,
			SearchTagsV2Handler:        mockHandler,
			SearchTagsValuesV2Handler:  mockHandler,
			MetricsQueryInstantHandler: mockHandler,
			MetricsQueryRangeHandler:   mockHandler,
		},
		logger:     log.NewNopLogger(),
		pathPrefix: "",
	}

	callAndTestResults := func(t *testing.T, req mcp.CallToolRequest, handler func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error), expected expectedResult) {
		ctx := context.Background()
		result, err := handler(ctx, req)

		require.NoError(t, err)
		require.NotNil(t, result)

		if expected.meta != nil {
			require.Equal(t, expected.meta, result.Meta.AdditionalFields)
		}

		if expected.err != "" {
			// Check if the result contains an error
			require.True(t, result.IsError)
			require.Len(t, result.Content, 1)
			textContent, ok := result.Content[0].(mcp.TextContent)
			require.True(t, ok)
			require.Equal(t, expected.err, textContent.Text)
			return
		}

		// For successful cases, verify we have text content
		require.False(t, result.IsError)
		require.NotEmpty(t, result.Content)

		// Parse and verify the request URL
		require.Equal(t, expected.path, lastRequest.URL.Path)

		actualParams := lastRequest.URL.Query()
		for key, expectedValue := range expected.params {
			require.Equal(t, expectedValue, actualParams.Get(key), "parameter %s", key)
		}
	}

	return server, callAndTestResults
}