✅ 完善自动迁移相关单元测试

Signed-off-by: liutianqi <[email protected]>
godoes · Jun 6, 2024 · 02050e6 · 02050e6
1 parent 7ec951e
commit 02050e6
Show file tree

Hide file tree

Showing 2 changed files with 202 additions and 0 deletions.
diff --git a/dameng_test.go b/dameng_test.go
@@ -99,6 +99,12 @@ func TestGormConnExample(t *testing.T) {
 	} else {
 		t.Logf("迁移表结构成功！")
 	}
+	// 测试重新迁移 schema
+	if err = db.AutoMigrate(&Product{}); err != nil {
+		t.Errorf("迁移表结构失败：%v", err)
+	} else {
+		t.Logf("迁移表结构成功！")
+	}
 
 	// Create
 	data := Product{Code: "D42", Price: 100, Remark1: "VARCHAR", Remark2: "CLOB"}

diff --git a/migrator_test.go b/migrator_test.go
@@ -0,0 +1,196 @@
+package dameng
+
+import (
+	"encoding/json"
+	"reflect"
+	"testing"
+	"time"
+
+	"gorm.io/gorm"
+)
+
+func TestMigrator_AutoMigrate(t *testing.T) {
+	options := map[string]string{
+		"schema":         dmSchema,
+		"appName":        "dm_TestMigrator",
+		"connectTimeout": "30000",
+	}
+	db, err := gorm.Open(New(Config{
+		DriverName:              DriverName,
+		DSN:                     BuildUrl(dmUsername, dmPassword, dmHost, dmPort, options),
+		VarcharSizeIsCharLength: true,
+	}))
+	if err != nil {
+		t.Fatalf("initialize db session based on dialector got error: %v", err)
+	}
+
+	type args struct {
+		drop     bool
+		models   []interface{}
+		comments []string
+	}
+	tests := []struct {
+		name    string
+		args    args
+		wantErr bool
+	}{
+		{name: "TestTableUser-create", args: args{models: []interface{}{TestTableUser{}}, comments: []string{"用户信息表"}}},
+		{name: "TestTableUser-alter", args: args{models: []interface{}{TestTableUser{}}, comments: []string{"用户信息表"}, drop: true}},
+		{name: "TestTableUserNoComments", args: args{models: []interface{}{TestTableUserNoComments{}}, comments: []string{"用户信息表"}}},
+		{name: "TestTableUserAddColumn", args: args{models: []interface{}{TestTableUserAddColumn{}}, comments: []string{"用户信息表"}}},
+		{name: "TestTableUserMigrateColumn", args: args{models: []interface{}{TestTableUserMigrateColumn{}}, comments: []string{"用户信息表"}, drop: true}},
+		{name: "TestTableColumnTypeModel-create", args: args{models: []interface{}{testTableColumnTypeModel{}}, comments: []string{"测试字段类型表"}}},
+		{name: "TestTableColumnTypeModel-alter", args: args{models: []interface{}{testTableColumnTypeModel{}}, comments: []string{"测试字段类型表"}, drop: true}},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if len(tt.args.models) == 0 {
+				t.Fatal("models is nil")
+			}
+			migrator := db.Set("gorm:table_comments", tt.args.comments).Migrator()
+
+			if tt.args.drop {
+				for _, model := range tt.args.models {
+					if !migrator.HasTable(model) {
+						continue
+					}
+					if err = migrator.DropTable(model); err != nil {
+						t.Fatalf("DropTable() got error = %v", err)
+					}
+				}
+			}
+
+			if err = migrator.AutoMigrate(tt.args.models...); (err != nil) != tt.wantErr {
+				t.Errorf("AutoMigrate() error = %v, wantErr %v", err, tt.wantErr)
+			} else if err == nil {
+				t.Log("AutoMigrate() success!")
+			}
+
+			if tt.name == "TestTableUserMigrateColumn" {
+				wantUser := TestTableUserMigrateColumn{
+					TestTableUser: TestTableUser{
+						UID:         "U0",
+						Name:        "someone",
+						Account:     "guest",
+						Password:    "MAkOvrJ8JV",
+						Email:       "",
+						PhoneNumber: "+8618888888888",
+						Sex:         "1",
+						UserType:    1,
+						Enabled:     true,
+						Remark:      "Ahmad",
+					},
+					AddNewColumn:       "AddNewColumnValue",
+					CommentSingleQuote: "CommentSingleQuoteValue",
+				}
+
+				result := db.Create(&wantUser)
+				if err = result.Error; err != nil {
+					t.Fatal(err)
+				}
+
+				var gotUser TestTableUserMigrateColumn
+				result.Where(&TestTableUser{UID: "U0"}).Find(&gotUser)
+				if err = result.Error; err != nil {
+					t.Fatal(err)
+				}
+				gotUserBytes, _ := json.Marshal(gotUser)
+				t.Logf("gotUser Result: %s", gotUserBytes)
+				if !reflect.DeepEqual(gotUser, wantUser) {
+					wantUserBytes, _ := json.Marshal(wantUser)
+					t.Errorf("wantUser Info: %s", wantUserBytes)
+				}
+			}
+		})
+	}
+}
+
+// TestTableUser 测试用户信息表模型
+type TestTableUser struct {
+	ID   uint64 `gorm:"column:id;size:64;not null;autoIncrement:true;autoIncrementIncrement:1;primaryKey;comment:自增 ID" json:"id"`
+	UID  string `gorm:"column:uid;type:varchar(50);comment:用户身份标识" json:"uid"`
+	Name string `gorm:"column:name;size:50;comment:用户姓名" json:"name"`
+
+	Account  string `gorm:"column:account;type:varchar(50);comment:登录账号" json:"account"`
+	Password string `gorm:"column:password;type:varchar(512);comment:登录密码（密文）" json:"password"`
+
+	Email       string `gorm:"column:email;type:varchar(128);comment:邮箱地址" json:"email"`
+	PhoneNumber string `gorm:"column:phone_number;type:varchar(15);comment:E.164" json:"phoneNumber"`
+
+	Sex      string     `gorm:"column:sex;type:char(1);comment:性别" json:"sex"`
+	Birthday *time.Time `gorm:"column:birthday;->:false;<-:create;comment:生日" json:"birthday,omitempty"`
+
+	UserType int `gorm:"column:user_type;size:8;comment:用户类型" json:"userType"`
+
+	Enabled bool   `gorm:"column:enabled;comment:是否可用" json:"enabled"`
+	Remark  string `gorm:"column:remark;size:1024;comment:备注信息" json:"remark"`
+}
+
+func (TestTableUser) TableName() string {
+	return "test_user"
+}
+
+type TestTableUserNoComments struct {
+	ID   uint64 `gorm:"column:id;size:64;not null;autoIncrement:true;autoIncrementIncrement:1;primaryKey" json:"id"`
+	UID  string `gorm:"column:name;type:varchar(50)" json:"uid"`
+	Name string `gorm:"column:name;size:50" json:"name"`
+
+	Account  string `gorm:"column:account;type:varchar(50)" json:"account"`
+	Password string `gorm:"column:password;type:varchar(512)" json:"password"`
+
+	Email       string `gorm:"column:email;type:varchar(128)" json:"email"`
+	PhoneNumber string `gorm:"column:phone_number;type:varchar(15)" json:"phoneNumber"`
+
+	Sex      string    `gorm:"column:sex;type:char(1)" json:"sex"`
+	Birthday time.Time `gorm:"column:birthday" json:"birthday"`
+
+	UserType int `gorm:"column:user_type;size:8" json:"userType"`
+
+	Enabled bool   `gorm:"column:enabled" json:"enabled"`
+	Remark  string `gorm:"column:remark;size:1024" json:"remark"`
+}
+
+func (TestTableUserNoComments) TableName() string {
+	return "test_user"
+}
+
+type TestTableUserAddColumn struct {
+	TestTableUser
+
+	AddNewColumn string `gorm:"column:add_new_column;type:varchar(100);comment:添加新字段"`
+}
+
+func (TestTableUserAddColumn) TableName() string {
+	return "test_user"
+}
+
+type TestTableUserMigrateColumn struct {
+	TestTableUser
+
+	AddNewColumn       string `gorm:"column:add_new_column;type:varchar(100);comment:测试添加新字段"`
+	CommentSingleQuote string `gorm:"column:comment_single_quote;comment:注释中存在单引号'[']'"`
+}
+
+func (TestTableUserMigrateColumn) TableName() string {
+	return "test_user"
+}
+
+type testTableColumnTypeModel struct {
+	ID   int64  `gorm:"column:id;size:64;not null;autoIncrement:true;autoIncrementIncrement:1;primaryKey"`
+	Name string `gorm:"column:name;size:50"`
+	Age  uint8  `gorm:"column:age;size:8"`
+
+	Avatar []byte `gorm:"column:avatar;"`
+
+	Balance float64 `gorm:"column:balance;type:decimal(18, 2)"`
+	Remark  string  `gorm:"column:remark;size:-1"`
+	Enabled bool    `gorm:"column:enabled;"`
+
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	DeletedAt gorm.DeletedAt
+}
+
+func (t testTableColumnTypeModel) TableName() string {
+	return "test_table_column_type"
+}