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 }