github.com/bshelton229/agent@v3.5.4+incompatible/agent/s3_downloader_test.go (about)

     1  package agent
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  )
     8  
     9  func TestS3DowloaderBucketPath(t *testing.T) {
    10  	t.Parallel()
    11  
    12  	s3Uploader := S3Downloader{Bucket: "s3://my-bucket-name/foo/bar"}
    13  	assert.Equal(t, s3Uploader.BucketPath(), "foo/bar")
    14  
    15  	s3Uploader.Bucket = "s3://starts-with-an-s/and-this-is-its/folder"
    16  	assert.Equal(t, s3Uploader.BucketPath(), "and-this-is-its/folder")
    17  }
    18  
    19  func TestS3DowloaderBucketName(t *testing.T) {
    20  	t.Parallel()
    21  
    22  	s3Uploader := S3Downloader{Bucket: "s3://my-bucket-name/foo/bar"}
    23  	assert.Equal(t, s3Uploader.BucketName(), "my-bucket-name")
    24  
    25  	s3Uploader.Bucket = "s3://starts-with-an-s"
    26  	assert.Equal(t, s3Uploader.BucketName(), "starts-with-an-s")
    27  }
    28  
    29  func TestS3DowloaderBucketFileLocation(t *testing.T) {
    30  	t.Parallel()
    31  
    32  	s3Uploader := S3Downloader{Bucket: "s3://my-bucket-name/s3/folder", Path: "here/please/right/now/"}
    33  	assert.Equal(t, s3Uploader.BucketFileLocation(), "s3/folder/here/please/right/now/")
    34  
    35  	s3Uploader.Path = ""
    36  	assert.Equal(t, s3Uploader.BucketFileLocation(), "s3/folder/")
    37  }