| 89 | } |
| 90 | |
| 91 | func TestCreateInBatchesWithDefaultSize(t *testing.T) { |
| 92 | users := []User{ |
| 93 | *GetUser("create_with_default_batch_size_1", Config{Account: true, Pets: 2, Toys: 3, Company: true, Manager: true, Team: 0, Languages: 1, Friends: 1}), |
| 94 | *GetUser("create_with_default_batch_sizs_2", Config{Account: false, Pets: 2, Toys: 4, Company: false, Manager: false, Team: 1, Languages: 3, Friends: 5}), |
| 95 | *GetUser("create_with_default_batch_sizs_3", Config{Account: true, Pets: 0, Toys: 3, Company: true, Manager: false, Team: 4, Languages: 0, Friends: 1}), |
| 96 | *GetUser("create_with_default_batch_sizs_4", Config{Account: true, Pets: 3, Toys: 0, Company: false, Manager: true, Team: 0, Languages: 3, Friends: 0}), |
| 97 | *GetUser("create_with_default_batch_sizs_5", Config{Account: false, Pets: 0, Toys: 3, Company: true, Manager: false, Team: 1, Languages: 3, Friends: 1}), |
| 98 | *GetUser("create_with_default_batch_sizs_6", Config{Account: true, Pets: 4, Toys: 3, Company: false, Manager: true, Team: 1, Languages: 3, Friends: 0}), |
| 99 | } |
| 100 | |
| 101 | result := DB.Session(&gorm.Session{CreateBatchSize: 2}).Create(&users) |
| 102 | if result.RowsAffected != int64(len(users)) { |
| 103 | t.Errorf("affected rows should be %v, but got %v", len(users), result.RowsAffected) |
| 104 | } |
| 105 | |
| 106 | for _, user := range users { |
| 107 | if user.ID == 0 { |
| 108 | t.Fatalf("failed to fill user's ID, got %v", user.ID) |
| 109 | } else { |
| 110 | var newUser User |
| 111 | if err := DB.Where("id = ?", user.ID).Preload(clause.Associations).First(&newUser).Error; err != nil { |
| 112 | t.Fatalf("errors happened when query: %v", err) |
| 113 | } else { |
| 114 | CheckUser(t, newUser, user) |
| 115 | } |
| 116 | } |
| 117 | } |
| 118 | } |
| 119 | |
| 120 | func TestCreateFromMap(t *testing.T) { |
| 121 | if err := DB.Model(&User{}).Create(map[string]interface{}{"Name": "create_from_map", "Age": 18}).Error; err != nil { |