mirror of https://github.com/status-im/migrate.git
83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
package awss3
|
|
|
|
import (
|
|
"errors"
|
|
"io/ioutil"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
|
"github.com/golang-migrate/migrate/v3/source"
|
|
st "github.com/golang-migrate/migrate/v3/source/testing"
|
|
)
|
|
|
|
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 := s3Driver{
|
|
bucket: "some-bucket",
|
|
prefix: "prod/migrations/",
|
|
migrations: source.NewMigrations(),
|
|
s3client: &s3Client,
|
|
}
|
|
err := driver.loadMigrations()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
st.Test(t, &driver)
|
|
}
|
|
|
|
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 := ioutil.NopCloser(strings.NewReader(data))
|
|
return &s3.GetObjectOutput{Body: body}, nil
|
|
}
|
|
return nil, errors.New("object not found")
|
|
}
|