testutils/s3managerv2_mock.go

package testutils

import (
	"context"
	"errors"
	"fmt"
	"io"
	"os"
	"strings"

	"github.com/aws/aws-sdk-go-v2/feature/s3/manager"
	"github.com/aws/aws-sdk-go-v2/service/s3"
)

type S3ManagerV2Mock struct {
	S3ManagerMockFileSystem map[string]string
	TestDataLocation        string
}

func (s *S3ManagerV2Mock) Upload(
	ctx context.Context,
	uploadInput *s3.PutObjectInput,
	opts ...func(uploader *manager.Uploader),
) (*manager.UploadOutput, error) {
	if ctx == nil {
		return nil, errors.New("cannot create context from nil parent")
	}
	if uploadInput.Bucket == nil || *uploadInput.Bucket == "" {
		return nil, errors.New("invalid bucket")
	}

	if s.S3ManagerMockFileSystem == nil {
		s.S3ManagerMockFileSystem = make(map[string]string)
	}

	buf := new(strings.Builder)
	_, err := io.Copy(buf, uploadInput.Body)
	if err != nil {
		fmt.Println(err)
	}
	s.S3ManagerMockFileSystem[*uploadInput.Key] = buf.String()

	return &manager.UploadOutput{
		Location: *uploadInput.Key,
	}, nil
}

func (s *S3ManagerV2Mock) Download(
	ctx context.Context,
	w io.WriterAt,
	input *s3.GetObjectInput,
	options ...func(*manager.Downloader),
) (n int64, err error) {
	if ctx == nil {
		return -1, errors.New("cannot create context from nil parent")
	}

	if *input.Key == "valid" {
		res, _ := os.ReadFile(s.TestDataLocation + "/flag-config.yaml")
		_, _ = w.WriteAt(res, 0)
		return 1, nil
	} else if *input.Key == "no-file" {
		return 0, errors.New("no file")
	}

	return 1, nil
}