f

Author: Ryan-Git

integration_test/tidb_tidb_test.go

@@ -6,8 +6,6 @@ import (
 	"testing"
 	"time"
 
-	"github.com/moiot/gravity/pkg/consts"
-
 	"github.com/moiot/gravity/pkg/inputs"
 	"github.com/moiot/gravity/pkg/outputs/mysql"
 	"github.com/moiot/gravity/pkg/sql_execution_engine"
@@ -41,8 +39,7 @@ func TestBidirection(t *testing.T) {
 	r.NoError(err)
 	defer sourceDB.Close()
 
-	_, err = sourceDB.Exec("drop database if exists " + consts.GravityDBName)
-	r.NoError(err)
+	r.NoError(mysql_test.SetupTestDB(sourceDB, sourceDBName))
 
 	err = utils.InitInternalTxnTags(sourceDB)
 	r.NoError(err)
@@ -51,6 +48,8 @@ func TestBidirection(t *testing.T) {
 	r.NoError(err)
 	defer targetDB.Close()
 
+	r.NoError(mysql_test.SetupTestDB(targetDB, targetDBName))
+
 	pipelineConfig := config.PipelineConfigV3{
 		PipelineName: t.Name(),
 		Version:      config.PipelineConfigV3Version,
@@ -103,9 +102,6 @@ func TestBidirection(t *testing.T) {
 
 	r.NoError(server.Start())
 
-	_, err = sourceDB.Exec(fmt.Sprintf("create database `%s`", sourceDBName))
-	r.NoError(err)
-
 	_, err = sourceDB.Exec(fmt.Sprintf("create table `%s`.t(id int(11), primary key(id)) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 ", sourceDBName))
 	r.NoError(err)
 

pkg/mysql_test/test.go

@@ -432,7 +432,7 @@ func SeedCompositePrimaryKeyInt(db *sql.DB, dbName string) {
 
 }
 
-func setupTestDB(db *sql.DB, dbName string) error {
+func SetupTestDB(db *sql.DB, dbName string) error {
 
 	// setup test tableNames
 	if _, err := db.Exec(dropDBStatement(dbName)); err != nil {
@@ -502,7 +502,7 @@ func MustCreateSourceDBConn() *sql.DB {
 // MustSetupSourceDB setup a test db, so that we can use different db in different test cases
 func MustSetupSourceDB(dbName string) *sql.DB {
 	db := MustCreateSourceDBConn()
-	err := setupTestDB(db, dbName)
+	err := SetupTestDB(db, dbName)
 	if err != nil {
 		log.Fatalf("failed to setup source db err: %v", errors.ErrorStack(err))
 	}
@@ -538,7 +538,7 @@ func MustCreateTargetDBConn() *sql.DB {
 func MustSetupTargetDB(dbName string) *sql.DB {
 	db := MustCreateTargetDBConn()
 
-	err := setupTestDB(db, dbName)
+	err := SetupTestDB(db, dbName)
 	if err != nil {
 		log.Fatalf("failed to setup source db1 err: %v", errors.ErrorStack(err))
 	}