Skip to content

Commit bdc14f3

Browse files
Use UserWithAddress struct instead if User struct in embedded_struct_test.go
1 parent fd8295a commit bdc14f3

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

tests/embedded_struct_test.go

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -332,20 +332,20 @@ func TestZeroValueEmbeddedStruct(t *testing.T) {
332332
City string
333333
State string
334334
}
335-
type User struct {
335+
type UserWithAddress struct {
336336
ID uint
337337
Address Address `gorm:"embedded"`
338338
}
339339

340-
DB.Migrator().DropTable(&User{})
341-
err := DB.Migrator().AutoMigrate(&User{})
340+
DB.Migrator().DropTable(&UserWithAddress{})
341+
err := DB.Migrator().AutoMigrate(&UserWithAddress{})
342342
tests.AssertEqual(t, err, nil)
343343

344-
user := User{}
344+
user := UserWithAddress{}
345345
err = DB.Save(&user).Error
346346
tests.AssertEqual(t, err, nil)
347347

348-
var loaded User
348+
var loaded UserWithAddress
349349
err = DB.First(&loaded, user.ID).Error
350350
tests.AssertEqual(t, err, nil)
351351

@@ -359,24 +359,23 @@ func TestUpdateEmbeddedFields(t *testing.T) {
359359
City string
360360
State string
361361
}
362-
type User struct {
362+
type UserWithAddress struct {
363363
ID uint
364364
Address Address `gorm:"embedded"`
365365
}
366366

367-
DB.Migrator().DropTable(&User{})
368-
err := DB.Migrator().AutoMigrate(&User{})
367+
DB.Migrator().DropTable(&UserWithAddress{})
368+
err := DB.Migrator().AutoMigrate(&UserWithAddress{})
369369
tests.AssertEqual(t, err, nil)
370370

371-
user := User{Address: Address{City: "Austin", State: "TX"}}
371+
user := UserWithAddress{Address: Address{City: "Austin", State: "TX"}}
372372
err = DB.Save(&user).Error
373373
tests.AssertEqual(t, err, nil)
374374

375-
// Update embedded field only
376-
err = DB.Model(&user).Updates(User{Address: Address{City: "Houston"}}).Error
375+
err = DB.Model(&user).Updates(UserWithAddress{Address: Address{City: "Houston"}}).Error
377376
tests.AssertEqual(t, err, nil)
378377

379-
var loaded User
378+
var loaded UserWithAddress
380379
err = DB.First(&loaded, user.ID).Error
381380
tests.AssertEqual(t, err, nil)
382381

0 commit comments

Comments
 (0)