Refact tests

This commit is contained in:
Jinzhu 2013-11-16 22:26:34 +08:00
parent ba6403f904
commit c9e8c8d16c
1 changed files with 158 additions and 189 deletions

View File

@ -105,28 +105,23 @@ func init() {
db.Exec("drop table addresses") db.Exec("drop table addresses")
db.Exec("drop table credit_cards") db.Exec("drop table credit_cards")
err = db.CreateTable(&User{}).Error if err = db.CreateTable(&User{}).Error; err != nil {
if err != nil {
panic(fmt.Sprintf("No error should happen when create table, but got %+v", err)) panic(fmt.Sprintf("No error should happen when create table, but got %+v", err))
} }
err = db.CreateTable(&Product{}).Error if err = db.CreateTable(&Product{}).Error; err != nil {
if err != nil {
panic(fmt.Sprintf("No error should happen when create table, but got %+v", err)) panic(fmt.Sprintf("No error should happen when create table, but got %+v", err))
} }
err = db.CreateTable(Email{}).Error if err = db.CreateTable(Email{}).Error; err != nil {
if err != nil {
panic(fmt.Sprintf("No error should happen when create table, but got %+v", err)) panic(fmt.Sprintf("No error should happen when create table, but got %+v", err))
} }
err = db.CreateTable(Address{}).Error if err = db.AutoMigrate(Address{}).Error; err != nil {
if err != nil {
panic(fmt.Sprintf("No error should happen when create table, but got %+v", err)) panic(fmt.Sprintf("No error should happen when create table, but got %+v", err))
} }
err = db.CreateTable(CreditCard{}).Error if err = db.AutoMigrate(&CreditCard{}).Error; err != nil {
if err != nil {
panic(fmt.Sprintf("No error should happen when create table, but got %+v", err)) panic(fmt.Sprintf("No error should happen when create table, but got %+v", err))
} }
@ -143,28 +138,11 @@ func init() {
db.Save(&User{Name: "5", Age: 26, Birthday: t4}) db.Save(&User{Name: "5", Age: 26, Birthday: t4})
} }
func TestSaveAndFind(t *testing.T) {
name := "save_and_find"
u := &User{Name: name, Age: 1}
db.Save(u)
if u.Id == 0 {
t.Errorf("Should have ID after create record")
}
user := &User{}
db.First(user, "name = ?", name)
if user.Name != name {
t.Errorf("User should be saved and fetched correctly")
}
users := []User{}
db.Find(&users)
}
func TestFirstAndLast(t *testing.T) { func TestFirstAndLast(t *testing.T) {
var user1, user2, user3, user4 User var user1, user2, user3, user4 User
db.First(&user1) db.First(&user1)
db.Order("id").Find(&user2) db.Order("id").Find(&user2)
db.Last(&user3) db.Last(&user3)
db.Order("id desc").Find(&user4) db.Order("id desc").Find(&user4)
if user1.Id != user2.Id || user3.Id != user4.Id { if user1.Id != user2.Id || user3.Id != user4.Id {
@ -178,28 +156,28 @@ func TestFirstAndLast(t *testing.T) {
} }
} }
func TestSaveAndUpdate(t *testing.T) { func TestCreateAndUpdate(t *testing.T) {
name, name2, new_name := "update", "update2", "new_update" name, name2, new_name := "update", "update2", "new_update"
user := User{Name: name, Age: 1} user := User{Name: name, Age: 1}
db.Save(&user) db.Save(&user)
db.Save(&User{Name: name2, Age: 1})
if user.Id == 0 { if user.Id == 0 {
t.Errorf("User Id should exist after create") t.Errorf("Should have ID after create")
} }
db.Save(&User{Name: name2, Age: 1})
user.Name = new_name user.Name = new_name
db.Save(&user) db.Save(&user)
if db.Where("name = ?", name).First(&User{}).Error != RecordNotFound { if db.Where("name = ?", name).First(&User{}).Error != RecordNotFound {
t.Errorf("Should raise error when looking for a existing user with an outdated name") t.Errorf("Should raise RecordNotFound error when looking with an outdated name")
} }
if db.Where("name = ?", new_name).First(&User{}).Error != nil { if db.Where("name = ?", new_name).First(&User{}).Error != nil {
t.Errorf("Shouldn't raise error when looking for a existing user with the new name") t.Errorf("Shouldn't raise error when looking with the new name")
} }
if db.Where("name = ?", name2).First(&User{}).Error != nil { if db.Where("name = ?", name2).First(&User{}).Error != nil {
t.Errorf("Shouldn't update other users") t.Errorf("Shouldn't update other users when update one")
} }
} }
@ -208,16 +186,17 @@ func TestDelete(t *testing.T) {
user := User{Name: name, Age: 1} user := User{Name: name, Age: 1}
db.Save(&user) db.Save(&user)
db.Save(&User{Name: name2, Age: 1}) db.Save(&User{Name: name2, Age: 1})
orm := db.Delete(&user)
orm = db.Where("name = ?", name).First(&User{}) if db.Delete(&user).Error != nil {
if orm.Error == nil { t.Errorf("Shouldn't raise any error when delete a user")
t.Errorf("User should be deleted successfully")
} }
orm = db.Where("name = ?", name2).First(&User{}) if db.Where("name = ?", name).First(&User{}).Error == nil {
if orm.Error != nil { t.Errorf("User can't be found after delete")
t.Errorf("User2 should not be deleted") }
if db.Where("name = ?", name2).First(&User{}).Error != nil {
t.Errorf("Other users shouldn't be deleted")
} }
} }
@ -228,38 +207,32 @@ func TestWhere(t *testing.T) {
user := &User{} user := &User{}
db.Where("name = ?", name).First(user) db.Where("name = ?", name).First(user)
if user.Name != name { if user.Name != name {
t.Errorf("Should found out user with name '%v'", name) t.Errorf("Search user with name")
} }
if db.Where(user.Id).First(&User{}).Error != nil { if db.Where(user.Id).First(&User{}).Error != nil {
t.Errorf("Should found out users only with id") t.Errorf("Search user with primary key")
} }
user = &User{} if db.Where("name LIKE ?", "%nonono%").First(&User{}).Error == nil {
orm := db.Where("name LIKE ?", "%nonono%").First(user) t.Errorf("Search non-existing user with regexp name")
if orm.Error == nil {
t.Errorf("Should return error when searching for none existing record, %+v", user)
} }
user = &User{} if db.Where("name LIKE ?", "%whe%").First(&User{}).Error != nil {
orm = db.Where("name LIKE ?", "%whe%").First(user) t.Errorf("Search user with regexp name")
if orm.Error != nil {
t.Errorf("Should not return error when searching for existing record, %+v", user)
} }
user = &User{} if db.Where("name = ?", "non-existing user").First(&User{}).Error == nil {
orm = db.Where("name = ?", "noexisting-user").First(user) t.Errorf("Search non-existing user should get error")
if orm.Error == nil {
t.Errorf("Should return error when looking for none existing record, %+v", user)
} }
users := []User{} var users []User
orm = db.Where("name = ?", "none-noexisting").Find(&users) if db.Where("name = ?", "none-noexisting").Find(&users).Error != nil {
if orm.Error != nil { t.Errorf("Shouldn't return error when looking for none existing records with find")
t.Errorf("Shouldn't return error when looking for none existing records, %+v", users)
} }
if len(users) != 0 { if len(users) != 0 {
t.Errorf("Shouldn't find anything when looking for none existing records, %+v", users) t.Errorf("Users should be empty")
} }
} }
@ -267,80 +240,80 @@ func TestComplexWhere(t *testing.T) {
var users []User var users []User
db.Where("age > ?", 20).Find(&users) db.Where("age > ?", 20).Find(&users)
if len(users) != 3 { if len(users) != 3 {
t.Errorf("Should only found 3 users that age > 20, but have %v", len(users)) t.Errorf("Should found 3 users that age > 20, but got %v", len(users))
} }
var user_ids []int64 var user_ids []int64
db.Table("users").Where("age > ?", 20).Pluck("id", &user_ids) db.Table("users").Where("age > ?", 20).Pluck("id", &user_ids)
if len(user_ids) != 3 { if len(user_ids) != 3 {
t.Errorf("Should only found 3 users that age > 20, but have %v", len(users)) t.Errorf("Should found 3 users that age > 20, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where(user_ids).Find(&users) db.Where(user_ids).Find(&users)
if len(users) != 3 { if len(users) != 3 {
t.Errorf("Should only found 3 users that age > 20 when search with id map, but have %v", len(users)) t.Errorf("Should found 3 users that age > 20 when search with primary keys, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("age >= ?", 20).Find(&users) db.Where("age >= ?", 20).Find(&users)
if len(users) != 4 { if len(users) != 4 {
t.Errorf("Should only found 4 users that age >= 20, but have %v", len(users)) t.Errorf("Should found 4 users that age >= 20, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("age = ?", 20).Find(&users) db.Where("age = ?", 20).Find(&users)
if len(users) != 1 { if len(users) != 1 {
t.Errorf("Should only found 1 users age == 20, but have %v", len(users)) t.Errorf("Should found 1 users age == 20, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("age <> ?", 20).Find(&users) db.Where("age <> ?", 20).Find(&users)
if len(users) < 3 { if len(users) < 3 {
t.Errorf("Should have more than 3 users age != 20, but have %v", len(users)) t.Errorf("Should found more than 3 users age != 20, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("name = ? and age >= ?", "3", 20).Find(&users) db.Where("name = ? and age >= ?", "3", 20).Find(&users)
if len(users) != 2 { if len(users) != 2 {
t.Errorf("Should only found 2 users that age >= 20 with name 3, but have %v", len(users)) t.Errorf("Should found 2 users that age >= 20 with name 3, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("name = ?", "3").Where("age >= ?", 20).Find(&users) db.Where("name = ?", "3").Where("age >= ?", 20).Find(&users)
if len(users) != 2 { if len(users) != 2 {
t.Errorf("Should only found 2 users that age >= 20 with name 3, but have %v", len(users)) t.Errorf("Should found 2 users that age >= 20 with name 3, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("birthday > ?", t2).Find(&users) db.Where("birthday > ?", t2).Find(&users)
if len(users) != 3 { if len(users) != 3 {
t.Errorf("Should only found 3 users's birthday >= t2", len(users)) t.Errorf("Should found 3 users's birthday >= t2, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("birthday > ?", "2002-10-10").Find(&users) db.Where("birthday > ?", "2002-10-10").Find(&users)
if len(users) != 3 { if len(users) != 3 {
t.Errorf("Should only found 3 users's birthday >= t2", len(users)) t.Errorf("Should found 3 users's birthday >= 2002-10-10, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("birthday >= ?", t1).Where("birthday < ?", t2).Find(&users) db.Where("birthday >= ?", t1).Where("birthday < ?", t2).Find(&users)
if len(users) != 1 { if len(users) != 1 {
t.Errorf("Should only found 1 users's birthday <= t2, but have %v", len(users)) t.Errorf("Should found 1 users's birthday <= t2, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("birthday >= ? and birthday <= ?", t1, t2).Find(&users) db.Where("birthday >= ? and birthday <= ?", t1, t2).Find(&users)
if len(users) != 2 { if len(users) != 2 {
t.Errorf("Should only found 2 users's birthday <= t2, but have %v", len(users)) t.Errorf("Should found 2 users's birthday <= t2, but got %v", len(users))
} }
users = []User{} users = []User{}
db.Where("name in (?)", []string{"1", "3"}).Find(&users) db.Where("name in (?)", []string{"1", "3"}).Find(&users)
if len(users) != 3 { if len(users) != 3 {
t.Errorf("Should only found 3 users's name in (1, 3), but have %v", len(users)) t.Errorf("Should found 3 users's name in (1, 3), but got %v", len(users))
} }
user_ids = []int64{} user_ids = []int64{}
@ -351,83 +324,72 @@ func TestComplexWhere(t *testing.T) {
users = []User{} users = []User{}
db.Where("id in (?)", user_ids).Find(&users) db.Where("id in (?)", user_ids).Find(&users)
if len(users) != 3 { if len(users) != 3 {
t.Errorf("Should only found 3 users's name in (1, 3) - search by id, but have %v", len(users)) t.Errorf("Should found 3 users's name in (1, 3), but got %v", len(users))
}
users = []User{}
db.Where("name in (?)", []string{"1", "2"}).Find(&users)
if len(users) != 2 {
t.Errorf("Should only found 2 users's name in (1, 2), but have %v", len(users))
}
user_ids = []int64{}
for _, user := range users {
user_ids = append(user_ids, user.Id)
}
users = []User{}
db.Where("id in (?)", user_ids).Find(&users)
if len(users) != 2 {
t.Errorf("Should only found 2 users's name in (1, 2) - search by id, but have %v", len(users))
} }
users = []User{} users = []User{}
db.Where("id in (?)", user_ids[0]).Find(&users) db.Where("id in (?)", user_ids[0]).Find(&users)
if len(users) != 1 { if len(users) != 1 {
t.Errorf("Should only found 1 users's name in (1, 2) - search by the first id, but have %v", len(users)) t.Errorf("Should found 1 users's name in (1), but got %v", len(users))
}
users = []User{}
db.Where("name in (?)", []string{"1", "2"}).Find(&users)
if len(users) != 2 {
t.Errorf("Should found 2 users's name in (1, 2), but got %v", len(users))
} }
} }
func TestWhereWithStruct(t *testing.T) { func TestSearchWithStruct(t *testing.T) {
var user User var user User
db.First(&user, &User{Name: "2"}) db.First(&user, &User{Name: "2"})
if user.Id == 0 || user.Name != "2" { if user.Id == 0 || user.Name != "2" {
t.Errorf("Should be able to search first record with inline struct") t.Errorf("Search first record with inline struct pointer")
} }
db.First(&user, User{Name: "2"}) db.First(&user, User{Name: "2"})
if user.Id == 0 || user.Name != "2" { if user.Id == 0 || user.Name != "2" {
t.Errorf("Should be able to search first record with inline struct") t.Errorf("Search first record with inline struct")
} }
db.Where(&User{Name: "2"}).First(&user) db.Where(&User{Name: "2"}).First(&user)
if user.Id == 0 || user.Name != "2" { if user.Id == 0 || user.Name != "2" {
t.Errorf("Should be able to search first record with where struct") t.Errorf("Search first record with where struct")
} }
var users []User var users []User
db.Find(&users, &User{Name: "3"}) db.Find(&users, &User{Name: "3"})
if len(users) != 2 { if len(users) != 2 {
t.Errorf("Should be able to search all record with inline struct") t.Errorf("Search all records with inline struct")
} }
db.Where(User{Name: "3"}).Find(&users) db.Where(User{Name: "3"}).Find(&users)
if user.Id == 0 || user.Name != "2" { if user.Id == 0 || user.Name != "2" {
t.Errorf("Should be able to search first record with where struct") t.Errorf("Search all records with where struct")
} }
} }
func TestWhereWithInterfaceMap(t *testing.T) { func TestSearchWithMap(t *testing.T) {
var user User var user User
db.First(&user, map[string]interface{}{"name": "2"}) db.First(&user, map[string]interface{}{"name": "2"})
if user.Id == 0 || user.Name != "2" { if user.Id == 0 || user.Name != "2" {
t.Errorf("Should be able to search first record with inline interface map") t.Errorf("Search first record with inline map")
} }
db.Where(map[string]interface{}{"name": "2"}).First(&user) db.Where(map[string]interface{}{"name": "2"}).First(&user)
if user.Id == 0 || user.Name != "2" { if user.Id == 0 || user.Name != "2" {
t.Errorf("Should be able to search first record with where interface map") t.Errorf("Search first record with where map")
} }
var users []User var users []User
db.Find(&users, map[string]interface{}{"name": "3"}) db.Find(&users, map[string]interface{}{"name": "3"})
if len(users) != 2 { if len(users) != 2 {
t.Errorf("Should be able to search all record with inline interface map") t.Errorf("Search all records with inline map")
} }
db.Where(map[string]interface{}{"name": "3"}).Find(&users) db.Where(map[string]interface{}{"name": "3"}).Find(&users)
if user.Id == 0 || user.Name != "2" { if user.Id == 0 || user.Name != "2" {
t.Errorf("Should be able to search first record with where interface map") t.Errorf("Search all records with where map")
} }
} }
@ -437,28 +399,28 @@ func TestInitlineCondition(t *testing.T) {
db.Where("name = ?", "3").First(&u3, "age = 22").First(&u4, "age = ?", 24).First(&u5, "age = ?", 26) db.Where("name = ?", "3").First(&u3, "age = 22").First(&u4, "age = ?", 24).First(&u5, "age = ?", 26)
if !((u5.Id == 0) && (u3.Age == 22 && u3.Name == "3") && (u4.Age == 24 && u4.Name == "3")) { if !((u5.Id == 0) && (u3.Age == 22 && u3.Name == "3") && (u4.Age == 24 && u4.Name == "3")) {
t.Errorf("Inline where condition for first when search") t.Errorf("Find first record with inline condition and where")
}
db.First(&u6, u1.Id)
if !(u6.Id == u1.Id && u6.Id != 0) {
t.Errorf("Find first record with primary key")
}
db.First(&u7, strconv.Itoa(int(u1.Id)))
if !(u6.Id == u1.Id && u6.Id != 0) {
t.Errorf("Find first record with string primary key")
} }
var us1, us2, us3, us4 []User var us1, us2, us3, us4 []User
db.Find(&us1, "age = 22").Find(&us2, "name = ?", "3").Find(&us3, "age > ?", 20) db.Find(&us1, "age = 22").Find(&us2, "name = ?", "3").Find(&us3, "age > ?", 20)
if !(len(us1) == 1 && len(us2) == 2 && len(us3) == 3) { if !(len(us1) == 1 && len(us2) == 2 && len(us3) == 3) {
t.Errorf("Inline where condition for find when search") t.Errorf("Find all records with inline condition and where")
} }
db.Find(&us4, "name = ? and age > ?", "3", "22") db.Find(&us4, "name = ? and age > ?", "3", "22")
if len(us4) != 1 { if len(us4) != 1 {
t.Errorf("More complex inline where condition for find, %v", us4) t.Errorf("Find all records with complex inline condition")
}
db.First(&u6, u1.Id)
if !(u6.Id == u1.Id && u6.Id != 0) {
t.Errorf("Should find out user with int id")
}
db.First(&u7, strconv.Itoa(int(u1.Id)))
if !(u6.Id == u1.Id && u6.Id != 0) {
t.Errorf("Should find out user with string id")
} }
} }
@ -466,10 +428,11 @@ func TestSelect(t *testing.T) {
var user User var user User
db.Where("name = ?", "3").Select("name").Find(&user) db.Where("name = ?", "3").Select("name").Find(&user)
if user.Id != 0 { if user.Id != 0 {
t.Errorf("Should not got ID because I am only looking for age, %+v", user.Id) t.Errorf("Should not have ID because only searching age, %+v", user.Id)
} }
if user.Name != "3" { if user.Name != "3" {
t.Errorf("Should got Name = 3 when searching it, %+v", user.Id) t.Errorf("Should got Name = 3 when searching with it, %+v", user.Id)
} }
query := db.Where("name = ?", "3").Select("nam;e") query := db.Where("name = ?", "3").Select("nam;e")
@ -479,29 +442,26 @@ func TestSelect(t *testing.T) {
} }
func TestOrderAndPluck(t *testing.T) { func TestOrderAndPluck(t *testing.T) {
var ages []int64 var ages, ages1, ages2, ages3, ages4, ages5 []int64
db.Model(&[]User{}).Order("age desc").Pluck("age", &ages) db.Model(&[]User{}).Order("age desc").Pluck("age", &ages)
if ages[0] != 26 { if ages[0] != 26 {
t.Errorf("The first age should be 26 because of ordered by") t.Errorf("The first age should be 26 when order with age desc")
} }
var ages1, ages2 []int64
db.Model([]User{}).Order("age desc").Pluck("age", &ages1).Order("age").Pluck("age", &ages2) db.Model([]User{}).Order("age desc").Pluck("age", &ages1).Order("age").Pluck("age", &ages2)
if !reflect.DeepEqual(ages1, ages2) { if !reflect.DeepEqual(ages1, ages2) {
t.Errorf("The first order is the primary order") t.Errorf("The first order is the primary order")
} }
var ages3, ages4 []int64
db.Model(&User{}).Order("age desc").Pluck("age", &ages3).Order("age", true).Pluck("age", &ages4) db.Model(&User{}).Order("age desc").Pluck("age", &ages3).Order("age", true).Pluck("age", &ages4)
if reflect.DeepEqual(ages3, ages4) { if reflect.DeepEqual(ages3, ages4) {
t.Errorf("Reorder should works well") t.Errorf("Reorder should works")
} }
ages = []int64{}
var names []string var names []string
db.Model(User{}).Order("name").Order("age desc").Pluck("age", &ages).Pluck("name", &names) db.Model(User{}).Order("name").Order("age desc").Pluck("age", &ages5).Pluck("name", &names)
if !(names[0] == "1" && names[2] == "3" && names[3] == "3" && ages[2] == 24 && ages[3] == 22) { if !(names[0] == "1" && names[2] == "3" && names[3] == "3" && ages5[2] == 24 && ages5[3] == 22) {
t.Errorf("Should be ordered correctly with multiple orders") t.Errorf("Order with multiple orders")
} }
db.Model(User{}).Select("name, age").Find(&[]User{}) db.Model(User{}).Select("name, age").Find(&[]User{})
@ -511,16 +471,17 @@ func TestLimit(t *testing.T) {
var users1, users2, users3 []User var users1, users2, users3 []User
db.Order("age desc").Limit(3).Find(&users1).Limit(5).Find(&users2).Limit(-1).Find(&users3) db.Order("age desc").Limit(3).Find(&users1).Limit(5).Find(&users2).Limit(-1).Find(&users3)
if !(len(users1) == 3 && len(users2) == 5 && len(users3) > 5) { if len(users1) != 3 || len(users2) != 5 || len(users3) <= 5 {
t.Errorf("Limit should works perfectly") t.Errorf("Limit should works")
} }
} }
func TestOffset(t *testing.T) { func TestOffset(t *testing.T) {
var users1, users2, users3, users4 []User var users1, users2, users3, users4 []User
db.Limit(100).Order("age desc").Find(&users1).Offset(3).Find(&users2).Offset(5).Find(&users3).Offset(-1).Find(&users4) db.Limit(100).Order("age desc").Find(&users1).Offset(3).Find(&users2).Offset(5).Find(&users3).Offset(-1).Find(&users4)
if !((len(users1) == len(users4)) && (len(users1)-len(users2) == 3) && (len(users1)-len(users3) == 5)) {
t.Errorf("Offset should works perfectly") if (len(users1) != len(users4)) || (len(users1)-len(users2) != 3) || (len(users1)-len(users3) != 5) {
t.Errorf("Offset should works")
} }
} }
@ -528,7 +489,7 @@ func TestOr(t *testing.T) {
var users []User var users []User
db.Where("name = ?", "1").Or("name = ?", "3").Find(&users) db.Where("name = ?", "1").Or("name = ?", "3").Find(&users)
if len(users) != 3 { if len(users) != 3 {
t.Errorf("Should find three users with name 1 and 3") t.Errorf("Find users with or")
} }
} }
@ -537,16 +498,16 @@ func TestCount(t *testing.T) {
var users []User var users []User
if err := db.Where("name = ?", "1").Or("name = ?", "3").Find(&users).Count(&count).Error; err != nil { if err := db.Where("name = ?", "1").Or("name = ?", "3").Find(&users).Count(&count).Error; err != nil {
t.Errorf("Count should have no error", err) t.Errorf("Count should works", err)
} }
if count != int64(len(users)) { if count != int64(len(users)) {
t.Errorf("Count() method should get same value of users count") t.Errorf("Count() method should get correct value")
} }
db.Model(&User{}).Where("name = ?", "1").Count(&count1).Or("name = ?", "3").Count(&count2) db.Model(&User{}).Where("name = ?", "1").Count(&count1).Or("name = ?", "3").Count(&count2)
if !(count1 == 1 && count2 == 3) { if count1 != 1 || count2 != 3 {
t.Errorf("Multiple count should works well also") t.Errorf("Multiple count should works")
} }
} }
@ -633,6 +594,7 @@ func (s *Product) AfterDelete() (err error) {
s.AfterDeleteCallTimes = s.AfterDeleteCallTimes + 1 s.AfterDeleteCallTimes = s.AfterDeleteCallTimes + 1
return return
} }
func (p *Product) GetCallTimes() []int64 { func (p *Product) GetCallTimes() []int64 {
return []int64{p.BeforeCreateCallTimes, p.BeforeSaveCallTimes, p.BeforeUpdateCallTimes, p.AfterCreateCallTimes, p.AfterSaveCallTimes, p.AfterUpdateCallTimes, p.BeforeDeleteCallTimes, p.AfterDeleteCallTimes} return []int64{p.BeforeCreateCallTimes, p.BeforeSaveCallTimes, p.BeforeUpdateCallTimes, p.AfterCreateCallTimes, p.AfterSaveCallTimes, p.AfterUpdateCallTimes, p.BeforeDeleteCallTimes, p.AfterDeleteCallTimes}
} }
@ -640,82 +602,86 @@ func (p *Product) GetCallTimes() []int64 {
func TestRunCallbacks(t *testing.T) { func TestRunCallbacks(t *testing.T) {
p := Product{Code: "unique_code", Price: 100} p := Product{Code: "unique_code", Price: 100}
db.Save(&p) db.Save(&p)
if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 1, 0, 1, 1, 0, 0, 0}) { if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 1, 0, 1, 1, 0, 0, 0}) {
t.Errorf("Some errors happened when run create callbacks, %v", p.GetCallTimes()) t.Errorf("Callbacks should be invoked successfully, %v", p.GetCallTimes())
} }
db.Where("Code = ?", "unique_code").First(&p) db.Where("Code = ?", "unique_code").First(&p)
if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 1, 0, 0, 0, 0, 0, 0}) { if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 1, 0, 0, 0, 0, 0, 0}) {
t.Errorf("Should be able to query about saved values in before filters, %v", p.GetCallTimes()) t.Errorf("After callbacks values are not saved, %v", p.GetCallTimes())
} }
p.Price = 200 p.Price = 200
db.Save(&p) db.Save(&p)
if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 2, 1, 0, 1, 1, 0, 0}) { if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 2, 1, 0, 1, 1, 0, 0}) {
t.Errorf("Some errors happened when run update callbacks, %v", p.GetCallTimes()) t.Errorf("After update callbacks should be invoked successfully, %v", p.GetCallTimes())
} }
db.Where("Code = ?", "unique_code").First(&p) db.Where("Code = ?", "unique_code").First(&p)
if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 2, 1, 0, 0, 0, 0, 0}) { if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 2, 1, 0, 0, 0, 0, 0}) {
t.Errorf("Some errors happened when run update callbacks, %v", p.GetCallTimes()) t.Errorf("After update callbacks values are not saved, %v", p.GetCallTimes())
} }
db.Delete(&p) db.Delete(&p)
if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 2, 1, 0, 0, 0, 1, 1}) { if !reflect.DeepEqual(p.GetCallTimes(), []int64{1, 2, 1, 0, 0, 0, 1, 1}) {
t.Errorf("Some errors happened when run update callbacks, %v", p.GetCallTimes()) t.Errorf("After delete callbacks should be invoked successfully, %v", p.GetCallTimes())
} }
if db.Where("Code = ?", "unique_code").First(&p).Error == nil { if db.Where("Code = ?", "unique_code").First(&p).Error == nil {
t.Errorf("Should get error when find an deleted record") t.Errorf("Can't find a deleted record")
} }
} }
func TestRunCallbacksAndGetErrors(t *testing.T) { func TestCallbacksWithErrors(t *testing.T) {
p := Product{Code: "Invalid", Price: 100} p := Product{Code: "Invalid", Price: 100}
if db.Save(&p).Error == nil { if db.Save(&p).Error == nil {
t.Errorf("An error from create callbacks expected when create") t.Errorf("An error from before create callbacks happened when create with invalid value")
} }
if db.Where("code = ?", "Invalid").First(&Product{}).Error == nil { if db.Where("code = ?", "Invalid").First(&Product{}).Error == nil {
t.Errorf("Should not save records that have errors") t.Errorf("Should not save record that have errors")
} }
if db.Save(&Product{Code: "dont_save", Price: 100}).Error == nil { if db.Save(&Product{Code: "dont_save", Price: 100}).Error == nil {
t.Errorf("An error from create callbacks expected when create") t.Errorf("An error from after create callbacks happened when create with invalid value")
} }
p2 := Product{Code: "update_callback", Price: 100} p2 := Product{Code: "update_callback", Price: 100}
db.Save(&p2) db.Save(&p2)
p2.Code = "dont_update" p2.Code = "dont_update"
if db.Save(&p2).Error == nil { if db.Save(&p2).Error == nil {
t.Errorf("An error from callbacks expected when update") t.Errorf("An error from before update callbacks happened when update with invalid value")
} }
if db.Where("code = ?", "update_callback").First(&Product{}).Error != nil { if db.Where("code = ?", "update_callback").First(&Product{}).Error != nil {
t.Errorf("Record Should not be updated due to errors happened in callback") t.Errorf("Record Should not be updated due to errors happened in before update callback")
} }
if db.Where("code = ?", "dont_update").First(&Product{}).Error == nil { if db.Where("code = ?", "dont_update").First(&Product{}).Error == nil {
t.Errorf("Record Should not be updated due to errors happened in callback") t.Errorf("Record Should not be updated due to errors happened in before update callback")
} }
p2.Code = "dont_save" p2.Code = "dont_save"
if db.Save(&p2).Error == nil { if db.Save(&p2).Error == nil {
t.Errorf("An error from before save callbacks expected when update") t.Errorf("An error from before save callbacks happened when update with invalid value")
} }
p3 := Product{Code: "dont_delete", Price: 100} p3 := Product{Code: "dont_delete", Price: 100}
db.Save(&p3) db.Save(&p3)
if db.Delete(&p3).Error == nil { if db.Delete(&p3).Error == nil {
t.Errorf("An error from before delete callbacks expected when delete") t.Errorf("An error from before delete callbacks happened when delete")
} }
if db.Where("Code = ?", "dont_delete").First(&p3).Error != nil { if db.Where("Code = ?", "dont_delete").First(&p3).Error != nil {
t.Errorf("Should not delete record due to errors happened in callback") t.Errorf("An error from before delete callbacks happened")
} }
p4 := Product{Code: "after_save_error", Price: 100} p4 := Product{Code: "after_save_error", Price: 100}
db.Save(&p4) db.Save(&p4)
if err := db.First(&Product{}, "code = ?", "after_save_error").Error; err == nil { if err := db.First(&Product{}, "code = ?", "after_save_error").Error; err == nil {
t.Errorf("Record should be reverted if get an error after save", err) t.Errorf("Record should be reverted if get an error in after save callback", err)
} }
p5 := Product{Code: "after_delete_error", Price: 100} p5 := Product{Code: "after_delete_error", Price: 100}
@ -726,7 +692,7 @@ func TestRunCallbacksAndGetErrors(t *testing.T) {
db.Delete(&p5) db.Delete(&p5)
if err := db.First(&Product{}, "code = ?", "after_delete_error").Error; err != nil { if err := db.First(&Product{}, "code = ?", "after_delete_error").Error; err != nil {
t.Errorf("Record should be found because failed to delete", err) t.Errorf("Record shouldn't be deleted because of an error happened in after delete callback", err)
} }
} }
@ -746,11 +712,12 @@ func TestFillSmallerStructCorrectly(t *testing.T) {
} }
} }
func TestNoUnExpectedHappenWithInvalidSql(t *testing.T) { func TestExceptionsWithInvalidSql(t *testing.T) {
var columns []string var columns []string
if db.Where("sdsd.zaaa = ?", "sd;;;aa").Pluck("aaa", &columns).Error == nil { if db.Where("sdsd.zaaa = ?", "sd;;;aa").Pluck("aaa", &columns).Error == nil {
t.Errorf("Should got error with invalid SQL") t.Errorf("Should got error with invalid SQL")
} }
if db.Model(&User{}).Where("sdsd.zaaa = ?", "sd;;;aa").Pluck("aaa", &columns).Error == nil { if db.Model(&User{}).Where("sdsd.zaaa = ?", "sd;;;aa").Pluck("aaa", &columns).Error == nil {
t.Errorf("Should got error with invalid SQL") t.Errorf("Should got error with invalid SQL")
} }
@ -761,6 +728,7 @@ func TestNoUnExpectedHappenWithInvalidSql(t *testing.T) {
db.Where("sdsd.zaaa = ?", "sd;;;aa").Find(&Article{}) db.Where("sdsd.zaaa = ?", "sd;;;aa").Find(&Article{})
db.Where("name = ?", "3").Find(&[]User{}) db.Where("name = ?", "3").Find(&[]User{})
var count1, count2 int64 var count1, count2 int64
db.Model(&User{}).Count(&count1) db.Model(&User{}).Count(&count1)
if count1 <= 0 { if count1 <= 0 {
@ -777,31 +745,31 @@ func TestNoUnExpectedHappenWithInvalidSql(t *testing.T) {
t.Errorf("Users should not be deleted by invalid SQL") t.Errorf("Users should not be deleted by invalid SQL")
} }
db.Where("unexisting = ?", "3").Find(&[]User{}) db.Where("non-existing = ?", "3").Find(&[]User{})
} }
func TestSetTableDirectly(t *testing.T) { func TestSetTableDirectly(t *testing.T) {
var ages []int64 var ages []int64
if db.Table("users").Pluck("age", &ages).Error != nil { if db.Table("users").Pluck("age", &ages).Error != nil {
t.Errorf("No errors should happen if only set table") t.Errorf("No errors should happen if set table for pluck")
} }
if len(ages) == 0 { if len(ages) == 0 {
t.Errorf("Should find out some records") t.Errorf("Should get some records")
} }
var users []User var users []User
if db.Table("users").Find(&users).Error != nil { if db.Table("users").Find(&users).Error != nil {
t.Errorf("No errors should happen if set table to an existing table") t.Errorf("No errors should happen if set table for find")
} }
if db.Table("unexisting_users_table").Find(&users).Error == nil { if db.Table("unexisting_users_table").Find(&users).Error == nil {
t.Errorf("Should got some errors if set table to an unexisting table") t.Errorf("Should got error if set table to an invalid table")
} }
db.Exec("drop table deleted_users;") db.Exec("drop table deleted_users;")
if db.Table("deleted_users").CreateTable(&User{}).Error != nil { if db.Table("deleted_users").CreateTable(&User{}).Error != nil {
t.Errorf("Should create deleted_users table") t.Errorf("Should create table with specified table")
} }
db.Table("deleted_users").Save(&User{Name: "DeletedUser"}) db.Table("deleted_users").Save(&User{Name: "DeletedUser"})
@ -809,19 +777,19 @@ func TestSetTableDirectly(t *testing.T) {
var deleted_users []User var deleted_users []User
db.Table("deleted_users").Find(&deleted_users) db.Table("deleted_users").Find(&deleted_users)
if len(deleted_users) != 1 { if len(deleted_users) != 1 {
t.Errorf("Should query from deleted_users table") t.Errorf("Query from specified table")
} }
var deleted_user User var deleted_user User
db.Table("deleted_users").Find(&deleted_user) db.Table("deleted_users").First(&deleted_user)
if deleted_user.Name != "DeletedUser" { if deleted_user.Name != "DeletedUser" {
t.Errorf("Should query from deleted_users table") t.Errorf("Query from specified table")
} }
var user1, user2, user3 User var user1, user2, user3 User
db.First(&user1).Table("deleted_users").First(&user2).Table("").First(&user3) db.First(&user1).Table("deleted_users").First(&user2).Table("").First(&user3)
if !((user1.Name != user2.Name) && (user1.Name == user3.Name)) { if (user1.Name == user2.Name) || (user1.Name != user3.Name) {
t.Errorf("Set Table Chain Should works well") t.Errorf("unset specified table with blank string")
} }
} }
@ -830,7 +798,7 @@ func TestUpdate(t *testing.T) {
product2 := Product{Code: "234"} product2 := Product{Code: "234"}
db.Save(&product1).Save(&product2).Update("code", "456") db.Save(&product1).Save(&product2).Update("code", "456")
if product2.Code != "456" { if product2.Code != "456" {
t.Errorf("Object should be updated also with update attributes") t.Errorf("Record should be updated with update attributes")
} }
db.First(&product1, product1.Id) db.First(&product1, product1.Id)
@ -841,19 +809,19 @@ func TestUpdate(t *testing.T) {
var product3 Product var product3 Product
db.First(&product3, product2.Id).Update("code", "456") db.First(&product3, product2.Id).Update("code", "456")
if updated_at2.Format(time.RFC3339Nano) != product3.UpdatedAt.Format(time.RFC3339Nano) { if updated_at2.Format(time.RFC3339Nano) != product3.UpdatedAt.Format(time.RFC3339Nano) {
t.Errorf("updated_at should not be updated if nothing really new") t.Errorf("updated_at should not be updated if nothing changed")
} }
if db.First(&Product{}, "code = '123'").Error != nil { if db.First(&Product{}, "code = '123'").Error != nil {
t.Errorf("Product 123's code should not be changed!") t.Errorf("Product 123 should not be updated")
} }
if db.First(&Product{}, "code = '234'").Error == nil { if db.First(&Product{}, "code = '234'").Error == nil {
t.Errorf("Product 234's code should be changed to 456!") t.Errorf("Product 234 should be changed to 456")
} }
if db.First(&Product{}, "code = '456'").Error != nil { if db.First(&Product{}, "code = '456'").Error != nil {
t.Errorf("Product 234's code should be 456 now!") t.Errorf("Product 234 should be changed to 456")
} }
db.Table("products").Where("code in (?)", []string{"123"}).Update("code", "789") db.Table("products").Where("code in (?)", []string{"123"}).Update("code", "789")
@ -861,19 +829,19 @@ func TestUpdate(t *testing.T) {
var product4 Product var product4 Product
db.First(&product4, product1.Id) db.First(&product4, product1.Id)
if updated_at1.Format(time.RFC3339Nano) != product4.UpdatedAt.Format(time.RFC3339Nano) { if updated_at1.Format(time.RFC3339Nano) != product4.UpdatedAt.Format(time.RFC3339Nano) {
t.Errorf("updated_at should be updated if something updated") t.Errorf("updated_at should be updated if something changed")
} }
if db.First(&Product{}, "code = '123'").Error == nil { if db.First(&Product{}, "code = '123'").Error == nil {
t.Errorf("Product 123's code should be changed to 789") t.Errorf("Product 123 should be changed to 789")
} }
if db.First(&Product{}, "code = '456'").Error != nil { if db.First(&Product{}, "code = '456'").Error != nil {
t.Errorf("Product 456's code should not be changed to 789") t.Errorf("Product 456 should not be changed to 789")
} }
if db.First(&Product{}, "code = '789'").Error != nil { if db.First(&Product{}, "code = '789'").Error != nil {
t.Errorf("We should have Product 789") t.Errorf("Product 456 should be changed to 789")
} }
} }
@ -882,8 +850,9 @@ func TestUpdates(t *testing.T) {
product2 := Product{Code: "cde", Price: 20} product2 := Product{Code: "cde", Price: 20}
db.Save(&product1).Save(&product2).Updates(map[string]interface{}{"code": "edf", "price": 100}) db.Save(&product1).Save(&product2).Updates(map[string]interface{}{"code": "edf", "price": 100})
if product2.Code != "edf" || product2.Price != 100 { if product2.Code != "edf" || product2.Price != 100 {
t.Errorf("Object should be updated also with update attributes") t.Errorf("Record should be updated also with update attributes")
} }
db.First(&product1, product1.Id) db.First(&product1, product1.Id)
db.First(&product2, product2.Id) db.First(&product2, product2.Id)
updated_at1 := product1.UpdatedAt updated_at1 := product1.UpdatedAt
@ -892,33 +861,34 @@ func TestUpdates(t *testing.T) {
var product3 Product var product3 Product
db.First(&product3, product2.Id).Updates(Product{Code: "edf", Price: 100}) db.First(&product3, product2.Id).Updates(Product{Code: "edf", Price: 100})
if product3.Code != "edf" || product3.Price != 100 { if product3.Code != "edf" || product3.Price != 100 {
t.Errorf("Object should be updated also with update attributes") t.Errorf("Record should be updated with update attributes")
} }
if updated_at2.Format(time.RFC3339Nano) != product3.UpdatedAt.Format(time.RFC3339Nano) { if updated_at2.Format(time.RFC3339Nano) != product3.UpdatedAt.Format(time.RFC3339Nano) {
t.Errorf("updated_at should not be updated if nothing really new") t.Errorf("updated_at should not be updated if nothing changed")
} }
if db.First(&Product{}, "code = 'abc' and price = 10").Error != nil { if db.First(&Product{}, "code = 'abc' and price = 10").Error != nil {
t.Errorf("Product abc should not be updated!") t.Errorf("Product abc should not be updated")
} }
if db.First(&Product{}, "code = 'cde'").Error == nil { if db.First(&Product{}, "code = 'cde'").Error == nil {
t.Errorf("Product cde should be renamed to edf!") t.Errorf("Product cde should be renamed to edf")
} }
if db.First(&Product{}, "code = 'edf' and price = 100").Error != nil { if db.First(&Product{}, "code = 'edf' and price = 100").Error != nil {
t.Errorf("We should have product edf!") t.Errorf("Product cde should be renamed to edf")
} }
db.Table("products").Where("code in (?)", []string{"abc"}).Updates(map[string]interface{}{"code": "fgh", "price": 200}) db.Table("products").Where("code in (?)", []string{"abc"}).Updates(map[string]interface{}{"code": "fgh", "price": 200})
if db.First(&Product{}, "code = 'abc'").Error == nil { if db.First(&Product{}, "code = 'abc'").Error == nil {
t.Errorf("Product abc's code should be changed to fgh") t.Errorf("Product abc's code should be changed to fgh")
} }
var product4 Product var product4 Product
db.First(&product4, product1.Id) db.First(&product4, product1.Id)
if updated_at1.Format(time.RFC3339Nano) != product4.UpdatedAt.Format(time.RFC3339Nano) { if updated_at1.Format(time.RFC3339Nano) != product4.UpdatedAt.Format(time.RFC3339Nano) {
t.Errorf("updated_at should be updated if something updated") t.Errorf("updated_at should be updated if something changed")
} }
if db.First(&Product{}, "code = 'edf' and price = ?", 100).Error != nil { if db.First(&Product{}, "code = 'edf' and price = ?", 100).Error != nil {
@ -947,16 +917,16 @@ func TestSoftDelete(t *testing.T) {
db.Delete(&order) db.Delete(&order)
if db.First(&Order{}, "amount = 1234").Error == nil { if db.First(&Order{}, "amount = 1234").Error == nil {
t.Errorf("Can't find the user because it is soft deleted") t.Errorf("Can't find order because it is soft deleted")
} }
if db.Unscoped().First(&Order{}, "amount = 1234").Error != nil { if db.Unscoped().First(&Order{}, "amount = 1234").Error != nil {
t.Errorf("Should be able to find out the soft deleted user with unscoped") t.Errorf("Should be able to find out the soft deleted order with unscoped")
} }
db.Unscoped().Delete(&order) db.Unscoped().Delete(&order)
if db.Unscoped().First(&Order{}, "amount = 1234").Error == nil { if db.Unscoped().First(&Order{}, "amount = 1234").Error == nil {
t.Errorf("Can't find out permanently deleted order") t.Errorf("Can't find permanently deleted order")
} }
} }
@ -984,11 +954,10 @@ func TestFindOrInitialize(t *testing.T) {
db.Where(&User{Name: "find or init"}).Assign("age", 44).FirstOrInit(&user4) db.Where(&User{Name: "find or init"}).Assign("age", 44).FirstOrInit(&user4)
if user4.Name != "find or init" || user4.Id != 0 || user4.Age != 44 { if user4.Name != "find or init" || user4.Id != 0 || user4.Age != 44 {
t.Errorf("user should be initialized with search value and assigned attrs") t.Errorf("user should be initialized with search value and assign attrs")
} }
db.Save(&User{Name: "find or init", Age: 33}) db.Save(&User{Name: "find or init", Age: 33})
db.Where(&User{Name: "find or init"}).Attrs("age", 44).FirstOrInit(&user5) db.Where(&User{Name: "find or init"}).Attrs("age", 44).FirstOrInit(&user5)
if user5.Name != "find or init" || user5.Id == 0 || user5.Age != 33 { if user5.Name != "find or init" || user5.Id == 0 || user5.Age != 33 {
t.Errorf("user should be found and not initialized by Attrs") t.Errorf("user should be found and not initialized by Attrs")
@ -1013,7 +982,7 @@ func TestFindOrCreate(t *testing.T) {
} }
db.Where(&User{Name: "find or create", Age: 33}).FirstOrCreate(&user2) db.Where(&User{Name: "find or create", Age: 33}).FirstOrCreate(&user2)
if user2.Name != "find or create" || user2.Id == 0 || user2.Age != 33 { if user1.Id != user2.Id || user2.Name != "find or create" || user2.Id == 0 || user2.Age != 33 {
t.Errorf("user should be created with search value") t.Errorf("user should be created with search value")
} }