mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-16 12:58:00 +00:00
f51f0ede5c
* fix: add assets to eventstore and event * fix: project member, grant member, app changed tests * fix: asset migrations * feat: add asset tests * feat: add asset tests * Update internal/eventstore/repository/repository.go Co-authored-by: Livio Amstutz <livio.a@gmail.com> * feat: add asset tests Co-authored-by: Livio Amstutz <livio.a@gmail.com>
58 lines
2.0 KiB
Go
58 lines
2.0 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/caos/zitadel/internal/eventstore/repository"
|
|
)
|
|
|
|
func NewRepo(t *testing.T) *MockRepository {
|
|
return NewMockRepository(gomock.NewController(t))
|
|
}
|
|
|
|
func (m *MockRepository) ExpectFilterNoEventsNoError() *MockRepository {
|
|
m.EXPECT().Filter(gomock.Any(), gomock.Any()).Return(nil, nil)
|
|
return m
|
|
}
|
|
|
|
func (m *MockRepository) ExpectFilterEvents(events ...*repository.Event) *MockRepository {
|
|
m.EXPECT().Filter(gomock.Any(), gomock.Any()).Return(events, nil)
|
|
return m
|
|
}
|
|
|
|
func (m *MockRepository) ExpectPush(expectedEvents []*repository.Event, expectedAssets []*repository.Asset, expectedUniqueConstraints ...*repository.UniqueConstraint) *MockRepository {
|
|
m.EXPECT().Push(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).DoAndReturn(
|
|
func(ctx context.Context, events []*repository.Event, assets []*repository.Asset, uniqueConstraints ...*repository.UniqueConstraint) error {
|
|
assert.Equal(m.ctrl.T, expectedEvents, events)
|
|
if expectedUniqueConstraints == nil {
|
|
expectedUniqueConstraints = []*repository.UniqueConstraint{}
|
|
}
|
|
assert.Equal(m.ctrl.T, expectedUniqueConstraints, uniqueConstraints)
|
|
|
|
assert.Equal(m.ctrl.T, expectedAssets, assets)
|
|
return nil
|
|
},
|
|
)
|
|
return m
|
|
}
|
|
|
|
func (m *MockRepository) ExpectPushFailed(err error, expectedEvents []*repository.Event, expectedAssets []*repository.Asset, expectedUniqueConstraints ...*repository.UniqueConstraint) *MockRepository {
|
|
m.EXPECT().Push(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).DoAndReturn(
|
|
func(ctx context.Context, events []*repository.Event, assets []*repository.Asset, uniqueConstraints ...*repository.UniqueConstraint) error {
|
|
assert.Equal(m.ctrl.T, expectedEvents, events)
|
|
if expectedUniqueConstraints == nil {
|
|
expectedUniqueConstraints = []*repository.UniqueConstraint{}
|
|
}
|
|
assert.Equal(m.ctrl.T, expectedUniqueConstraints, uniqueConstraints)
|
|
|
|
assert.Equal(m.ctrl.T, expectedAssets, assets)
|
|
return err
|
|
},
|
|
)
|
|
return m
|
|
}
|