add tests for label time range limit

Signed-off-by: Ahmed Hassan <afayekhassan@gmail.com>
cortexproject · Oct 2, 2024 · 70ae7b9 · 70ae7b9
1 parent 274f583
commit 70ae7b9
Showing 1 changed file with 73 additions and 0 deletions.
diff --git a/pkg/querier/querier_test.go b/pkg/querier/querier_test.go
@@ -928,6 +928,79 @@ func TestQuerier_ValidateQueryTimeRange_MaxQueryLength_Series(t *testing.T) {
 	require.True(t, strings.Contains(ss.Err().Error(), "the query time range exceeds the limit (query length: 721h0m0s, limit: 720h0m0s)"))
 }
 
+func TestQuerier_ValidateQueryTimeRange_MaxQueryLength_Labels(t *testing.T) {
+	t.Parallel()
+	const maxQueryLength = 30 * 24 * time.Hour
+	tests := map[string]struct {
+		startTime            time.Time
+		endTime              time.Time
+		expected             error
+		ignoreMaxQueryLength bool
+	}{
+		"time range shorter than maxQueryLength": {
+			startTime:            time.Now().Add(-maxQueryLength).Add(time.Hour),
+			endTime:              time.Now(),
+			expected:             nil,
+			ignoreMaxQueryLength: false,
+		},
+		"time range longer than maxQueryLength": {
+			startTime:            time.Now().Add(-maxQueryLength).Add(-time.Hour),
+			endTime:              time.Now(),
+			expected:             validation.LimitError("expanding series: the query time range exceeds the limit (query length: 721h0m0s, limit: 720h0m0s)"),
+			ignoreMaxQueryLength: false,
+		},
+		"time range longer than maxQueryLength and ignoreMaxQueryLength is true": {
+			startTime:            time.Now().Add(-maxQueryLength).Add(-time.Hour),
+			endTime:              time.Now(),
+			expected:             validation.LimitError("expanding series: the query time range exceeds the limit (query length: 721h0m0s, limit: 720h0m0s)"),
+			ignoreMaxQueryLength: true,
+		},
+	}
+
+	for testName, testData := range tests {
+		t.Run(testName, func(t *testing.T) {
+			var cfg Config
+			flagext.DefaultValues(&cfg)
+			cfg.ActiveQueryTrackerDir = ""
+			cfg.IgnoreMaxQueryLength = testData.ignoreMaxQueryLength
+
+			limits := DefaultLimitsConfig()
+			limits.MaxQueryLength = model.Duration(maxQueryLength)
+			overrides, err := validation.NewOverrides(limits, nil)
+			require.NoError(t, err)
+
+			chunkStore := &emptyChunkStore{}
+			distributor := &emptyDistributor{}
+
+			queryables := []QueryableWithFilter{UseAlwaysQueryable(NewMockStoreQueryable(chunkStore))}
+			queryable, _, _ := New(cfg, overrides, distributor, queryables, nil, log.NewNopLogger())
+
+			ctx := user.InjectOrgID(context.Background(), "test")
+
+			q, err := queryable.Querier(util.TimeToMillis(testData.startTime), util.TimeToMillis(testData.endTime))
+			require.NoError(t, err)
+
+			_, _, err = q.LabelNames(ctx, &storage.LabelHints{Limit: 0})
+
+			if testData.expected != nil {
+				require.NotNil(t, err)
+				assert.True(t, strings.Contains(testData.expected.Error(), err.Error()))
+			} else {
+				assert.Nil(t, err)
+			}
+
+			_, _, err = q.LabelValues(ctx, labels.MetricName, &storage.LabelHints{Limit: 0})
+
+			if testData.expected != nil {
+				require.NotNil(t, err)
+				assert.True(t, strings.Contains(testData.expected.Error(), err.Error()))
+			} else {
+				assert.Nil(t, err)
+			}
+		})
+	}
+}
+
 func TestQuerier_ValidateQueryTimeRange_MaxQueryLookback(t *testing.T) {
 	t.Parallel()
 	const (