vendor/github.com/jinzhu/gorm/delete_test.go

   1 package gorm_test
   2
   3 import (
   4         "testing"
   5         "time"
   6 )
   7
   8 func TestDelete(t *testing.T) {
   9         user1, user2 := User{Name: "delete1"}, User{Name: "delete2"}
  10         DB.Save(&user1)
  11         DB.Save(&user2)
  12
  13         if err := DB.Delete(&user1).Error; err != nil {
  14                 t.Errorf("No error should happen when delete a record, err=%s", err)
  15         }
  16
  17         if !DB.Where("name = ?", user1.Name).First(&User{}).RecordNotFound() {
  18                 t.Errorf("User can't be found after delete")
  19         }
  20
  21         if DB.Where("name = ?", user2.Name).First(&User{}).RecordNotFound() {
  22                 t.Errorf("Other users that not deleted should be found-able")
  23         }
  24 }
  25
  26 func TestInlineDelete(t *testing.T) {
  27         user1, user2 := User{Name: "inline_delete1"}, User{Name: "inline_delete2"}
  28         DB.Save(&user1)
  29         DB.Save(&user2)
  30
  31         if DB.Delete(&User{}, user1.Id).Error != nil {
  32                 t.Errorf("No error should happen when delete a record")
  33         } else if !DB.Where("name = ?", user1.Name).First(&User{}).RecordNotFound() {
  34                 t.Errorf("User can't be found after delete")
  35         }
  36
  37         if err := DB.Delete(&User{}, "name = ?", user2.Name).Error; err != nil {
  38                 t.Errorf("No error should happen when delete a record, err=%s", err)
  39         } else if !DB.Where("name = ?", user2.Name).First(&User{}).RecordNotFound() {
  40                 t.Errorf("User can't be found after delete")
  41         }
  42 }
  43
  44 func TestSoftDelete(t *testing.T) {
  45         type User struct {
  46                 Id        int64
  47                 Name      string
  48                 DeletedAt *time.Time
  49         }
  50         DB.AutoMigrate(&User{})
  51
  52         user := User{Name: "soft_delete"}
  53         DB.Save(&user)
  54         DB.Delete(&user)
  55
  56         if DB.First(&User{}, "name = ?", user.Name).Error == nil {
  57                 t.Errorf("Can't find a soft deleted record")
  58         }
  59
  60         if err := DB.Unscoped().First(&User{}, "name = ?", user.Name).Error; err != nil {
  61                 t.Errorf("Should be able to find soft deleted record with Unscoped, but err=%s", err)
  62         }
  63
  64         DB.Unscoped().Delete(&user)
  65         if !DB.Unscoped().First(&User{}, "name = ?", user.Name).RecordNotFound() {
  66                 t.Errorf("Can't find permanently deleted record")
  67         }
  68 }
  69
  70 func TestSoftDeleteWithCustomizedDeletedAtColumnName(t *testing.T) {
  71         creditCard := CreditCard{Number: "411111111234567"}
  72         DB.Save(&creditCard)
  73         DB.Delete(&creditCard)
  74
  75         if deletedAtField, ok := DB.NewScope(&CreditCard{}).FieldByName("DeletedAt"); !ok || deletedAtField.DBName != "deleted_time" {
  76                 t.Errorf("CreditCard's DeletedAt's column name should be `deleted_time`")
  77         }
  78
  79         if DB.First(&CreditCard{}, "number = ?", creditCard.Number).Error == nil {
  80                 t.Errorf("Can't find a soft deleted record")
  81         }
  82
  83         if err := DB.Unscoped().First(&CreditCard{}, "number = ?", creditCard.Number).Error; err != nil {
  84                 t.Errorf("Should be able to find soft deleted record with Unscoped, but err=%s", err)
  85         }
  86
  87         DB.Unscoped().Delete(&creditCard)
  88         if !DB.Unscoped().First(&CreditCard{}, "number = ?", creditCard.Number).RecordNotFound() {
  89                 t.Errorf("Can't find permanently deleted record")
  90         }
  91 }