2020-05-23 11:08:50 +03:00
package tests_test
import (
"testing"
. "github.com/jinzhu/gorm/tests"
)
func TestCreate ( t * testing . T ) {
2020-05-23 11:38:13 +03:00
var user = * GetUser ( "create" , Config { } )
2020-05-23 11:08:50 +03:00
if err := DB . Create ( & user ) . Error ; err != nil {
t . Fatalf ( "errors happened when create: %v" , err )
}
if user . ID == 0 {
t . Errorf ( "user's primary key should has value after create, got : %v" , user . ID )
}
if user . CreatedAt . IsZero ( ) {
t . Errorf ( "user's created at should be not zero" )
}
if user . UpdatedAt . IsZero ( ) {
t . Errorf ( "user's updated at should be not zero" )
}
var newUser User
if err := DB . Where ( "id = ?" , user . ID ) . First ( & newUser ) . Error ; err != nil {
2020-05-23 11:38:13 +03:00
t . Fatalf ( "errors happened when query: %v" , err )
2020-05-23 11:08:50 +03:00
} else {
2020-05-23 11:38:13 +03:00
CheckUser ( t , newUser , user )
2020-05-23 11:08:50 +03:00
}
}
2020-05-23 11:38:13 +03:00
func TestCreateWithAssociations ( t * testing . T ) {
2020-05-23 16:03:28 +03:00
var user = * GetUser ( "create_with_associations" , Config {
2020-05-23 11:38:13 +03:00
Account : true ,
Pets : 2 ,
Toys : 3 ,
Company : true ,
Manager : true ,
Team : 4 ,
Languages : 3 ,
Friends : 1 ,
} )
if err := DB . Create ( & user ) . Error ; err != nil {
t . Fatalf ( "errors happened when create: %v" , err )
2020-05-23 11:08:50 +03:00
}
2020-05-23 11:38:13 +03:00
CheckUser ( t , user , user )
2020-05-23 11:08:50 +03:00
2020-05-23 11:38:13 +03:00
var user2 User
2020-05-23 16:03:28 +03:00
DB . Debug ( ) . Preload ( "Account" ) . Preload ( "Pets" ) . Preload ( "Toys" ) . Preload ( "Company" ) . Preload ( "Manager" ) . Preload ( "Team" ) . Preload ( "Languages" ) . Preload ( "Friends" ) . Find ( & user2 , "id = ?" , user . ID )
2020-05-23 11:38:13 +03:00
CheckUser ( t , user2 , user )
}
2020-05-23 11:08:50 +03:00
2020-05-23 16:03:28 +03:00
func TestBulkCreateWithAssociations ( t * testing . T ) {
users := [ ] User {
* GetUser ( "bulk_1" , Config { Account : true , Pets : 2 , Toys : 3 , Company : true , Manager : true , Team : 0 , Languages : 1 , Friends : 1 } ) ,
* GetUser ( "bulk_2" , Config { Account : false , Pets : 2 , Toys : 4 , Company : false , Manager : false , Team : 1 , Languages : 3 , Friends : 5 } ) ,
* GetUser ( "bulk_3" , Config { Account : true , Pets : 0 , Toys : 3 , Company : true , Manager : false , Team : 4 , Languages : 0 , Friends : 1 } ) ,
* GetUser ( "bulk_4" , Config { Account : true , Pets : 3 , Toys : 0 , Company : false , Manager : true , Team : 0 , Languages : 3 , Friends : 0 } ) ,
* GetUser ( "bulk_5" , Config { Account : false , Pets : 0 , Toys : 3 , Company : true , Manager : false , Team : 1 , Languages : 3 , Friends : 1 } ) ,
* GetUser ( "bulk_6" , Config { Account : true , Pets : 4 , Toys : 3 , Company : false , Manager : true , Team : 1 , Languages : 3 , Friends : 0 } ) ,
* GetUser ( "bulk_7" , Config { Account : true , Pets : 1 , Toys : 3 , Company : true , Manager : true , Team : 4 , Languages : 3 , Friends : 1 } ) ,
* GetUser ( "bulk_8" , Config { Account : false , Pets : 0 , Toys : 0 , Company : false , Manager : false , Team : 0 , Languages : 0 , Friends : 0 } ) ,
}
if err := DB . Create ( & users ) . Error ; err != nil {
t . Fatalf ( "errors happened when create: %v" , err )
}
var userIDs [ ] uint
for _ , user := range users {
userIDs = append ( userIDs , user . ID )
CheckUser ( t , user , user )
}
var users2 [ ] User
DB . Preload ( "Account" ) . Preload ( "Pets" ) . Preload ( "Toys" ) . Preload ( "Company" ) . Preload ( "Manager" ) . Preload ( "Team" ) . Preload ( "Languages" ) . Preload ( "Friends" ) . Find ( & users2 , "id IN ?" , userIDs )
for idx , user := range users2 {
CheckUser ( t , user , users [ idx ] )
}
}
2020-05-23 11:38:13 +03:00
// func TestBulkCreatePtrDataWithBelongsTo(t *testing.T) {
// users := []*User{
// GetUser("create_with_belongs_to_1", Config{Company: true, Manager: true}),
// GetUser("create_with_belongs_to_2", Config{Company: true, Manager: false}),
// GetUser("create_with_belongs_to_3", Config{Company: false, Manager: true}),
// GetUser("create_with_belongs_to_4", Config{Company: true, Manager: true}),
// }
2020-05-23 11:08:50 +03:00
2020-05-23 11:38:13 +03:00
// if err := DB.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// var userIDs []uint
// for _, user := range users {
// userIDs = append(userIDs, user.ID)
// CheckUser(t, *user, *user)
// }
// var users2 []User
// DB.Preload("Company").Preload("Manager").Find(&users2, "id IN ?", userIDs)
// for idx, user := range users2 {
// CheckUser(t, user, *users[idx])
// }
// }
// func TestBulkCreateWithoutPtrWithBelongsTo(t *testing.T) {
// users := []*User{
// GetUser("create_with_belongs_to_1", Config{Company: true, Manager: true}),
// GetUser("create_with_belongs_to_2", Config{Company: true, Manager: false}),
// GetUser("create_with_belongs_to_3", Config{Company: false, Manager: true}),
// GetUser("create_with_belongs_to_4", Config{Company: true, Manager: true}),
// }
// if err := DB.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// var userIDs []uint
// for _, user := range users {
// userIDs = append(userIDs, user.ID)
// CheckUser(t, *user, *user)
// }
// }
2020-05-23 11:08:50 +03:00
// func TestCreateHasOneAssociations(t *testing.T, db *gorm.DB) {
// check := func(t *testing.T, user User, old User) {
// if user.Account.ID == 0 {
// t.Errorf("Account should be saved")
// } else if user.Account.UserID.Int64 != int64(user.ID) {
// t.Errorf("Account's foreign key should be saved")
// } else {
// var account Account
// db.First(&account, "id = ?", user.Account.ID)
// if account.Number != user.Account.Number {
// t.Errorf("Account's number should be same")
// } else if user.Account.Number != old.Account.Number {
// t.Errorf("Account's number should be same")
// }
// }
// }
// t.Run("HasOne", func(t *testing.T) {
// var user = User{
// Name: "create",
// Age: 18,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association"},
// }
// if err := db.Create(&user).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// check(t, user, user)
// var user2 User
// db.Preload("Account").Find(&user2, "id = ?", user.ID)
// check(t, user2, user)
// })
// t.Run("HasOneForBulkInsert", func(t *testing.T) {
// var users = []User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-1"},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-2"},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-3"},
// }}
// if err := db.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// var userIDs []uint
// for _, user := range users {
// userIDs = append(userIDs, user.ID)
// check(t, user, user)
// }
// var users2 []User
// db.Preload("Account").Find(&users2, "id IN (?)", userIDs)
// for idx, user := range users2 {
// check(t, user, users[idx])
// }
// })
// t.Run("HasOneForBulkInsertPtrData", func(t *testing.T) {
// var users = []*User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-1"},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-2"},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-3"},
// }}
// if err := db.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, user := range users {
// check(t, *user, *user)
// }
// })
// t.Run("HasOneForBulkInsertWithoutPtr", func(t *testing.T) {
// var users = []User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-1"},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-2"},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Account: Account{Number: "account-has-one-association-3"},
// }}
// if err := db.Create(users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, user := range users {
// check(t, user, user)
// }
// })
// checkPet := func(t *testing.T, pet Pet, old Pet) {
// if pet.Toy.OwnerID != fmt.Sprint(pet.ID) || pet.Toy.OwnerType != "pets" {
// t.Errorf("Failed to create polymorphic has one association - toy owner id %v, owner type %v", pet.Toy.OwnerID, pet.Toy.OwnerType)
// } else {
// var toy Toy
// db.First(&toy, "owner_id = ? and owner_type = ?", pet.Toy.OwnerID, pet.Toy.OwnerType)
// if toy.Name != pet.Toy.Name {
// t.Errorf("Failed to query saved polymorphic has one association")
// } else if old.Toy.Name != pet.Toy.Name {
// t.Errorf("Failed to query saved polymorphic has one association")
// }
// }
// }
// t.Run("PolymorphicHasOne", func(t *testing.T) {
// var pet = Pet{
// Name: "create",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic"},
// }
// if err := db.Create(&pet).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// checkPet(t, pet, pet)
// var pet2 Pet
// db.Preload("Toy").Find(&pet2, "id = ?", pet.ID)
// checkPet(t, pet2, pet)
// })
// t.Run("PolymorphicHasOneForBulkInsert", func(t *testing.T) {
// var pets = []Pet{{
// Name: "create-1",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-1"},
// }, {
// Name: "create-2",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-2"},
// }, {
// Name: "create-3",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-3"},
// }}
// if err := db.Create(&pets).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// var petIDs []uint
// for _, pet := range pets {
// petIDs = append(petIDs, pet.ID)
// checkPet(t, pet, pet)
// }
// var pets2 []Pet
// db.Preload("Toy").Find(&pets2, "id IN (?)", petIDs)
// for idx, pet := range pets2 {
// checkPet(t, pet, pets[idx])
// }
// })
// t.Run("PolymorphicHasOneForBulkInsertPtrData", func(t *testing.T) {
// var pets = []*Pet{{
// Name: "create-1",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-1"},
// }, {
// Name: "create-2",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-2"},
// }, {
// Name: "create-3",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-3"},
// }}
// if err := db.Create(&pets).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, pet := range pets {
// checkPet(t, *pet, *pet)
// }
// })
// t.Run("PolymorphicHasOneForBulkInsertWithoutPtr", func(t *testing.T) {
// var pets = []*Pet{{
// Name: "create-1",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-1"},
// }, {
// Name: "create-2",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-2"},
// }, {
// Name: "create-3",
// Toy: Toy{Name: "Create-HasOneAssociation-Polymorphic-3"},
// }}
// if err := db.Create(pets).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, pet := range pets {
// checkPet(t, *pet, *pet)
// }
// })
// }
// func TestCreateHasManyAssociations(t *testing.T, db *gorm.DB) {
// check := func(t *testing.T, user User, old User) {
// for idx, pet := range user.Pets {
// if pet.ID == 0 {
// t.Errorf("Pet's foreign key should be saved")
// }
// var result Pet
// db.First(&result, "id = ?", pet.ID)
// if result.Name != pet.Name {
// t.Errorf("Pet's name should be same")
// } else if result.UserID != user.ID {
// t.Errorf("Pet's foreign key should be saved")
// } else if result.Name != old.Pets[idx].Name {
// t.Errorf("Pet's name should be same")
// }
// }
// }
// t.Run("HasMany", func(t *testing.T) {
// var user = User{
// Name: "create",
// Age: 18,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet1"}, {Name: "pet2"}},
// }
// if err := db.Create(&user).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// check(t, user, user)
// var user2 User
// db.Preload("Pets").Find(&user2, "id = ?", user.ID)
// check(t, user2, user)
// })
// t.Run("HasManyForBulkInsert", func(t *testing.T) {
// var users = []User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-1-1"}, {Name: "pet-1-2"}},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-2-1"}},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-3-1"}, {Name: "pet-3-2"}},
// }}
// if err := db.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// var userIDs []uint
// for _, user := range users {
// userIDs = append(userIDs, user.ID)
// check(t, user, user)
// }
// var users2 []User
// db.Preload("Pets").Find(&users2, "id IN (?)", userIDs)
// for idx, user := range users2 {
// check(t, user, users[idx])
// }
// })
// t.Run("HasManyForBulkInsertPtrData", func(t *testing.T) {
// var users = []*User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-1-1"}, {Name: "pet-1-2"}},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-2-1"}},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-3-1"}, {Name: "pet-3-2"}},
// }}
// if err := db.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, user := range users {
// check(t, *user, *user)
// }
// })
// t.Run("HasManyForBulkInsertWithoutPtr", func(t *testing.T) {
// var users = []*User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-1-1"}, {Name: "pet-1-2"}},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-2-1"}},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Pets: []*Pet{{Name: "pet-3-1"}, {Name: "pet-3-2"}},
// }}
// if err := db.Create(users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, user := range users {
// check(t, *user, *user)
// }
// })
// checkToy := func(t *testing.T, user User, old User) {
// for idx, toy := range user.Toys {
// if toy.ID == 0 {
// t.Fatalf("Failed to create toy #%v", idx)
// }
// var result Toy
// db.First(&result, "id = ?", toy.ID)
// if result.Name != toy.Name {
// t.Errorf("Failed to query saved toy")
// } else if result.Name != old.Toys[idx].Name {
// t.Errorf("Failed to query saved toy")
// } else if result.OwnerID != fmt.Sprint(user.ID) || result.OwnerType != "users" {
// t.Errorf("Failed to save relation")
// }
// }
// }
// t.Run("PolymorphicHasMany", func(t *testing.T) {
// var user = User{
// Name: "create",
// Age: 18,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy1"}, {Name: "toy2"}},
// }
// if err := db.Create(&user).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// checkToy(t, user, user)
// var user2 User
// db.Preload("Toys").Find(&user2, "id = ?", user.ID)
// check(t, user2, user)
// })
// t.Run("PolymorphicHasManyForBulkInsert", func(t *testing.T) {
// var users = []User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-1-1"}, {Name: "toy-1-2"}},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-2-1"}, {Name: "toy-2-2"}},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-3-1"}, {Name: "toy-3-2"}},
// }}
// if err := db.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// var userIDs []uint
// for _, user := range users {
// userIDs = append(userIDs, user.ID)
// checkToy(t, user, user)
// }
// var users2 []User
// db.Preload("Toys").Find(&users2, "id IN (?)", userIDs)
// for idx, user := range users2 {
// check(t, user, users[idx])
// }
// })
// t.Run("PolymorphicHasManyForBulkInsertPtrData", func(t *testing.T) {
// var users = []*User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-1-1"}, {Name: "toy-1-2"}},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-2-1"}, {Name: "toy-2-2"}},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-3-1"}, {Name: "toy-3-2"}},
// }}
// if err := db.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, user := range users {
// checkToy(t, *user, *user)
// }
// })
// t.Run("PolymorphicHasManyForBulkInsertWithoutPtr", func(t *testing.T) {
// var users = []User{{
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-1-1"}, {Name: "toy-1-2"}},
// }, {
// Name: "create-2",
// Age: 28,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-2-1"}, {Name: "toy-2-2"}},
// }, {
// Name: "create-3",
// Age: 38,
// Birthday: Now(),
// Toys: []Toy{{Name: "toy-3-1"}, {Name: "toy-3-2"}},
// }}
// if err := db.Create(users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// for _, user := range users {
// checkToy(t, user, user)
// }
// })
// }
// func TestCreateMany2ManyAssociations(t *testing.T, db *gorm.DB) {
// check := func(t *testing.T, user User, old User) {
// for idx, language := range user.Languages {
// var result Language
// db.First(&result, "code = ?", language.Code)
// if result.Name != language.Name {
// t.Errorf("Language's name should be same")
// } else if result.Name != old.Languages[idx].Name {
// t.Errorf("Language's name should be same")
// }
// }
// for idx, f := range user.Friends {
// if f.ID == 0 {
// t.Errorf("Friend's foreign key should be saved")
// }
// var result User
// db.First(&result, "id = ?", f.ID)
// if result.Name != f.Name {
// t.Errorf("Friend's name should be same")
// } else if result.Name != old.Friends[idx].Name {
// t.Errorf("Language's name should be same")
// }
// }
// }
// db.Create(&[]Language{{Code: "zh-CN", Name: "Chinese"}, {Code: "en", Name: "English"}})
// t.Run("Many2Many", func(t *testing.T) {
// var user = User{
// Name: "create",
// Age: 18,
// Birthday: Now(),
// Languages: []Language{{Code: "zh-CN", Name: "Chinese"}, {Code: "en", Name: "English"}},
// Friends: []*User{{Name: "friend-1"}, {Name: "friend-2"}},
// }
// if err := db.Create(&user).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// check(t, user, user)
// var user2 User
// db.Preload("Languages").Preload("Friends").Find(&user2, "id = ?", user.ID)
// check(t, user2, user)
// })
// t.Run("Many2ManyForBulkInsert", func(t *testing.T) {
// var users = []User{
// {
// Name: "create-1",
// Age: 18,
// Birthday: Now(),
// Languages: []Language{{Code: "zh-CN", Name: "Chinese"}, {Code: "en", Name: "English"}},
// Friends: []*User{{Name: "friend-1-1"}, {Name: "friend-1-2"}},
// },
// {
// Name: "create-2",
// Age: 18,
// Birthday: Now(),
// Languages: []Language{{Code: "zh-CN", Name: "Chinese"}},
// Friends: []*User{{Name: "friend-2-1"}},
// },
// {
// Name: "create-3",
// Age: 18,
// Birthday: Now(),
// Languages: []Language{{Code: "zh-CN", Name: "Chinese"}, {Code: "en", Name: "English"}},
// Friends: []*User{{Name: "friend-3-1"}, {Name: "friend-3-2"}, {Name: "friend-3-3"}},
// },
// }
// if err := db.Create(&users).Error; err != nil {
// t.Fatalf("errors happened when create: %v", err)
// }
// var userIDs []uint
// for _, user := range users {
// userIDs = append(userIDs, user.ID)
// check(t, user, user)
// }
// var users2 []User
// db.Preload("Languages").Preload("Friends").Find(&users2, "id IN (?)", userIDs)
// for idx, user := range users2 {
// check(t, user, users[idx])
// }
// })
// }