Function TestAcceptHeaderIsSet

modules/frontend/mcp_tools_test.go:453–546

(t *testing.T)

}

func TestAcceptHeaderIsSet(t *testing.T) {
	tests := []struct {
		name    string
		request mcp.CallToolRequest
		handler func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error)
	}{
		{
			name: "handleSearch sets Accept header",
			request: callToolRequest(map[string]any{
				"query": "{ span.foo = \"bar\" }",
			}),
		},
		{
			name: "handleInstantQuery sets Accept header",
			request: callToolRequest(map[string]any{
				"query": "{} | rate()",
			}),
		},
		{
			name: "handleRangeQuery sets Accept header",
			request: callToolRequest(map[string]any{
				"query": "{} | rate()",
			}),
		},
		{
			name: "handleGetTrace sets Accept header",
			request: callToolRequest(map[string]any{
				"trace_id": "12345678abcdef90",
			}),
		},
		{
			name:    "handleGetAttributeNames sets Accept header",
			request: callToolRequest(map[string]any{}),
		},
		{
			name: "handleGetAttributeValues sets Accept header",
			request: callToolRequest(map[string]any{
				"name": "service.name",
			}),
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var capturedRequest *http.Request

			// Mock handler that captures the request
			mockHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				capturedRequest = r
				w.WriteHeader(http.StatusOK)
			})

			server := &MCPServer{
				frontend: &QueryFrontend{
					SearchHandler:              mockHandler,
					TraceByIDHandlerV2:         mockHandler,
					SearchTagsV2Handler:        mockHandler,
					SearchTagsValuesV2Handler:  mockHandler,