mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-14 03:54:21 +00:00
41e1a7cc7b
* one concurrent task * disable spooler * fix: improve concurrency in spooler * fix: dont block lock * fix: break if lock failed * fix: check if handler is working * fix: worker id * fix: test * fix: use limit for spoolers configured in startup.yaml * fix test * fix: factory * fix(key): only reduce if not expired * fix(searchQueryFactory): check for string-slice in aggregateID * fix(migrations): combine migrations * fix: allow saving multiple objects in one request * fix(eventstore): logging * fix(eventstore): rethink insert i locks table * fix: ignore failed tests for the moment * fix: tuubel * fix: for tests in io * fix: ignore tests for io * fix: rename concurrent tasks to workers * fix: incomment tests and remove some tests * fix: refert changes for io * refactor(eventstore): combine types of sql in one file * refactor(eventstore): logs, TODO's, tests * fix(eventstore): sql package * test(eventstore): add tests for search query factory * chore: logs * fix(spooler): optimize lock query chore(migrations): rename locks.object_type to view_name chore(migrations): refactor migrations * test: incomment tests * fix: rename PrepareSaves to PrepareBulkSave * chore: go dependencies * fix(migrations): add id in events table * refactor(lock): less magic numbers Co-authored-by: Livio Amstutz <livio.a@gmail.com>
114 lines
4.2 KiB
Go
114 lines
4.2 KiB
Go
package eventsourcing
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
mock_cache "github.com/caos/zitadel/internal/cache/mock"
|
|
"github.com/caos/zitadel/internal/eventstore/mock"
|
|
es_models "github.com/caos/zitadel/internal/eventstore/models"
|
|
"github.com/caos/zitadel/internal/id"
|
|
"github.com/caos/zitadel/internal/usergrant/repository/eventsourcing/model"
|
|
"github.com/golang/mock/gomock"
|
|
)
|
|
|
|
func GetMockedEventstore(ctrl *gomock.Controller, mockEs *mock.MockEventstore) *UserGrantEventStore {
|
|
return &UserGrantEventStore{
|
|
Eventstore: mockEs,
|
|
userGrantCache: GetMockCache(ctrl),
|
|
idGenerator: GetSonyFlacke(),
|
|
}
|
|
}
|
|
|
|
func GetMockCache(ctrl *gomock.Controller) *UserGrantCache {
|
|
mockCache := mock_cache.NewMockCache(ctrl)
|
|
mockCache.EXPECT().Get(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()
|
|
mockCache.EXPECT().Set(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()
|
|
return &UserGrantCache{userGrantCache: mockCache}
|
|
}
|
|
|
|
func GetSonyFlacke() id.Generator {
|
|
return id.SonyFlakeGenerator
|
|
}
|
|
|
|
func GetMockUserGrantByIDOK(ctrl *gomock.Controller) *UserGrantEventStore {
|
|
user := model.UserGrant{
|
|
UserID: "UserID",
|
|
ProjectID: "ProjectID",
|
|
RoleKeys: []string{"Key"},
|
|
}
|
|
data, _ := json.Marshal(user)
|
|
events := []*es_models.Event{
|
|
&es_models.Event{AggregateID: "AggregateID", Sequence: 1, Type: model.UserGrantAdded, Data: data},
|
|
}
|
|
mockEs := mock.NewMockEventstore(ctrl)
|
|
mockEs.EXPECT().FilterEvents(gomock.Any(), gomock.Any()).Return(events, nil)
|
|
return GetMockedEventstore(ctrl, mockEs)
|
|
}
|
|
|
|
func GetMockUserGrantByIDRemoved(ctrl *gomock.Controller) *UserGrantEventStore {
|
|
user := model.UserGrant{
|
|
UserID: "UserID",
|
|
ProjectID: "ProjectID",
|
|
RoleKeys: []string{"Key"},
|
|
}
|
|
data, _ := json.Marshal(user)
|
|
events := []*es_models.Event{
|
|
&es_models.Event{AggregateID: "AggregateID", Sequence: 1, Type: model.UserGrantAdded, Data: data},
|
|
&es_models.Event{AggregateID: "AggregateID", Sequence: 1, Type: model.UserGrantRemoved},
|
|
}
|
|
mockEs := mock.NewMockEventstore(ctrl)
|
|
mockEs.EXPECT().FilterEvents(gomock.Any(), gomock.Any()).Return(events, nil)
|
|
return GetMockedEventstore(ctrl, mockEs)
|
|
}
|
|
|
|
func GetMockUserGrantByIDNoEvents(ctrl *gomock.Controller) *UserGrantEventStore {
|
|
events := []*es_models.Event{}
|
|
mockEs := mock.NewMockEventstore(ctrl)
|
|
mockEs.EXPECT().FilterEvents(gomock.Any(), gomock.Any()).Return(events, nil)
|
|
return GetMockedEventstore(ctrl, mockEs)
|
|
}
|
|
|
|
func GetMockManipulateUserGrant(ctrl *gomock.Controller) *UserGrantEventStore {
|
|
user := model.UserGrant{
|
|
UserID: "UserID",
|
|
ProjectID: "ProjectID",
|
|
RoleKeys: []string{"Key"},
|
|
}
|
|
data, _ := json.Marshal(user)
|
|
events := []*es_models.Event{
|
|
&es_models.Event{AggregateID: "AggregateID", Sequence: 1, Type: model.UserGrantAdded, Data: data},
|
|
}
|
|
mockEs := mock.NewMockEventstore(ctrl)
|
|
mockEs.EXPECT().FilterEvents(gomock.Any(), gomock.Any()).Return(events, nil)
|
|
mockEs.EXPECT().AggregateCreator().Return(es_models.NewAggregateCreator("TEST"))
|
|
mockEs.EXPECT().PushAggregates(gomock.Any(), gomock.Any()).Return(nil)
|
|
return GetMockedEventstore(ctrl, mockEs)
|
|
}
|
|
|
|
func GetMockManipulateUserGrantInactive(ctrl *gomock.Controller) *UserGrantEventStore {
|
|
user := model.UserGrant{
|
|
UserID: "UserID",
|
|
ProjectID: "ProjectID",
|
|
RoleKeys: []string{"Key"},
|
|
}
|
|
data, _ := json.Marshal(user)
|
|
events := []*es_models.Event{
|
|
&es_models.Event{AggregateID: "AggregateID", Sequence: 1, Type: model.UserGrantAdded, Data: data},
|
|
&es_models.Event{AggregateID: "AggregateID", Sequence: 1, Type: model.UserGrantDeactivated},
|
|
}
|
|
mockEs := mock.NewMockEventstore(ctrl)
|
|
mockEs.EXPECT().FilterEvents(gomock.Any(), gomock.Any()).Return(events, nil)
|
|
mockEs.EXPECT().AggregateCreator().Return(es_models.NewAggregateCreator("TEST"))
|
|
mockEs.EXPECT().PushAggregates(gomock.Any(), gomock.Any()).Return(nil)
|
|
return GetMockedEventstore(ctrl, mockEs)
|
|
}
|
|
|
|
func GetMockManipulateUserGrantNoEvents(ctrl *gomock.Controller) *UserGrantEventStore {
|
|
events := []*es_models.Event{}
|
|
mockEs := mock.NewMockEventstore(ctrl)
|
|
mockEs.EXPECT().FilterEvents(gomock.Any(), gomock.Any()).Return(events, nil)
|
|
mockEs.EXPECT().AggregateCreator().Return(es_models.NewAggregateCreator("TEST"))
|
|
mockEs.EXPECT().PushAggregates(gomock.Any(), gomock.Any()).Return(nil)
|
|
return GetMockedEventstore(ctrl, mockEs)
|
|
}
|