mirror of
https://github.com/Luzifer/mondash.git
synced 2024-12-24 13:01:19 +00:00
122 lines
3.4 KiB
Go
122 lines
3.4 KiB
Go
|
package rds
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"io/ioutil"
|
||
|
"net/url"
|
||
|
"regexp"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/aws/aws-sdk-go/aws"
|
||
|
"github.com/aws/aws-sdk-go/aws/request"
|
||
|
"github.com/aws/aws-sdk-go/awstesting"
|
||
|
"github.com/aws/aws-sdk-go/awstesting/unit"
|
||
|
)
|
||
|
|
||
|
func TestPresignWithPresignNotSet(t *testing.T) {
|
||
|
reqs := map[string]*request.Request{}
|
||
|
svc := New(unit.Session, &aws.Config{Region: aws.String("us-west-2")})
|
||
|
|
||
|
f := func() {
|
||
|
// Doesn't panic on nil input
|
||
|
req, _ := svc.CopyDBSnapshotRequest(nil)
|
||
|
req.Sign()
|
||
|
}
|
||
|
if paniced, p := awstesting.DidPanic(f); paniced {
|
||
|
t.Errorf("expect no panic, got %v", p)
|
||
|
}
|
||
|
|
||
|
reqs[opCopyDBSnapshot], _ = svc.CopyDBSnapshotRequest(&CopyDBSnapshotInput{
|
||
|
SourceRegion: aws.String("us-west-1"),
|
||
|
SourceDBSnapshotIdentifier: aws.String("foo"),
|
||
|
TargetDBSnapshotIdentifier: aws.String("bar"),
|
||
|
})
|
||
|
|
||
|
reqs[opCreateDBInstanceReadReplica], _ = svc.CreateDBInstanceReadReplicaRequest(&CreateDBInstanceReadReplicaInput{
|
||
|
SourceRegion: aws.String("us-west-1"),
|
||
|
SourceDBInstanceIdentifier: aws.String("foo"),
|
||
|
DBInstanceIdentifier: aws.String("bar"),
|
||
|
})
|
||
|
|
||
|
for op, req := range reqs {
|
||
|
req.Sign()
|
||
|
b, _ := ioutil.ReadAll(req.HTTPRequest.Body)
|
||
|
q, _ := url.ParseQuery(string(b))
|
||
|
|
||
|
u, _ := url.QueryUnescape(q.Get("PreSignedUrl"))
|
||
|
|
||
|
exp := fmt.Sprintf(`^https://rds.us-west-1\.amazonaws\.com/\?Action=%s.+?DestinationRegion=us-west-2.+`, op)
|
||
|
if re, a := regexp.MustCompile(exp), u; !re.MatchString(a) {
|
||
|
t.Errorf("expect %s to match %s", re, a)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestPresignWithPresignSet(t *testing.T) {
|
||
|
reqs := map[string]*request.Request{}
|
||
|
svc := New(unit.Session, &aws.Config{Region: aws.String("us-west-2")})
|
||
|
|
||
|
f := func() {
|
||
|
// Doesn't panic on nil input
|
||
|
req, _ := svc.CopyDBSnapshotRequest(nil)
|
||
|
req.Sign()
|
||
|
}
|
||
|
if paniced, p := awstesting.DidPanic(f); paniced {
|
||
|
t.Errorf("expect no panic, got %v", p)
|
||
|
}
|
||
|
|
||
|
reqs[opCopyDBSnapshot], _ = svc.CopyDBSnapshotRequest(&CopyDBSnapshotInput{
|
||
|
SourceRegion: aws.String("us-west-1"),
|
||
|
SourceDBSnapshotIdentifier: aws.String("foo"),
|
||
|
TargetDBSnapshotIdentifier: aws.String("bar"),
|
||
|
PreSignedUrl: aws.String("presignedURL"),
|
||
|
})
|
||
|
|
||
|
reqs[opCreateDBInstanceReadReplica], _ = svc.CreateDBInstanceReadReplicaRequest(&CreateDBInstanceReadReplicaInput{
|
||
|
SourceRegion: aws.String("us-west-1"),
|
||
|
SourceDBInstanceIdentifier: aws.String("foo"),
|
||
|
DBInstanceIdentifier: aws.String("bar"),
|
||
|
PreSignedUrl: aws.String("presignedURL"),
|
||
|
})
|
||
|
|
||
|
for _, req := range reqs {
|
||
|
req.Sign()
|
||
|
|
||
|
b, _ := ioutil.ReadAll(req.HTTPRequest.Body)
|
||
|
q, _ := url.ParseQuery(string(b))
|
||
|
|
||
|
u, _ := url.QueryUnescape(q.Get("PreSignedUrl"))
|
||
|
if e, a := "presignedURL", u; !strings.Contains(a, e) {
|
||
|
t.Errorf("expect %s to be in %s", e, a)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestPresignWithSourceNotSet(t *testing.T) {
|
||
|
reqs := map[string]*request.Request{}
|
||
|
svc := New(unit.Session, &aws.Config{Region: aws.String("us-west-2")})
|
||
|
|
||
|
f := func() {
|
||
|
// Doesn't panic on nil input
|
||
|
req, _ := svc.CopyDBSnapshotRequest(nil)
|
||
|
req.Sign()
|
||
|
}
|
||
|
if paniced, p := awstesting.DidPanic(f); paniced {
|
||
|
t.Errorf("expect no panic, got %v", p)
|
||
|
}
|
||
|
|
||
|
reqs[opCopyDBSnapshot], _ = svc.CopyDBSnapshotRequest(&CopyDBSnapshotInput{
|
||
|
SourceDBSnapshotIdentifier: aws.String("foo"),
|
||
|
TargetDBSnapshotIdentifier: aws.String("bar"),
|
||
|
})
|
||
|
|
||
|
for _, req := range reqs {
|
||
|
_, err := req.Presign(5 * time.Minute)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
}
|
||
|
}
|