43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
package mocks
|
|
|
|
import (
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type MockS3Client struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *MockS3Client) PresignedDownloadUrl(bucket, key, filename string) (string, error) {
|
|
args := m.Called(bucket, key, filename)
|
|
return args.String(0), args.Error(1)
|
|
}
|
|
|
|
func (m *MockS3Client) UploadFile(bucketName, pathToFile, fileName, targetFileName string) (*s3.PutObjectOutput, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (m *MockS3Client) DownloadAndUnzipFile(bucketName, targetFolder, inputFileName, targetFileName, unzippedFiledName string) (string, error) {
|
|
|
|
args := m.Called(bucketName, targetFolder, inputFileName, targetFileName, unzippedFiledName)
|
|
return args.String(0), args.Error(1)
|
|
}
|
|
|
|
func (m *MockS3Client) DownloadFile(bucketName, targetFolder, inputFileName, targetFileName string) error {
|
|
|
|
args := m.Called(bucketName, targetFolder, inputFileName, targetFileName)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *MockS3Client) CheckIfPresent(bucketName, fileName string) (bool, error) {
|
|
|
|
args := m.Called(bucketName, fileName)
|
|
return args.Bool(0), args.Error(1)
|
|
}
|
|
|
|
func (m *MockS3Client) PreSignedUploadUrl(bucket, sessionId, extension, contentType string) (string, error) {
|
|
args := m.Called(bucket, sessionId, extension, contentType)
|
|
return args.String(0), args.Error(1)
|
|
}
|