mirror of
https://github.com/hanzoai/migrate.git
synced 2026-09-30 22:55:10 +00:00
Abhinav Nair
baec8ed6b6
As of Go 1.16, the same functionality is now provided by package io or package os, and those implementations should be preferred in new code. So replacing all usage of ioutil pkg with io & os.
128 lines
3.1 KiB
Go
128 lines
3.1 KiB
Go
package awss3
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
|
st "github.com/golang-migrate/migrate/v4/source/testing"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test(t *testing.T) {
|
|
s3Client := fakeS3{
|
|
bucket: "some-bucket",
|
|
objects: map[string]string{
|
|
"staging/migrations/1_foobar.up.sql": "1 up",
|
|
"staging/migrations/1_foobar.down.sql": "1 down",
|
|
"prod/migrations/1_foobar.up.sql": "1 up",
|
|
"prod/migrations/1_foobar.down.sql": "1 down",
|
|
"prod/migrations/3_foobar.up.sql": "3 up",
|
|
"prod/migrations/4_foobar.up.sql": "4 up",
|
|
"prod/migrations/4_foobar.down.sql": "4 down",
|
|
"prod/migrations/5_foobar.down.sql": "5 down",
|
|
"prod/migrations/7_foobar.up.sql": "7 up",
|
|
"prod/migrations/7_foobar.down.sql": "7 down",
|
|
"prod/migrations/not-a-migration.txt": "",
|
|
"prod/migrations/0-random-stuff/whatever.txt": "",
|
|
},
|
|
}
|
|
driver, err := WithInstance(&s3Client, &Config{
|
|
Bucket: "some-bucket",
|
|
Prefix: "prod/migrations/",
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
st.Test(t, driver)
|
|
}
|
|
|
|
func TestParseURI(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
uri string
|
|
config *Config
|
|
}{
|
|
{
|
|
"with prefix, no trailing slash",
|
|
"s3://migration-bucket/production",
|
|
&Config{
|
|
Bucket: "migration-bucket",
|
|
Prefix: "production/",
|
|
},
|
|
},
|
|
{
|
|
"without prefix, no trailing slash",
|
|
"s3://migration-bucket",
|
|
&Config{
|
|
Bucket: "migration-bucket",
|
|
},
|
|
},
|
|
{
|
|
"with prefix, trailing slash",
|
|
"s3://migration-bucket/production/",
|
|
&Config{
|
|
Bucket: "migration-bucket",
|
|
Prefix: "production/",
|
|
},
|
|
},
|
|
{
|
|
"without prefix, trailing slash",
|
|
"s3://migration-bucket/",
|
|
&Config{
|
|
Bucket: "migration-bucket",
|
|
},
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
actual, err := parseURI(test.uri)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, test.config, actual)
|
|
})
|
|
}
|
|
}
|
|
|
|
type fakeS3 struct {
|
|
s3.S3
|
|
bucket string
|
|
objects map[string]string
|
|
}
|
|
|
|
func (s *fakeS3) ListObjects(input *s3.ListObjectsInput) (*s3.ListObjectsOutput, error) {
|
|
bucket := aws.StringValue(input.Bucket)
|
|
if bucket != s.bucket {
|
|
return nil, errors.New("bucket not found")
|
|
}
|
|
prefix := aws.StringValue(input.Prefix)
|
|
delimiter := aws.StringValue(input.Delimiter)
|
|
var output s3.ListObjectsOutput
|
|
for name := range s.objects {
|
|
if strings.HasPrefix(name, prefix) {
|
|
if delimiter == "" || !strings.Contains(strings.Replace(name, prefix, "", 1), delimiter) {
|
|
output.Contents = append(output.Contents, &s3.Object{
|
|
Key: aws.String(name),
|
|
})
|
|
}
|
|
}
|
|
}
|
|
return &output, nil
|
|
}
|
|
|
|
func (s *fakeS3) GetObject(input *s3.GetObjectInput) (*s3.GetObjectOutput, error) {
|
|
bucket := aws.StringValue(input.Bucket)
|
|
if bucket != s.bucket {
|
|
return nil, errors.New("bucket not found")
|
|
}
|
|
if data, ok := s.objects[aws.StringValue(input.Key)]; ok {
|
|
body := io.NopCloser(strings.NewReader(data))
|
|
return &s3.GetObjectOutput{Body: body}, nil
|
|
}
|
|
return nil, errors.New("object not found")
|
|
}
|