testhelpers

type DataWriter interface {
	Run()
	Stop()
	Wait()
	SetDB(*sql.DB)
}

type ErrorHandler struct {
	LastError error
}

type GhostferryUnitTestSuite struct {
	suite.Suite

	TestFerry *TestFerry
	Ferry     *ghostferry.Ferry
}

type IntegrationTestCase struct {
	T *testing.T

	SetupAction                   func(*TestFerry, *sql.DB, *sql.DB)
	AfterStartBinlogStreaming     func(*TestFerry, *sql.DB, *sql.DB)
	AfterRowCopyIsComplete        func(*TestFerry, *sql.DB, *sql.DB)
	BeforeStoppingBinlogStreaming func(*TestFerry, *sql.DB, *sql.DB)
	AfterStoppedBinlogStreaming   func(*TestFerry, *sql.DB, *sql.DB)
	CustomVerifyAction            func(*TestFerry, *sql.DB, *sql.DB)

	DisableChecksumVerifier bool

	SourceDB   *sql.DB
	TargetDB   *sql.DB
	DataWriter DataWriter
	Ferry      *TestFerry
	Verifier   *ghostferry.ChecksumTableVerifier
	// contains filtered or unexported fields
}

type MixedActionDataWriter struct {
	ProbabilityOfInsert float32
	ProbabilityOfUpdate float32
	ProbabilityOfDelete float32

	ModifyIdRange [2]int64

	NumberOfWriters int
	Db              *sql.DB
	Tables          []string

	ExtraInsertData func(string, map[string]interface{})
	ExtraUpdateData func(string, map[string]interface{})
	// contains filtered or unexported fields
}

type TestFerry struct {
	*ghostferry.Ferry

	BeforeBatchCopyListener   func(batch *ghostferry.RowBatch) error
	BeforeBinlogApplyListener func(events []ghostferry.DMLEvent) error
	BeforeRowCopyDoneListener func() error

	AfterBatchCopyListener   func(batch *ghostferry.RowBatch) error
	AfterBinlogApplyListener func(events []ghostferry.DMLEvent) error
	AfterRowCopyDoneListener func() error
}

type TestTableFilter struct {
	DbsFunc    func([]string) []string
	TablesFunc func([]*ghostferry.TableSchema) []*ghostferry.TableSchema
}