package testcases

import (
	"time"

	"github.com/pingcap/test-infra/caselib/pkg/model/workload"
	"github.com/pingcap/test-infra/caselib/pkg/spec"
	"github.com/pingcap/test-infra/caselib/pkg/steps"
	"github.com/pingcap/test-infra/caselib/pkg/utils"
	_ "github.com/pingcap/test-infra/sdk/resource/impl/k8s"
)

var (
	taskName       = "dm_migrate_incremental_ddl"
	sourceName0    = "source-0"
	sourceNameList = []string{sourceName0}
	name           = "dm_migrate_incremental_ddl"
	env            = newCommonEnv()
	executeSql     string
)
var _ = AddTestCase(spec.CaseName(name), func() steps.Step {
	env.init().b.
		CreateDatabase(env.UpStreams[0], "test").
		LoadParamFromEnv(utils.ParamPrefix+"executeSql", &executeSql).
		DmRunWorkload(env.UpStreams[0], env.Dm, env.Workload, &workload.QihooOption{
			Command:        workload.CommandTypeLoad,
			PropertyFile:   "/workloads/workloadd",
			PropertyValues: []string{"recordcount=10000", "autocommit=1", "databasename=test", "tablecount=1"},
			Threads:        20,
			Duration:       3 * time.Minute,
		}).
		DmCreateSources(env.Dm, env.DmHost, Path.SourcePaths).
		DmStartTasks(env.Dm, env.DmHost, Path.TaskPaths).
		DmWaitTaskSynced(env.DmHost, taskName, sourceNameList).
		DmCheckSyncDiff(env.Dm, env.SyncDiff, env.DmHost, Path.DiffPaths[0]).
		ExecuteSqlStatements(env.UpStreams[0], &executeSql).
		DmRunWorkload(env.UpStreams[0], env.Dm, env.Workload, &workload.QihooOption{
			Command:        workload.CommandTypeRun,
			PropertyFile:   "/workloads/workloadd",
			PropertyValues: []string{"operationcount=500000", "autocommit=1", "databasename=test", "tablecount=1"},
			Threads:        20,
			Duration:       5 * time.Minute,
		}).
		DmWaitTaskSynced(env.DmHost, taskName, sourceNameList).
		DmCheckSyncDiff(env.Dm, env.SyncDiff, env.DmHost, Path.DiffPaths[0])
	return env.toBuilder().Build()
})