update inte_test/一つのファイルに複数のモデル

digeon-inc · Jun 12, 2024 · da3cc8d · da3cc8d
1 parent cf66499
commit da3cc8d
Show file tree

Hide file tree

Showing 4 changed files with 71 additions and 1 deletion.
diff --git a/integration_test/expected_output.md b/integration_test/expected_output.md
@@ -11,6 +11,18 @@ Stores basic information about orders
 | user_id | int | YES | FOREIGN KEY | [users](#users) |  |  |  |
 | quantity | int | YES |  |  | 1 |  | Quantity of the product being ordered, defaults to 1 |
 
+## user_details
+
+Stores basic information about users details
+
+| Name | Type | Nullable | Constraints | Referenced | Default | Extra | Comment |
+| ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- |
+| user_detail_id | varchar(30) | NO | PRIMARY KEY |  |  |  |  |
+| user_id | int | YES | FOREIGN KEY | [users](#users) |  |  |  |
+| name | varchar(255) | NO |  |  |  |  |  |
+| created_at | datetime(3) | YES |  |  |  |  |  |
+| updated_at | datetime(3) | YES |  |  |  |  |  |
+
 ## users
 
 Stores basic information about users

diff --git a/integration_test/helper_test.go b/integration_test/helper_test.go
@@ -24,11 +24,14 @@ func init() {
 		log.Fatal(err)
 	}
 
-	// テーブルを自動的に作成または更新する
 	err = db.Set("gorm:table_options", "COMMENT='Stores basic information about users'").AutoMigrate(&testdata.Users{})
 	if err != nil {
 		log.Fatal(err)
 	}
+	err = db.Set("gorm:table_options", "COMMENT='Stores basic information about users details'").AutoMigrate(&testdata.UserDetail{})
+	if err != nil {
+		log.Fatal(err)
+	}
 	err = db.Set("gorm:table_options", "COMMENT='Stores basic information about orders'").AutoMigrate(&another_testdata.Orders{})
 	if err != nil {
 		log.Fatal(err)

diff --git a/integration_test/integration_test.go b/integration_test/integration_test.go
@@ -74,6 +74,28 @@ func TestIntegration(t *testing.T) {
 						},
 					},
 				},
+				{
+					TableName: "user_details",
+					Comment:   "Stores basic information about users details",
+					Columns: []pipe.Column{
+						{ColumnName: "created_at", IsNullable: "YES", ColumnType: "datetime(3)"},
+						{ColumnName: "name", IsNullable: "NO", ColumnType: "varchar(255)"},
+						{ColumnName: "updated_at", IsNullable: "YES", ColumnType: "datetime(3)"},
+						{
+							ColumnName:      "user_detail_id",
+							IsNullable:      "NO",
+							ColumnType:      "varchar(30)",
+							ConstraintTypes: "PRIMARY KEY",
+						},
+						{
+							ColumnName:          "user_id",
+							IsNullable:          "YES",
+							ColumnType:          "int",
+							ReferencedTableName: "users",
+							ConstraintTypes:     "FOREIGN KEY",
+						},
+					},
+				},
 				{
 					TableName: "users",
 					Comment:   "Stores basic information about users",
@@ -158,6 +180,28 @@ func TestIntegration(t *testing.T) {
 						},
 					},
 				},
+				{
+					TableName: "user_details",
+					Comment:   "Stores basic information about users details",
+					Columns: []pipe.Column{
+						{
+							ColumnName:      "user_detail_id",
+							IsNullable:      "NO",
+							ColumnType:      "varchar(30)",
+							ConstraintTypes: "PRIMARY KEY",
+						},
+						{
+							ColumnName:          "user_id",
+							IsNullable:          "YES",
+							ColumnType:          "int",
+							ReferencedTableName: "users",
+							ConstraintTypes:     "FOREIGN KEY",
+						},
+						{ColumnName: "name", IsNullable: "NO", ColumnType: "varchar(255)"},
+						{ColumnName: "created_at", IsNullable: "YES", ColumnType: "datetime(3)"},
+						{ColumnName: "updated_at", IsNullable: "YES", ColumnType: "datetime(3)"},
+					},
+				},
 				{
 					TableName: "users",
 					Comment:   "Stores basic information about users",

diff --git a/integration_test/test_data/users.go b/integration_test/test_data/users.go
@@ -1,7 +1,18 @@
 package testdata
 
+import "time"
+
 type Users struct {
 	ID    int32  `gorm:"primaryKey;autoIncrement;"`
 	Name  string `gorm:"size:255;not null;"`
 	Email string `gorm:"size:255;not null;unique;"`
 }
+
+type UserDetail struct {
+	UserDetailID string `gorm:"primaryKey;size:30;not null"`
+	UserID       int32  // 外部キー
+	User         Users  `gorm:"foreignKey:UserID"` // Usersとの関連
+	Name         string `gorm:"size:255;not null"`
+	CreatedAt    time.Time
+	UpdatedAt    time.Time
+}