Function waitForClusterReady

dgraph/cmd/dgraphimport/import_test.go:458–482

waitForClusterReady ensures the cluster is fully operational before proceeding

(t *testing.T, cluster *dgraphtest.LocalCluster, gc *dgraphapi.GrpcClient, timeout time.Duration)

// waitForClusterReady ensures the cluster is fully operational before proceeding
func waitForClusterReady(t *testing.T, cluster *dgraphtest.LocalCluster, gc *dgraphapi.GrpcClient, timeout time.Duration) error {
	deadline := time.Now().Add(timeout)
	retryDelay := 500 * time.Millisecond

	for time.Now().Before(deadline) {
		if _, err := gc.Query("schema{}"); err != nil {
			t.Logf("Cluster not ready yet: %v, retrying in %v", err, retryDelay)
			time.Sleep(retryDelay)
			retryDelay = min(retryDelay*2, 5*time.Second)
			continue
		}

		if err := cluster.HealthCheck(false); err != nil {
			t.Logf("Health check failed: %v, retrying in %v", err, retryDelay)
			time.Sleep(retryDelay)
			retryDelay = min(retryDelay*2, 5*time.Second)
			continue
		}

		t.Log("Cluster is ready")
		return nil
	}

	return fmt.Errorf("cluster not ready within %v timeout", timeout)
}

// waitForClusterStable ensures remaining alphas are accessible after some are shut down
func waitForClusterStable(t *testing.T, cluster *dgraphtest.LocalCluster, timeout time.Duration) error {