forked from mirror/gorm
44 lines
976 B
Go
44 lines
976 B
Go
|
package tests_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
. "github.com/jinzhu/gorm/tests"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestUpdateHasOne(t *testing.T) {
|
||
|
|
var user = *GetUser("update-has-one", Config{})
|
||
|
|
|
||
|
|
if err := DB.Create(&user).Error; err != nil {
|
||
|
|
t.Fatalf("errors happened when create: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
user.Account = Account{Number: "account-has-one-association"}
|
||
|
|
|
||
|
|
if err := DB.Save(&user).Error; err != nil {
|
||
|
|
t.Fatalf("errors happened when update: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var user2 User
|
||
|
|
DB.Preload("Account").Find(&user2, "id = ?", user.ID)
|
||
|
|
CheckUser(t, user2, user)
|
||
|
|
|
||
|
|
t.Run("Polymorphic", func(t *testing.T) {
|
||
|
|
var pet = Pet{Name: "create"}
|
||
|
|
|
||
|
|
if err := DB.Create(&pet).Error; err != nil {
|
||
|
|
t.Fatalf("errors happened when create: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
pet.Toy = Toy{Name: "Update-HasOneAssociation-Polymorphic"}
|
||
|
|
|
||
|
|
if err := DB.Save(&pet).Error; err != nil {
|
||
|
|
t.Fatalf("errors happened when create: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var pet2 Pet
|
||
|
|
DB.Preload("Toy").Find(&pet2, "id = ?", pet.ID)
|
||
|
|
CheckPet(t, pet2, pet)
|
||
|
|
})
|
||
|
|
}
|