mirror of
https://github.com/Luzifer/cloudkeys-go.git
synced 2024-09-20 08:02:57 +00:00
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
|
package ec2_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"net/url"
|
||
|
|
"regexp"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go/aws"
|
||
|
|
"github.com/aws/aws-sdk-go/awstesting/unit"
|
||
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCopySnapshotPresignedURL(t *testing.T) {
|
||
|
|
svc := ec2.New(unit.Session, &aws.Config{Region: aws.String("us-west-2")})
|
||
|
|
|
||
|
|
func() {
|
||
|
|
defer func() {
|
||
|
|
if r := recover(); r != nil {
|
||
|
|
t.Fatalf("expect CopySnapshotRequest with nill")
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
// Doesn't panic on nil input
|
||
|
|
req, _ := svc.CopySnapshotRequest(nil)
|
||
|
|
req.Sign()
|
||
|
|
}()
|
||
|
|
|
||
|
|
req, _ := svc.CopySnapshotRequest(&ec2.CopySnapshotInput{
|
||
|
|
SourceRegion: aws.String("us-west-1"),
|
||
|
|
SourceSnapshotId: aws.String("snap-id"),
|
||
|
|
})
|
||
|
|
req.Sign()
|
||
|
|
|
||
|
|
b, _ := ioutil.ReadAll(req.HTTPRequest.Body)
|
||
|
|
q, _ := url.ParseQuery(string(b))
|
||
|
|
u, _ := url.QueryUnescape(q.Get("PresignedUrl"))
|
||
|
|
if e, a := "us-west-2", q.Get("DestinationRegion"); e != a {
|
||
|
|
t.Errorf("expect %v, got %v", e, a)
|
||
|
|
}
|
||
|
|
if e, a := "us-west-1", q.Get("SourceRegion"); e != a {
|
||
|
|
t.Errorf("expect %v, got %v", e, a)
|
||
|
|
}
|
||
|
|
|
||
|
|
r := regexp.MustCompile(`^https://ec2\.us-west-1\.amazonaws\.com/.+&DestinationRegion=us-west-2`)
|
||
|
|
if !r.MatchString(u) {
|
||
|
|
t.Errorf("expect %v to match, got %v", r.String(), u)
|
||
|
|
}
|
||
|
|
}
|