package datarecovery

import (
	"github.com/pingcap/endless/pkg/util"
	"github.com/pingcap/endless/testcase/oltp/oltppkg"
	"github.com/pingcap/test-infra/sdk/resource"
)

func TestSuite(caseParams *oltppkg.CaseParams) {
	util.TestCase("should recover the QPS in 1 minutes and system keep normal when one tikv down", "oltp_data_online_recovery_fun_001", func() {
		OLTP_DATA_ONLINE_RECOVERY_FUN_001(caseParams.Ctx, caseParams.Tc.(resource.TiDBCluster), caseParams.BenchToolset, caseParams.WorkloadRunParms, caseParams.CaseCmdParms)
	})
	util.TestCase("should recover the QPS in 1 minutes and system keep normal when one tikv down", "oltp_sst_corruption_recovery_fun_001", func() {
		OLTP_SST_CORRUPTION_RECOVERY_FUN_001(caseParams.Ctx, caseParams.Tc.(resource.TiDBCluster), caseParams.BenchToolset, caseParams.WorkloadRunParms, caseParams.CaseCmdParms)
	})
}