// Copyright 2021 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package testcases

import (
	"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"
)

func init() {
	var (
		taskName             = "dm_operate_schema_auto_incremental_id"
		sourceName0          = "source-0"
		sourceName1          = "source-1"
		sourceNameList       = []string{sourceName0, sourceName1}
		name                 = "dm_operate_schema_auto_incremental_id"
		env                  = newCommonEnv()
		upStream1ExecuteSql  string
		upStream2ExecuteSql  string
		downStreamExecuteSql string
	)

	var _ = AddTestCase(spec.CaseName(name), func() steps.Step {
		env.init().b.
			LoadParamFromEnv(utils.ParamPrefix+"upStream1ExecuteSql", &upStream1ExecuteSql).
			LoadParamFromEnv(utils.ParamPrefix+"upStream2ExecuteSql", &upStream2ExecuteSql).
			LoadParamFromEnv(utils.ParamPrefix+"downStreamExecuteSql", &downStreamExecuteSql).
			CreateDatabase(env.UpStreams[0], name).
			CreateDatabase(env.UpStreams[1], name).
			CreateDatabase(env.DownStream, name).
			ExecuteSqlStatements(env.DownStream, &downStreamExecuteSql).
			DmCreateSources(env.Dm, env.DmHost, Path.SourcePaths).
			DmStartTasks(env.Dm, env.DmHost, Path.TaskPaths).
			Parallel(
				// Create table and insert data
				steps.NewExecuteSqlStatements(env.UpStreams[0], &upStream1ExecuteSql),
				steps.NewExecuteSqlStatements(env.UpStreams[1], &upStream2ExecuteSql),
			).
			DmWaitTaskSynced(env.DmHost, taskName, sourceNameList).
			DmCheckSyncDiff(env.Dm, env.SyncDiff, env.DmHost, Path.DiffPaths[0])
		return env.toBuilder().Build()

	})
}