mirror of https://github.com/go-gorm/gorm.git
Refactor create tests
This commit is contained in:
Jinzhu
parent
5ec4fee797
commit
590f622674
|
|
@ -109,12 +109,11 @@ func GetIdentityFieldValuesMap(reflectValue reflect.Value, fields []*Field) (map
|
|||
|
||||
dataResults[utils.ToStringKey(results[0]...)] = []reflect.Value{reflectValue}
|
||||
case reflect.Slice, reflect.Array:
|
||||
fieldValues := make([]interface{}, len(fields))
|
||||
|
||||
for i := 0; i < reflectValue.Len(); i++ {
|
||||
fieldValues := make([]interface{}, len(fields))
|
||||
notZero = false
|
||||
for idx, field := range fields {
|
||||
fieldValues[idx], zero = field.ValueOf(reflectValue.Index(idx))
|
||||
fieldValues[idx], zero = field.ValueOf(reflectValue.Index(i))
|
||||
notZero = notZero || !zero
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -66,6 +66,15 @@ func GetUser(name string, config Config) *User {
|
|||
}
|
||||
|
||||
func CheckPet(t *testing.T, pet Pet, expect Pet) {
|
||||
if pet.ID != 0 {
|
||||
var newPet Pet
|
||||
if err := DB.Where("id = ?", pet.ID).First(&newPet).Error; err != nil {
|
||||
t.Fatalf("errors happened when query: %v", err)
|
||||
} else {
|
||||
AssertObjEqual(t, newPet, pet, "ID", "CreatedAt", "UpdatedAt", "DeletedAt", "UserID", "Name")
|
||||
}
|
||||
}
|
||||
|
||||
AssertObjEqual(t, pet, expect, "ID", "CreatedAt", "UpdatedAt", "DeletedAt", "UserID", "Name")
|
||||
|
||||
AssertObjEqual(t, pet.Toy, expect.Toy, "ID", "CreatedAt", "UpdatedAt", "DeletedAt", "Name", "OwnerID", "OwnerType")
|
||||
|
|
|
|||
|
|
@ -52,7 +52,7 @@ func TestCreateWithAssociations(t *testing.T) {
|
|||
CheckUser(t, user, user)
|
||||
|
||||
var user2 User
|
||||
DB.Debug().Preload("Account").Preload("Pets").Preload("Toys").Preload("Company").Preload("Manager").Preload("Team").Preload("Languages").Preload("Friends").Find(&user2, "id = ?", user.ID)
|
||||
DB.Preload("Account").Preload("Pets").Preload("Toys").Preload("Company").Preload("Manager").Preload("Team").Preload("Languages").Preload("Friends").Find(&user2, "id = ?", user.ID)
|
||||
CheckUser(t, user2, user)
|
||||
}
|
||||
|
||||
|
|
@ -85,620 +85,100 @@ func TestBulkCreateWithAssociations(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
// 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}),
|
||||
// }
|
||||
|
||||
// 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)
|
||||
// }
|
||||
// }
|
||||
|
||||
// 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])
|
||||
// }
|
||||
// })
|
||||
// }
|
||||
func TestBulkCreatePtrDataWithAssociations(t *testing.T) {
|
||||
users := []*User{
|
||||
GetUser("bulk_ptr_1", Config{Account: true, Pets: 2, Toys: 3, Company: true, Manager: true, Team: 0, Languages: 1, Friends: 1}),
|
||||
GetUser("bulk_ptr_2", Config{Account: false, Pets: 2, Toys: 4, Company: false, Manager: false, Team: 1, Languages: 3, Friends: 5}),
|
||||
GetUser("bulk_ptr_3", Config{Account: true, Pets: 0, Toys: 3, Company: true, Manager: false, Team: 4, Languages: 0, Friends: 1}),
|
||||
GetUser("bulk_ptr_4", Config{Account: true, Pets: 3, Toys: 0, Company: false, Manager: true, Team: 0, Languages: 3, Friends: 0}),
|
||||
GetUser("bulk_ptr_5", Config{Account: false, Pets: 0, Toys: 3, Company: true, Manager: false, Team: 1, Languages: 3, Friends: 1}),
|
||||
GetUser("bulk_ptr_6", Config{Account: true, Pets: 4, Toys: 3, Company: false, Manager: true, Team: 1, Languages: 3, Friends: 0}),
|
||||
GetUser("bulk_ptr_7", Config{Account: true, Pets: 1, Toys: 3, Company: true, Manager: true, Team: 4, Languages: 3, Friends: 1}),
|
||||
GetUser("bulk_ptr_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])
|
||||
}
|
||||
}
|
||||
|
||||
func TestPolymorphicHasOne(t *testing.T) {
|
||||
t.Run("Struct", func(t *testing.T) {
|
||||
var pet = Pet{
|
||||
Name: "PolymorphicHasOne",
|
||||
Toy: Toy{Name: "Toy-PolymorphicHasOne"},
|
||||
}
|
||||
|
||||
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("Slice", func(t *testing.T) {
|
||||
var pets = []Pet{{
|
||||
Name: "PolymorphicHasOne-Slice-1",
|
||||
Toy: Toy{Name: "Toy-PolymorphicHasOne-Slice-1"},
|
||||
}, {
|
||||
Name: "PolymorphicHasOne-Slice-2",
|
||||
Toy: Toy{Name: "Toy-PolymorphicHasOne-Slice-2"},
|
||||
}, {
|
||||
Name: "PolymorphicHasOne-Slice-3",
|
||||
Toy: Toy{Name: "Toy-PolymorphicHasOne-Slice-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("SliceOfPtr", func(t *testing.T) {
|
||||
var pets = []*Pet{{
|
||||
Name: "PolymorphicHasOne-Slice-1",
|
||||
Toy: Toy{Name: "Toy-PolymorphicHasOne-Slice-1"},
|
||||
}, {
|
||||
Name: "PolymorphicHasOne-Slice-2",
|
||||
Toy: Toy{Name: "Toy-PolymorphicHasOne-Slice-2"},
|
||||
}, {
|
||||
Name: "PolymorphicHasOne-Slice-3",
|
||||
Toy: Toy{Name: "Toy-PolymorphicHasOne-Slice-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)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
|
|
|||
|
|
@ -3,6 +3,7 @@ package utils
|
|||
import (
|
||||
"database/sql/driver"
|
||||
"fmt"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"regexp"
|
||||
"runtime"
|
||||
|
|
@ -11,8 +12,13 @@ import (
|
|||
"unicode"
|
||||
)
|
||||
|
||||
var goSrcRegexp = regexp.MustCompile(`jinzhu/gorm(@.*)?/.*.go`)
|
||||
var goTestRegexp = regexp.MustCompile(`jinzhu/gorm(@.*)?/.*test.go`)
|
||||
var goSrcRegexp, goTestRegexp *regexp.Regexp
|
||||
|
||||
func init() {
|
||||
_, file, _, _ := runtime.Caller(0)
|
||||
goSrcRegexp = regexp.MustCompile(filepath.Join(filepath.Dir(filepath.Dir(file)), ".*.go"))
|
||||
goTestRegexp = regexp.MustCompile(filepath.Join(filepath.Dir(filepath.Dir(file)), ".*test.go"))
|
||||
}
|
||||
|
||||
func FileWithLineNum() string {
|
||||
for i := 2; i < 15; i++ {
|
||||
|
|
|
|||
Loading…
Reference in New Issue