Function TestLimits

pkg/querier/querier_test.go:387–540

By testing limits, we are ensuring queries with multiple selects share the query limiter

(t *testing.T)

// By testing limits, we are ensuring queries with multiple selects share the query limiter
func TestLimits(t *testing.T) {
	t.Parallel()
	start := time.Now().Add(-2 * time.Hour)
	end := time.Now()
	ctx := user.InjectOrgID(context.Background(), "0")
	var cfg Config
	flagext.DefaultValues(&cfg)
	const chunks = 1

	for _, enc := range encodings {
		chks, _ := makeMockChunks(t, chunks, enc, model.TimeFromUnix(start.Unix()))
		clientChks, err := chunkcompat.ToChunks(chks)
		require.NoError(t, err)
		streamResponse := client.QueryStreamResponse{
			Chunkseries: []client.TimeSeriesChunk{
				{
					Labels: []cortexpb.LabelAdapter{
						{Name: model.MetricNameLabel, Value: "foo"},
						{Name: "order", Value: "2"},
					},
					Chunks: clientChks,
				},
				{
					Labels: []cortexpb.LabelAdapter{
						{Name: model.MetricNameLabel, Value: "foo"},
						{Name: "order", Value: "1"},
					},
					Chunks: clientChks,
				},
				{
					Labels: []cortexpb.LabelAdapter{
						{Name: model.MetricNameLabel, Value: "foo"},
						{Name: "orders", Value: "3"},
					},
					Chunks: clientChks,
				},
				{
					Labels: []cortexpb.LabelAdapter{
						{Name: model.MetricNameLabel, Value: "bar"},
						{Name: "orders", Value: "2"},
					},
					Chunks: clientChks,
				},
				{
					Labels: []cortexpb.LabelAdapter{
						{Name: model.MetricNameLabel, Value: "bar"},
						{Name: "orders", Value: "1"},
					},
					Chunks: clientChks,
				},
			},
		}

		distributor := &MockLimitingDistributor{
			response: &streamResponse,
		}

		distributorQueryableStreaming := newDistributorQueryable(distributor, cfg.IngesterMetadataStreaming, cfg.IngesterLabelNamesWithMatchers, batch.NewChunkMergeIterator, cfg.QueryIngestersWithin, nil, 1)