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(&notes).Error)
    32  	assert.Len(t, notes, 1)
    33  	assert.NoError(t, database.DeleteNote(clonedNote))
    34  	assert.NoError(t, database.DB.Find(&notes).Error)
    35  	assert.Len(t, notes, 0)
    36  }