mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-14 20:08:02 +00:00
2f91679623
This change adds a core_generate_all make target. It installs the required tools and runs generate on the complete project. `golang/mock` is no longer maintained and a fork is available from the Uber folks. So the latter is used as tool. All the mock files have been regenerated and are part of the PR. The obsolete `tools` directory has been removed, as all the tools are now part of specific make targets. Co-authored-by: Silvan <silvan.reusser@gmail.com>
64 lines
1.8 KiB
Go
64 lines
1.8 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"testing"
|
|
"time"
|
|
|
|
"go.uber.org/mock/gomock"
|
|
|
|
caos_errors "github.com/zitadel/zitadel/internal/errors"
|
|
"github.com/zitadel/zitadel/internal/static"
|
|
)
|
|
|
|
func NewStorage(t *testing.T) *MockStorage {
|
|
return NewMockStorage(gomock.NewController(t))
|
|
}
|
|
|
|
func (m *MockStorage) ExpectPutObject() *MockStorage {
|
|
m.EXPECT().
|
|
PutObject(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
|
|
DoAndReturn(func(ctx context.Context, instanceID, location, resourceOwner, name, contentType string, objectType static.ObjectType, object io.Reader, objectSize int64) (*static.Asset, error) {
|
|
hash, _ := io.ReadAll(object)
|
|
return &static.Asset{
|
|
InstanceID: instanceID,
|
|
Name: name,
|
|
Hash: string(hash),
|
|
Size: objectSize,
|
|
LastModified: time.Now(),
|
|
Location: location,
|
|
ContentType: contentType,
|
|
}, nil
|
|
})
|
|
return m
|
|
}
|
|
|
|
func (m *MockStorage) ExpectPutObjectError() *MockStorage {
|
|
m.EXPECT().
|
|
PutObject(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
|
|
Return(nil, caos_errors.ThrowInternal(nil, "", ""))
|
|
return m
|
|
}
|
|
|
|
func (m *MockStorage) ExpectRemoveObjectNoError() *MockStorage {
|
|
m.EXPECT().
|
|
RemoveObject(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
|
|
Return(nil)
|
|
return m
|
|
}
|
|
|
|
func (m *MockStorage) ExpectRemoveObjectsNoError() *MockStorage {
|
|
m.EXPECT().
|
|
RemoveObjects(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
|
|
Return(nil)
|
|
return m
|
|
}
|
|
|
|
func (m *MockStorage) ExpectRemoveObjectError() *MockStorage {
|
|
m.EXPECT().
|
|
RemoveObject(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).
|
|
Return(caos_errors.ThrowInternal(nil, "", ""))
|
|
return m
|
|
}
|