119 lines
3.3 KiB
Go
119 lines
3.3 KiB
Go
package aws
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
awsCredentials "github.com/aws/aws-sdk-go/aws/credentials"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
)
|
|
|
|
func TestGetSupportedEC2Platforms(t *testing.T) {
|
|
ec2Endpoints := []*awsMockEndpoint{
|
|
&awsMockEndpoint{
|
|
Request: &awsMockRequest{"POST", "/", "Action=DescribeAccountAttributes&" +
|
|
"AttributeName.1=supported-platforms&Version=2016-11-15"},
|
|
Response: &awsMockResponse{200, test_ec2_describeAccountAttributes_response, "text/xml"},
|
|
},
|
|
}
|
|
closeFunc, sess, err := getMockedAwsApiSession("EC2", ec2Endpoints)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer closeFunc()
|
|
conn := ec2.New(sess)
|
|
|
|
platforms, err := GetSupportedEC2Platforms(conn)
|
|
if err != nil {
|
|
t.Fatalf("Expected no error, received: %s", err)
|
|
}
|
|
expectedPlatforms := []string{"VPC", "EC2"}
|
|
if !reflect.DeepEqual(platforms, expectedPlatforms) {
|
|
t.Fatalf("Received platforms: %q\nExpected: %q\n", platforms, expectedPlatforms)
|
|
}
|
|
}
|
|
|
|
// getMockedAwsApiSession establishes a httptest server to simulate behaviour
|
|
// of a real AWS API server
|
|
func getMockedAwsApiSession(svcName string, endpoints []*awsMockEndpoint) (func(), *session.Session, error) {
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
buf := new(bytes.Buffer)
|
|
buf.ReadFrom(r.Body)
|
|
requestBody := buf.String()
|
|
|
|
log.Printf("[DEBUG] Received %s API %q request to %q: %s",
|
|
svcName, r.Method, r.RequestURI, requestBody)
|
|
|
|
for _, e := range endpoints {
|
|
if r.Method == e.Request.Method && r.RequestURI == e.Request.Uri && requestBody == e.Request.Body {
|
|
log.Printf("[DEBUG] Mocked %s API responding with %d: %s",
|
|
svcName, e.Response.StatusCode, e.Response.Body)
|
|
|
|
w.WriteHeader(e.Response.StatusCode)
|
|
w.Header().Set("Content-Type", e.Response.ContentType)
|
|
w.Header().Set("X-Amzn-Requestid", "1b206dd1-f9a8-11e5-becf-051c60f11c4a")
|
|
w.Header().Set("Date", time.Now().Format(time.RFC1123))
|
|
|
|
fmt.Fprintln(w, e.Response.Body)
|
|
return
|
|
}
|
|
}
|
|
|
|
w.WriteHeader(400)
|
|
return
|
|
}))
|
|
|
|
sc := awsCredentials.NewStaticCredentials("accessKey", "secretKey", "")
|
|
|
|
sess, err := session.NewSession(&aws.Config{
|
|
Credentials: sc,
|
|
Region: aws.String("us-east-1"),
|
|
Endpoint: aws.String(ts.URL),
|
|
CredentialsChainVerboseErrors: aws.Bool(true),
|
|
})
|
|
|
|
return ts.Close, sess, err
|
|
}
|
|
|
|
type awsMockEndpoint struct {
|
|
Request *awsMockRequest
|
|
Response *awsMockResponse
|
|
}
|
|
|
|
type awsMockRequest struct {
|
|
Method string
|
|
Uri string
|
|
Body string
|
|
}
|
|
|
|
type awsMockResponse struct {
|
|
StatusCode int
|
|
Body string
|
|
ContentType string
|
|
}
|
|
|
|
var test_ec2_describeAccountAttributes_response = `<DescribeAccountAttributesResponse xmlns="http://ec2.amazonaws.com/doc/2016-11-15/">
|
|
<requestId>7a62c49f-347e-4fc4-9331-6e8eEXAMPLE</requestId>
|
|
<accountAttributeSet>
|
|
<item>
|
|
<attributeName>supported-platforms</attributeName>
|
|
<attributeValueSet>
|
|
<item>
|
|
<attributeValue>VPC</attributeValue>
|
|
</item>
|
|
<item>
|
|
<attributeValue>EC2</attributeValue>
|
|
</item>
|
|
</attributeValueSet>
|
|
</item>
|
|
</accountAttributeSet>
|
|
</DescribeAccountAttributesResponse>`
|