forked from mirror/gorm
122 lines
3.9 KiB
Go
122 lines
3.9 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"gorm.io/gorm"
|
|
. "gorm.io/gorm/utils/tests"
|
|
)
|
|
|
|
func TestDelete(t *testing.T) {
|
|
var users = []User{*GetUser("delete", Config{}), *GetUser("delete", Config{}), *GetUser("delete", Config{})}
|
|
|
|
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]} {
|
|
result = User{}
|
|
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)
|
|
}
|
|
}
|
|
|
|
for _, user := range []User{users[0], users[2]} {
|
|
result = User{}
|
|
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(users[0]).Error; err != nil {
|
|
t.Errorf("errors happened when delete: %v", err)
|
|
}
|
|
|
|
if err := DB.Where("id = ?", users[0].ID).First(&result).Error; err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Errorf("should returns record not found error, but got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDeleteWithTable(t *testing.T) {
|
|
type UserWithDelete struct {
|
|
gorm.Model
|
|
Name string
|
|
}
|
|
|
|
DB.Table("deleted_users").Migrator().DropTable(UserWithDelete{})
|
|
DB.Table("deleted_users").AutoMigrate(UserWithDelete{})
|
|
|
|
user := UserWithDelete{Name: "delete1"}
|
|
DB.Table("deleted_users").Create(&user)
|
|
|
|
var result UserWithDelete
|
|
if err := DB.Table("deleted_users").First(&result).Error; err != nil {
|
|
t.Errorf("failed to find deleted user, got error %v", err)
|
|
}
|
|
|
|
AssertEqual(t, result, user)
|
|
|
|
if err := DB.Table("deleted_users").Delete(&result).Error; err != nil {
|
|
t.Errorf("failed to delete user, got error %v", err)
|
|
}
|
|
|
|
var result2 UserWithDelete
|
|
if err := DB.Table("deleted_users").First(&result2, user.ID).Error; !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Errorf("should raise record not found error, but got error %v", err)
|
|
}
|
|
|
|
var result3 UserWithDelete
|
|
if err := DB.Table("deleted_users").Unscoped().First(&result3, user.ID).Error; err != nil {
|
|
t.Fatalf("failed to find record, got error %v", err)
|
|
}
|
|
|
|
if err := DB.Table("deleted_users").Unscoped().Delete(&result).Error; err != nil {
|
|
t.Errorf("failed to delete user with unscoped, got error %v", err)
|
|
}
|
|
|
|
var result4 UserWithDelete
|
|
if err := DB.Table("deleted_users").Unscoped().First(&result4, user.ID).Error; !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Errorf("should raise record not found error, but got error %v", err)
|
|
}
|
|
}
|
|
|
|
func TestInlineCondDelete(t *testing.T) {
|
|
user1 := *GetUser("inline_delete_1", Config{})
|
|
user2 := *GetUser("inline_delete_2", Config{})
|
|
DB.Save(&user1).Save(&user2)
|
|
|
|
if DB.Delete(&User{}, user1.ID).Error != nil {
|
|
t.Errorf("No error should happen when delete a record")
|
|
} else if err := DB.Where("name = ?", user1.Name).First(&User{}).Error; !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Errorf("User can't be found after delete")
|
|
}
|
|
|
|
if err := DB.Delete(&User{}, "name = ?", user2.Name).Error; err != nil {
|
|
t.Errorf("No error should happen when delete a record, err=%s", err)
|
|
} else if err := DB.Where("name = ?", user2.Name).First(&User{}).Error; !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Errorf("User can't be found after delete")
|
|
}
|
|
}
|
|
|
|
func TestBlockGlobalDelete(t *testing.T) {
|
|
if err := DB.Delete(&User{}).Error; err == nil || !errors.Is(err, gorm.ErrMissingWhereClause) {
|
|
t.Errorf("should returns missing WHERE clause while deleting error")
|
|
}
|
|
}
|