github.com/aavshr/aws-sdk-go@v1.41.3/service/s3/s3manager/integ_shared_test.go (about)

     1  //go:build integration
     2  // +build integration
     3  
     4  package s3manager_test
     5  
     6  import (
     7  	"crypto/md5"
     8  	"fmt"
     9  	"io"
    10  	"os"
    11  	"testing"
    12  
    13  	"github.com/aavshr/aws-sdk-go/aws"
    14  	"github.com/aavshr/aws-sdk-go/aws/session"
    15  	"github.com/aavshr/aws-sdk-go/awstesting/integration"
    16  	"github.com/aavshr/aws-sdk-go/awstesting/integration/s3integ"
    17  	"github.com/aavshr/aws-sdk-go/service/s3"
    18  	"github.com/aavshr/aws-sdk-go/service/s3/s3manager"
    19  )
    20  
    21  func init() {
    22  	integSess = integration.SessionWithDefaultRegion("us-west-2")
    23  }
    24  
    25  var integSess *session.Session
    26  var bucketName *string
    27  var svc *s3.S3
    28  
    29  func TestMain(m *testing.M) {
    30  	svc = s3.New(integSess)
    31  	bucketName = aws.String(s3integ.GenerateBucketName())
    32  	if err := s3integ.SetupBucket(svc, *bucketName); err != nil {
    33  		panic(err)
    34  	}
    35  
    36  	var result int
    37  	defer func() {
    38  		if err := s3integ.CleanupBucket(svc, *bucketName); err != nil {
    39  			fmt.Fprintln(os.Stderr, err)
    40  		}
    41  		if r := recover(); r != nil {
    42  			fmt.Fprintln(os.Stderr, "S3 integrationt tests paniced,", r)
    43  			result = 1
    44  		}
    45  		os.Exit(result)
    46  	}()
    47  
    48  	result = m.Run()
    49  }
    50  
    51  type dlwriter struct {
    52  	buf []byte
    53  }
    54  
    55  func newDLWriter(size int) *dlwriter {
    56  	return &dlwriter{buf: make([]byte, size)}
    57  }
    58  
    59  func (d dlwriter) WriteAt(p []byte, pos int64) (n int, err error) {
    60  	if pos > int64(len(d.buf)) {
    61  		return 0, io.EOF
    62  	}
    63  
    64  	written := 0
    65  	for i, b := range p {
    66  		if i >= len(d.buf) {
    67  			break
    68  		}
    69  		d.buf[pos+int64(i)] = b
    70  		written++
    71  	}
    72  	return written, nil
    73  }
    74  
    75  func validate(t *testing.T, key string, md5value string) {
    76  	mgr := s3manager.NewDownloader(integSess)
    77  	params := &s3.GetObjectInput{Bucket: bucketName, Key: &key}
    78  
    79  	w := newDLWriter(1024 * 1024 * 20)
    80  	n, err := mgr.Download(w, params)
    81  	if err != nil {
    82  		t.Fatalf("expect no error, got %v", err)
    83  	}
    84  	if e, a := md5value, fmt.Sprintf("%x", md5.Sum(w.buf[0:n])); e != a {
    85  		t.Errorf("expect %s md5 value, got %s", e, a)
    86  	}
    87  }