1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
package customizations_test
import (
"context"
"errors"
"net/http"
"net/http/httptest"
"testing"
"time"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/service/ec2"
"github.com/aws/smithy-go"
)
func Test_EmptyResponse(t *testing.T) {
cases := map[string]struct {
status int
responseBody []byte
expectError bool
}{
"success case with no response body": {
status: 200,
responseBody: []byte(``),
},
"error case with no response body": {
status: 400,
responseBody: []byte(``),
expectError: true,
},
}
for name, c := range cases {
t.Run(name, func(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(
func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(c.status)
w.Write(c.responseBody)
}))
defer server.Close()
ctx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
defer cancelFn()
cfg := aws.Config{
Region: "us-east-1",
EndpointResolver: aws.EndpointResolverFunc(func(service, region string) (aws.Endpoint, error) {
return aws.Endpoint{
URL: server.URL,
SigningName: "ec2",
}, nil
}),
Retryer: func() aws.Retryer {
return aws.NopRetryer{}
},
Credentials: &fakeCredentials{},
}
client := ec2.NewFromConfig(cfg)
params := &ec2.DeleteFleetsInput{
FleetIds: []string{"mockid"},
TerminateInstances: aws.Bool(true),
}
_, err := client.DeleteFleets(ctx, params)
if c.expectError {
var apiErr smithy.APIError
if !errors.As(err, &apiErr) {
t.Fatalf("expect error to be API error, was not, %v", err)
}
if len(apiErr.ErrorCode()) == 0 {
t.Errorf("expect non-empty error code")
}
if len(apiErr.ErrorMessage()) == 0 {
t.Errorf("expect non-empty error message")
}
} else {
if err != nil {
t.Errorf("expected no error, got %v", err.Error())
}
}
})
}
}
type fakeCredentials struct{}
func (*fakeCredentials) Retrieve(_ context.Context) (aws.Credentials, error) {
return aws.Credentials{}, nil
}
|
