Function waitUntilReady

integration/query_fuzz_test.go:1705–1736

waitUntilReady is a helper function to wait and check if both servers to test load the expected data.

(t *testing.T, ctx context.Context, c1, c2 *e2ecortex.Client, query string, start, end time.Time)

// waitUntilReady is a helper function to wait and check if both servers to test load the expected data.
func waitUntilReady(t *testing.T, ctx context.Context, c1, c2 *e2ecortex.Client, query string, start, end time.Time) {
	retries := backoff.New(ctx, backoff.Config{
		MinBackoff: 5 * time.Second,
		MaxBackoff: 10 * time.Second,
		MaxRetries: 5,
	})

	var (
		labelSet1 []model.LabelSet
		labelSet2 []model.LabelSet
		err       error
	)
	// Wait until both Cortex and Prometheus load the block.
	for retries.Ongoing() {
		labelSet1, err = c1.Series([]string{query}, start, end)
		require.NoError(t, err)
		labelSet2, err = c2.Series([]string{query}, start, end)
		require.NoError(t, err)

		// Make sure series can be queried.
		if len(labelSet1) > 0 {
			if cmp.Equal(labelSet1, labelSet2, labelSetsComparer) {
				break
			}
		}

		retries.Wait()
	}
	if err := retries.Err(); err != nil {
		t.Fatalf("failed to wait for ready, error: %v", err)
	}
}

// runQueryFuzzTestCases executes the fuzz test for the specified number of runs for both instant and range queries.
func runQueryFuzzTestCases(t *testing.T, ps *promqlsmith.PromQLSmith, c1, c2 *e2ecortex.Client, queryTime, start, end time.Time, step time.Duration, run int, skipStdAggregations bool) {