mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-14 11:58:02 +00:00
8bfa1a083c
* feat: get views and failed events * feat: get views and failed events * feat: get views and failed events * Update internal/view/repository/sequence.go Co-authored-by: Livio Amstutz <livio.a@gmail.com> * Update internal/view/repository/general_query.go Co-authored-by: Livio Amstutz <livio.a@gmail.com> Co-authored-by: Livio Amstutz <livio.a@gmail.com>
567 lines
12 KiB
Go
567 lines
12 KiB
Go
package repository
|
|
|
|
import (
|
|
caos_errs "github.com/caos/zitadel/internal/errors"
|
|
"github.com/caos/zitadel/internal/model"
|
|
"github.com/jinzhu/gorm"
|
|
"testing"
|
|
)
|
|
|
|
func TestPrepareGetByKey(t *testing.T) {
|
|
type args struct {
|
|
table string
|
|
key ColumnKey
|
|
value string
|
|
}
|
|
type res struct {
|
|
result Test
|
|
wantErr bool
|
|
errFunc func(err error) bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
db *dbMock
|
|
args args
|
|
res res
|
|
}{
|
|
{
|
|
"ok",
|
|
mockDB(t).
|
|
expectGetByID("TESTTABLE", "test", "VALUE"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
key: TestSearchKey_TEST,
|
|
value: "VALUE",
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"not found",
|
|
mockDB(t).
|
|
expectGetByIDErr("TESTTABLE", "test", "VALUE", gorm.ErrRecordNotFound),
|
|
args{
|
|
table: "TESTTABLE",
|
|
key: TestSearchKey_TEST,
|
|
value: "VALUE",
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsNotFound,
|
|
},
|
|
},
|
|
{
|
|
"db err",
|
|
mockDB(t).
|
|
expectGetByIDErr("TESTTABLE", "test", "VALUE", gorm.ErrUnaddressable),
|
|
args{
|
|
table: "TESTTABLE",
|
|
key: TestSearchKey_TEST,
|
|
value: "VALUE",
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsInternal,
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
res := &Test{}
|
|
getByID := PrepareGetByKey(tt.args.table, tt.args.key, tt.args.value)
|
|
err := getByID(tt.db.db, res)
|
|
|
|
if !tt.res.wantErr && err != nil {
|
|
t.Errorf("got wrong err should be nil: %v ", err)
|
|
}
|
|
|
|
if tt.res.wantErr && !tt.res.errFunc(err) {
|
|
t.Errorf("got wrong err: %v ", err)
|
|
}
|
|
if err := tt.db.mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
|
|
tt.db.close()
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPrepareGetByQuery(t *testing.T) {
|
|
type args struct {
|
|
table string
|
|
searchQuery SearchQuery
|
|
}
|
|
type res struct {
|
|
result Test
|
|
wantErr bool
|
|
errFunc func(err error) bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
db *dbMock
|
|
args args
|
|
res res
|
|
}{
|
|
{
|
|
"search with equals case insensitive",
|
|
mockDB(t).
|
|
expectGetByQuery("TESTTABLE", "test", "=", "VALUE"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodEqualsIgnoreCase, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"search with equals case sensitive",
|
|
mockDB(t).
|
|
expectGetByQueryCaseSensitive("TESTTABLE", "test", "=", "VALUE"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodEquals, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"search with startswith, case insensitive",
|
|
mockDB(t).
|
|
expectGetByQuery("TESTTABLE", "test", "LIKE", "VALUE%"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodStartsWithIgnoreCase, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"search with startswith case sensitive",
|
|
mockDB(t).
|
|
expectGetByQueryCaseSensitive("TESTTABLE", "test", "LIKE", "VALUE%"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodStartsWith, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"search with contains case insensitive",
|
|
mockDB(t).
|
|
expectGetByQuery("TESTTABLE", "test", "LIKE", "%VALUE%"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodContainsIgnoreCase, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"search with contains case sensitive",
|
|
mockDB(t).
|
|
expectGetByQueryCaseSensitive("TESTTABLE", "test", "LIKE", "%VALUE%"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodContains, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"search expect not found err",
|
|
mockDB(t).
|
|
expectGetByQueryErr("TESTTABLE", "test", "LIKE", "%VALUE%", gorm.ErrRecordNotFound),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodContainsIgnoreCase, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsNotFound,
|
|
},
|
|
},
|
|
{
|
|
"search expect internal err",
|
|
mockDB(t).
|
|
expectGetByQueryErr("TESTTABLE", "test", "LIKE", "%VALUE%", gorm.ErrUnaddressable),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_TEST, method: model.SearchMethodContainsIgnoreCase, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsInternal,
|
|
},
|
|
},
|
|
{
|
|
"search with invalid column",
|
|
mockDB(t).
|
|
expectGetByQuery("TESTTABLE", "", "=", "VALUE"),
|
|
args{
|
|
table: "TESTTABLE",
|
|
searchQuery: TestSearchQuery{key: TestSearchKey_UNDEFINED, method: model.SearchMethodEqualsIgnoreCase, value: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsErrorInvalidArgument,
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
res := &Test{}
|
|
getByQuery := PrepareGetByQuery(tt.args.table, tt.args.searchQuery)
|
|
err := getByQuery(tt.db.db, res)
|
|
|
|
if !tt.res.wantErr && err != nil {
|
|
t.Errorf("got wrong err should be nil: %v ", err)
|
|
}
|
|
|
|
if tt.res.wantErr && !tt.res.errFunc(err) {
|
|
t.Errorf("got wrong err: %v ", err)
|
|
}
|
|
if err := tt.db.mock.ExpectationsWereMet(); !tt.res.wantErr && err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
|
|
tt.db.close()
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPreparePut(t *testing.T) {
|
|
type args struct {
|
|
table string
|
|
object *Test
|
|
}
|
|
type res struct {
|
|
result Test
|
|
wantErr bool
|
|
errFunc func(err error) bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
db *dbMock
|
|
args args
|
|
res res
|
|
}{
|
|
{
|
|
"ok",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectSave("TESTTABLE", Test{ID: "AggregateID", Test: "VALUE"}).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
object: &Test{ID: "AggregateID", Test: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"db error",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectSaveErr("TESTTABLE", Test{ID: "AggregateID", Test: "VALUE"}, gorm.ErrUnaddressable).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
object: &Test{ID: "AggregateID", Test: "VALUE"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsInternal,
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
getPut := PrepareSave(tt.args.table)
|
|
err := getPut(tt.db.db, tt.args.object)
|
|
|
|
if !tt.res.wantErr && err != nil {
|
|
t.Errorf("got wrong err should be nil: %v ", err)
|
|
}
|
|
|
|
if tt.res.wantErr && !tt.res.errFunc(err) {
|
|
t.Errorf("got wrong err: %v ", err)
|
|
}
|
|
if err := tt.db.mock.ExpectationsWereMet(); !tt.res.wantErr && err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
|
|
tt.db.close()
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPrepareDelete(t *testing.T) {
|
|
type args struct {
|
|
table string
|
|
key ColumnKey
|
|
value string
|
|
}
|
|
type res struct {
|
|
result Test
|
|
wantErr bool
|
|
errFunc func(err error) bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
db *dbMock
|
|
args args
|
|
res res
|
|
}{
|
|
{
|
|
"delete",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemove("TESTTABLE", "id", "VALUE").
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
key: TestSearchKey_ID,
|
|
value: "VALUE",
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"db error",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemoveErr("TESTTABLE", "id", "VALUE", gorm.ErrUnaddressable).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
key: TestSearchKey_ID,
|
|
value: "VALUE",
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsInternal,
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
getDelete := PrepareDeleteByKey(tt.args.table, tt.args.key, tt.args.value)
|
|
err := getDelete(tt.db.db)
|
|
|
|
if !tt.res.wantErr && err != nil {
|
|
t.Errorf("got wrong err should be nil: %v ", err)
|
|
}
|
|
|
|
if tt.res.wantErr && !tt.res.errFunc(err) {
|
|
t.Errorf("got wrong err: %v ", err)
|
|
}
|
|
if err := tt.db.mock.ExpectationsWereMet(); !tt.res.wantErr && err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
|
|
tt.db.close()
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPrepareDeleteByKeys(t *testing.T) {
|
|
type args struct {
|
|
table string
|
|
keys []Key
|
|
}
|
|
type res struct {
|
|
result Test
|
|
wantErr bool
|
|
errFunc func(err error) bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
db *dbMock
|
|
args args
|
|
res res
|
|
}{
|
|
{
|
|
"delete single key",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemoveKeys("TESTTABLE", Key{Key: TestSearchKey_ID, Value: "VALUE"}).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
keys: []Key{
|
|
{Key: TestSearchKey_ID, Value: "VALUE"},
|
|
},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"delete multiple keys",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemoveKeys("TESTTABLE", Key{Key: TestSearchKey_ID, Value: "VALUE"}, Key{Key: TestSearchKey_TEST, Value: "VALUE2"}).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
keys: []Key{
|
|
{Key: TestSearchKey_ID, Value: "VALUE"},
|
|
{Key: TestSearchKey_TEST, Value: "VALUE2"},
|
|
},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"db error",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemoveErr("TESTTABLE", "id", "VALUE", gorm.ErrUnaddressable).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
keys: []Key{
|
|
{Key: TestSearchKey_ID, Value: "VALUE"},
|
|
},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsInternal,
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
getDelete := PrepareDeleteByKeys(tt.args.table, tt.args.keys...)
|
|
err := getDelete(tt.db.db)
|
|
|
|
if !tt.res.wantErr && err != nil {
|
|
t.Errorf("got wrong err should be nil: %v ", err)
|
|
}
|
|
|
|
if tt.res.wantErr && !tt.res.errFunc(err) {
|
|
t.Errorf("got wrong err: %v ", err)
|
|
}
|
|
if err := tt.db.mock.ExpectationsWereMet(); !tt.res.wantErr && err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
|
|
tt.db.close()
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPrepareDeleteByObject(t *testing.T) {
|
|
type args struct {
|
|
table string
|
|
object interface{}
|
|
}
|
|
type res struct {
|
|
result Test
|
|
wantErr bool
|
|
errFunc func(err error) bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
db *dbMock
|
|
args args
|
|
res res
|
|
}{
|
|
{
|
|
"delete",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemoveByObject("TESTTABLE", Test{ID: "VALUE", Test: "TEST"}).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
object: &Test{ID: "VALUE", Test: "TEST"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"delete multiple PK",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemoveByObjectMultiplePKs("TESTTABLE", TestMultiplePK{TestID: "TESTID", HodorID: "HODORID", Test: "TEST"}).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
object: &TestMultiplePK{TestID: "TESTID", HodorID: "HODORID", Test: "TEST"},
|
|
},
|
|
res{
|
|
wantErr: false,
|
|
},
|
|
},
|
|
{
|
|
"db error",
|
|
mockDB(t).
|
|
expectBegin(nil).
|
|
expectRemoveErr("TESTTABLE", "id", "VALUE", gorm.ErrUnaddressable).
|
|
expectCommit(nil),
|
|
args{
|
|
table: "TESTTABLE",
|
|
object: &Test{ID: "VALUE", Test: "TEST"},
|
|
},
|
|
res{
|
|
result: Test{ID: "VALUE"},
|
|
wantErr: true,
|
|
errFunc: caos_errs.IsInternal,
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
getDelete := PrepareDeleteByObject(tt.args.table, tt.args.object)
|
|
err := getDelete(tt.db.db)
|
|
|
|
if !tt.res.wantErr && err != nil {
|
|
t.Errorf("got wrong err should be nil: %v ", err)
|
|
}
|
|
|
|
if tt.res.wantErr && !tt.res.errFunc(err) {
|
|
t.Errorf("got wrong err: %v ", err)
|
|
}
|
|
if err := tt.db.mock.ExpectationsWereMet(); !tt.res.wantErr && err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
|
|
tt.db.close()
|
|
})
|
|
}
|
|
}
|