mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-16 21:08:00 +00:00
fa9f581d56
* chore: move to new org * logging * fix: org rename caos -> zitadel Co-authored-by: adlerhurst <silvan.reusser@gmail.com>
64 lines
1.8 KiB
Go
64 lines
1.8 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/golang/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
|
|
}
|