61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package common
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
)
|
|
|
|
func testAccessConfig() *AccessConfig {
|
|
return &AccessConfig{}
|
|
}
|
|
|
|
func TestAccessConfigPrepare_Region(t *testing.T) {
|
|
c := testAccessConfig()
|
|
c.RawRegion = ""
|
|
if err := c.Prepare(nil); err != nil {
|
|
t.Fatalf("shouldn't have err: %s", err)
|
|
}
|
|
|
|
c.RawRegion = "us-east-12"
|
|
if err := c.Prepare(nil); err == nil {
|
|
t.Fatal("should have error")
|
|
}
|
|
|
|
c.RawRegion = "us-east-1"
|
|
if err := c.Prepare(nil); err != nil {
|
|
t.Fatalf("shouldn't have err: %s", err)
|
|
}
|
|
|
|
c.RawRegion = "custom"
|
|
if err := c.Prepare(nil); err == nil {
|
|
t.Fatalf("should have err")
|
|
}
|
|
|
|
c.RawRegion = "custom"
|
|
c.SkipValidation = true
|
|
if err := c.Prepare(nil); err != nil {
|
|
t.Fatalf("shouldn't have err: %s", err)
|
|
}
|
|
c.SkipValidation = false
|
|
|
|
}
|
|
|
|
func TestAccessConfigPrepare_RegionRestricted(t *testing.T) {
|
|
c := testAccessConfig()
|
|
|
|
// Create a Session with a custom region
|
|
c.session = session.Must(session.NewSession(&aws.Config{
|
|
Region: aws.String("us-gov-west-1"),
|
|
}))
|
|
|
|
if err := c.Prepare(nil); err != nil {
|
|
t.Fatalf("shouldn't have err: %s", err)
|
|
}
|
|
|
|
if !c.IsGovCloud() {
|
|
t.Fatal("We should be in gov region.")
|
|
}
|
|
}
|