Function TestValidateMaxQueryLength

pkg/querier/querier_test.go:1260–1314

Test max query length limit works with new validateQueryTimeRange function.

(t *testing.T)

// Test max query length limit works with new validateQueryTimeRange function.
func TestValidateMaxQueryLength(t *testing.T) {
	t.Parallel()
	ctx := context.Background()
	now := time.Now()
	for _, tc := range []struct {
		name              string
		start             time.Time
		end               time.Time
		expectedStartMs   int64
		expectedEndMs     int64
		maxQueryLength    time.Duration
		exceedQueryLength bool
	}{
		{
			name:              "normal params, not hit max query length",
			start:             now.Add(-time.Hour),
			end:               now,
			expectedStartMs:   util.TimeToMillis(now.Add(-time.Hour)),
			expectedEndMs:     util.TimeToMillis(now),
			maxQueryLength:    24 * time.Hour,
			exceedQueryLength: false,
		},
		{
			name:              "normal params, hit max query length",
			start:             now.Add(-100 * time.Hour),
			end:               now,
			expectedStartMs:   util.TimeToMillis(now.Add(-100 * time.Hour)),
			expectedEndMs:     util.TimeToMillis(now),
			maxQueryLength:    24 * time.Hour,
			exceedQueryLength: true,
		},
		{
			name:              "negative start",
			start:             time.Unix(-1000, 0),
			end:               now,
			expectedStartMs:   0,
			expectedEndMs:     util.TimeToMillis(now),
			maxQueryLength:    24 * time.Hour,
			exceedQueryLength: true,
		},
	} {
		t.Run(tc.name, func(t *testing.T) {
			//parallel testing causes data race
			limits := DefaultLimitsConfig()
			overrides := validation.NewOverrides(limits, nil)
			startMs, endMs, err := validateQueryTimeRange(ctx, "test", util.TimeToMillis(tc.start), util.TimeToMillis(tc.end), overrides, 0)
			require.NoError(t, err)
			startTime := model.Time(startMs)
			endTime := model.Time(endMs)
			if tc.maxQueryLength > 0 {
				require.Equal(t, tc.exceedQueryLength, endTime.Sub(startTime) > tc.maxQueryLength)
			}
		})
	}
}

// mockDistibutorFor duplicates the chunks in the mockChunkStore into the mockDistributor
// so we can test everything is dedupe correctly.