github.com/monkeswag33/noter-go@v0.0.0-20220505233910-9d72ccb0bdb6/db/delete_test.go (about) 1 package db 2 3 import ( 4 "testing" 5 6 "github.com/stretchr/testify/assert" 7 ) 8 9 func TestDeleteUser(t *testing.T) { 10 database, err := InitTesterDB() 11 assert.NoError(t, err) 12 var clonedUser User = user 13 assert.NoError(t, database.DB.Create(&clonedUser).Error) 14 var users []User 15 assert.NoError(t, database.DB.Find(&users).Error) 16 assert.Len(t, users, 1) 17 assert.NoError(t, database.DeleteUser(clonedUser)) 18 assert.NoError(t, database.DB.Find(&users).Error) 19 assert.Len(t, users, 0) 20 } 21 22 func TestDeleteNote(t *testing.T) { 23 database, err := InitTesterDB() 24 assert.NoError(t, err) 25 var clonedUser User = user 26 var clonedNote Note = note 27 assert.NoError(t, database.DB.Create(&clonedUser).Error) 28 clonedNote.User = clonedUser 29 assert.NoError(t, database.DB.Create(&clonedNote).Error) 30 var notes []Note 31 assert.NoError(t, database.DB.Find(¬es).Error) 32 assert.Len(t, notes, 1) 33 assert.NoError(t, database.DeleteNote(clonedNote)) 34 assert.NoError(t, database.DB.Find(¬es).Error) 35 assert.Len(t, notes, 0) 36 }