Function TestSkipASTTransformationsMerge

modules/frontend/search_sharder_test.go:1413–1495

TestSkipASTTransformationsMerge verifies that the per-request skip_ast_transformations URL parameter is merged with the global config list rather than overwritten by it.

(t *testing.T)

// TestSkipASTTransformationsMerge verifies that the per-request skip_ast_transformations
// URL parameter is merged with the global config list rather than overwritten by it.
func TestSkipASTTransformationsMerge(t *testing.T) {
	tests := []struct {
		name         string
		globalSkip   []string
		requestSkip  string // URL param value; empty means param is absent
		expectedSkip []string
	}{
		{
			name:         "global config only",
			globalSkip:   []string{"global_skip"},
			requestSkip:  "",
			expectedSkip: []string{"global_skip"},
		},
		{
			name:         "per-request only",
			globalSkip:   nil,
			requestSkip:  "per_req_skip",
			expectedSkip: []string{"per_req_skip"},
		},
		{
			name:         "merge global and per-request",
			globalSkip:   []string{"global_skip"},
			requestSkip:  "per_req_skip",
			expectedSkip: []string{"global_skip", "per_req_skip"},
		},
		{
			name:         "duplicated entries",
			globalSkip:   []string{"or_to_in"},
			requestSkip:  "or_to_in",
			expectedSkip: []string{"or_to_in"},
		},
		{
			name:         "neither",
			globalSkip:   nil,
			requestSkip:  "",
			expectedSkip: nil,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			var capturedSkip []string

			next := pipeline.AsyncRoundTripperFunc[combiner.PipelineResponse](func(r pipeline.Request) (pipeline.Responses[combiner.PipelineResponse], error) {
				req, err := api.ParseSearchRequest(r.HTTPRequest())
				if err == nil {
					capturedSkip = req.SkipASTTransformations
				}
				return pipeline.NewAsyncResponse(nil), nil
			})

			o, err := overrides.NewOverrides(overrides.Config{}, nil, prometheus.DefaultRegisterer)
			require.NoError(t, err)

			sharder := newAsyncSearchSharder(&mockReader{}, o, SearchSharderConfig{
				ConcurrentRequests:    defaultConcurrentRequests,
				TargetBytesPerRequest: defaultTargetBytesPerRequest,
			}, tc.globalSkip, newJobsPerQueryHistogram(), log.NewNopLogger())