gorm/tests/delete_test.go

49 lines
1.4 KiB
Go

package tests_test
import (
"errors"
"testing"
"github.com/jinzhu/gorm"
. "github.com/jinzhu/gorm/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]} {
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)
}
}
}