forked from mirror/gorm
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
|
package tests
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDelete(t *testing.T, db *gorm.DB) {
|
||
|
|
db.Migrator().DropTable(&User{})
|
||
|
|
db.AutoMigrate(&User{})
|
||
|
|
|
||
|
|
t.Run("Delete", func(t *testing.T) {
|
||
|
|
var users = []User{{
|
||
|
|
Name: "find",
|
||
|
|
Age: 1,
|
||
|
|
Birthday: Now(),
|
||
|
|
}, {
|
||
|
|
Name: "find",
|
||
|
|
Age: 2,
|
||
|
|
Birthday: Now(),
|
||
|
|
}, {
|
||
|
|
Name: "find",
|
||
|
|
Age: 3,
|
||
|
|
Birthday: Now(),
|
||
|
|
}}
|
||
|
|
|
||
|
|
if err := db.Create(&users).Error; err != nil {
|
||
|
|
t.Errorf("errors happened when create: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, user := range users {
|
||
|
|
if user.ID == 0 {
|
||
|
|
t.Fatalf("user's primary key should has value after create, got : %v", user.ID)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := db.Delete(&users[1]).Error; err != nil {
|
||
|
|
t.Errorf("errors happened when delete: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var result User
|
||
|
|
if err := db.Where("id = ?", users[1].ID).First(&result).Error; err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
|
||
|
|
t.Errorf("should returns record not found error, but got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, user := range []User{users[0], users[2]} {
|
||
|
|
if err := db.Where("id = ?", user.ID).First(&result).Error; err != nil {
|
||
|
|
t.Errorf("no error should returns when query %v, but got %v", user.ID, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := db.Delete(&User{}).Error; err == nil || !errors.Is(err, gorm.ErrMissingWhereClause) {
|
||
|
|
t.Errorf("should returns missing WHERE clause while deleting error")
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, user := range []User{users[0], users[2]} {
|
||
|
|
if err := db.Where("id = ?", user.ID).First(&result).Error; err != nil {
|
||
|
|
t.Errorf("no error should returns when query %v, but got %v", user.ID, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|