package availzone

import (
	"context"
	"time"

	. "github.com/onsi/gomega"
	"github.com/pingcap/endless/testcase/oltp/oltppkg"
	"github.com/pingcap/test-infra/sdk/resource"
)

func WaitForRegionHealth(ctx context.Context, tc resource.TiDBCluster, timeOut time.Duration, interval time.Duration) {
	Eventually(func() bool {
		return oltppkg.CheckRegionHealth(tc)
	}, timeOut, interval).Should(BeTrue())
}