package s3manager_test import ( "io" . "github.com/mastertinner/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: 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 }