@@ -93,34 +93,45 @@ func TestSetAndGet(t *testing.T) {
9393}
9494
9595func TestInsertScenarios (t * testing.T ) {
96- if err := DB .Migrator ().DropTable (& User {}); err != nil {
96+ type UserWithAge struct {
97+ ID uint `gorm:"column:ID;primaryKey"`
98+ Name string `gorm:"column:NAME"`
99+ Age int `gorm:"column:AGE"`
100+ }
101+
102+ if err := DB .Migrator ().DropTable (& UserWithAge {}); err != nil {
97103 t .Fatalf ("Failed to drop table: %v" , err )
98104 }
99- if err := DB .AutoMigrate (& User {}); err != nil {
105+ if err := DB .AutoMigrate (& UserWithAge {}); err != nil {
100106 t .Fatalf ("Failed to migrate table: %v" , err )
101107 }
102108
103- user1 := User {Name : "Alice" , Age : 30 }
109+ user1 := UserWithAge {Name : "Alice" , Age : 30 }
104110 if err := DB .Create (& user1 ).Error ; err != nil {
105111 t .Errorf ("Basic insert failed: %v" , err )
106112 }
107113
108- user2 := User {Name : "Bob" }
114+ user2 := UserWithAge {Name : "Bob" }
109115 if err := DB .Create (& user2 ).Error ; err != nil {
110116 t .Errorf ("Insert with NULL failed: %v" , err )
111117 }
112118
113- user3 := User {Name : "O'Reilly" , Age : 45 }
119+ user3 := UserWithAge {Name : "O'Reilly" , Age : 45 }
114120 if err := DB .Create (& user3 ).Error ; err != nil {
115121 t .Errorf ("Insert with special characters failed: %v" , err )
116122 }
117123
118124 type UserWithTime struct {
119- ID uint
120- Name string
121- CreatedAt time.Time
125+ ID uint `gorm:"column:ID;primaryKey"`
126+ Name string `gorm:"column:NAME"`
127+ CreatedAt time.Time `gorm:"column:CREATED_AT"`
122128 }
123- user4 := User {Name : "Charlie" }
129+
130+ if err := DB .AutoMigrate (& UserWithTime {}); err != nil {
131+ t .Fatalf ("Failed to migrate UserWithTime table: %v" , err )
132+ }
133+
134+ user4 := UserWithTime {Name : "Charlie" }
124135 if err := DB .Create (& user4 ).Error ; err != nil {
125136 t .Errorf ("Insert with default timestamp failed: %v" , err )
126137 }
0 commit comments