61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestExistsMethod(t *testing.T) {
|
|
t.Run("Without repo options", func(t *testing.T) {
|
|
sqldb, db, mock := NewMockDB()
|
|
defer sqldb.Close()
|
|
|
|
repo := RepoBase[MyModel]{}
|
|
repo.Init(db, nil)
|
|
|
|
id := uuid.New()
|
|
filter := MyModelFilter{
|
|
Id: &id,
|
|
}
|
|
|
|
sql := "SELECT id FROM my_models WHERE my_models.id = $1 LIMIT $2"
|
|
mock.ExpectQuery(fmt.Sprintf("^%s$", regexp.QuoteMeta(sql))).WithArgs(id, 1)
|
|
|
|
result, err := repo.Exists(filter)
|
|
assert.False(t, result)
|
|
assert.Nil(t, err)
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
})
|
|
|
|
t.Run("With id field set in repo options", func(t *testing.T) {
|
|
sqldb, db, mock := NewMockDB()
|
|
defer sqldb.Close()
|
|
|
|
repo := RepoBase[MyModel]{}
|
|
repo.Init(db, &RepoOptions{IdField: "some_other_pk"})
|
|
|
|
id := uuid.New()
|
|
filter := MyModelFilter{
|
|
Id: &id,
|
|
}
|
|
|
|
sql := "SELECT some_other_pk FROM my_models WHERE my_models.id = $1 LIMIT $2"
|
|
mock.ExpectQuery(fmt.Sprintf("^%s$", regexp.QuoteMeta(sql))).WithArgs(id, 1)
|
|
|
|
result, err := repo.Exists(filter)
|
|
assert.False(t, result)
|
|
assert.Nil(t, err)
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
})
|
|
}
|