88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package s3manager_test
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/mastertinner/s3manager/internal/app/s3manager"
|
|
minio "github.com/minio/minio-go"
|
|
)
|
|
|
|
// s3Mock is a mocked S3 client.
|
|
type s3Mock struct {
|
|
Buckets []minio.BucketInfo
|
|
Objects []minio.ObjectInfo
|
|
Err error
|
|
}
|
|
|
|
// CopyObject mocks minio.Client.CopyObject.
|
|
func (s *s3Mock) CopyObject(minio.DestinationInfo, minio.SourceInfo) error {
|
|
return s.Err
|
|
}
|
|
|
|
// GetObject mocks minio.Client.GetObject.
|
|
func (s *s3Mock) GetObject(bucketName string, objectName string, opts minio.GetObjectOptions) (*minio.Object, error) {
|
|
if s.Err != nil {
|
|
return nil, s.Err
|
|
}
|
|
|
|
return &minio.Object{}, nil
|
|
}
|
|
|
|
// ListBuckets mocks minio.Client.ListBuckets.
|
|
func (s *s3Mock) ListBuckets() ([]minio.BucketInfo, error) {
|
|
return s.Buckets, s.Err
|
|
}
|
|
|
|
// ListObjectsV2 mocks minio.Client.ListObjectsV2.
|
|
func (s *s3Mock) ListObjectsV2(name string, p string, r bool, d <-chan struct{}) <-chan minio.ObjectInfo {
|
|
// Add error if exists
|
|
if s.Err != nil {
|
|
s.Objects = append(s.Objects, minio.ObjectInfo{
|
|
Err: s.Err,
|
|
})
|
|
}
|
|
|
|
// Check if bucket exists
|
|
found := false
|
|
for _, b := range s.Buckets {
|
|
if b.Name == name {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
s.Objects = append(s.Objects, minio.ObjectInfo{
|
|
Err: s3manager.ErrBucketDoesNotExist,
|
|
})
|
|
|
|
}
|
|
|
|
objCh := make(chan minio.ObjectInfo, len(s.Objects))
|
|
defer close(objCh)
|
|
|
|
for _, obj := range s.Objects {
|
|
objCh <- obj
|
|
}
|
|
|
|
return objCh
|
|
}
|
|
|
|
// MakeBucket mocks minio.Client.MakeBucket.
|
|
func (s *s3Mock) MakeBucket(string, string) error {
|
|
return s.Err
|
|
}
|
|
|
|
// PutObject mocks minio.Client.PutObject.
|
|
func (s *s3Mock) PutObject(string, string, io.Reader, int64, minio.PutObjectOptions) (int64, error) {
|
|
return 0, s.Err
|
|
}
|
|
|
|
// RemoveBucket mocks minio.Client.RemoveBucket.
|
|
func (s *s3Mock) RemoveBucket(string) error {
|
|
return s.Err
|
|
}
|
|
|
|
// RemoveObject mocks minio.Client.RemoveObject.
|
|
func (s *s3Mock) RemoveObject(string, string) error {
|
|
return s.Err
|
|
}
|