// Code generated by smithy-go-codegen DO NOT EDIT.
package awsrestjson
import (
"bytes"
"context"
"github.com/aws/aws-sdk-go-v2/aws"
awshttp "github.com/aws/aws-sdk-go-v2/aws/transport/http"
smithydocument "github.com/aws/smithy-go/document"
"github.com/aws/smithy-go/middleware"
"github.com/aws/smithy-go/ptr"
smithyrand "github.com/aws/smithy-go/rand"
smithytesting "github.com/aws/smithy-go/testing"
smithyhttp "github.com/aws/smithy-go/transport/http"
"github.com/google/go-cmp/cmp"
"github.com/google/go-cmp/cmp/cmpopts"
"io"
"io/ioutil"
"math"
"net/http"
"net/http/httptest"
"net/url"
"strconv"
"testing"
)
func TestClient_HttpStringPayload_awsRestjson1Serialize(t *testing.T) {
cases := map[string]struct {
Params *HttpStringPayloadInput
ExpectMethod string
ExpectURIPath string
ExpectQuery []smithytesting.QueryItem
RequireQuery []string
ForbidQuery []string
ExpectHeader http.Header
RequireHeader []string
ForbidHeader []string
Host *url.URL
BodyMediaType string
BodyAssert func(io.Reader) error
}{
"StringPayloadRequest": {
Params: &HttpStringPayloadInput{
Payload: ptr.String("rawstring"),
},
ExpectMethod: "POST",
ExpectURIPath: "/StringPayload",
ExpectQuery: []smithytesting.QueryItem{},
BodyAssert: func(actual io.Reader) error {
return smithytesting.CompareReaderBytes(actual, []byte(`rawstring`))
},
},
}
for name, c := range cases {
t.Run(name, func(t *testing.T) {
var actualReq *http.Request
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
actualReq = r.Clone(r.Context())
if len(actualReq.URL.RawPath) == 0 {
actualReq.URL.RawPath = actualReq.URL.Path
}
if v := actualReq.ContentLength; v != 0 {
actualReq.Header.Set("Content-Length", strconv.FormatInt(v, 10))
}
var buf bytes.Buffer
if _, err := io.Copy(&buf, r.Body); err != nil {
t.Errorf("failed to read request body, %v", err)
}
actualReq.Body = ioutil.NopCloser(&buf)
w.WriteHeader(200)
}))
defer server.Close()
serverURL := server.URL
if c.Host != nil {
u, err := url.Parse(serverURL)
if err != nil {
t.Fatalf("expect no error, got %v", err)
}
u.Path = c.Host.Path
u.RawPath = c.Host.RawPath
u.RawQuery = c.Host.RawQuery
serverURL = u.String()
}
client := New(Options{
APIOptions: []func(*middleware.Stack) error{
func(s *middleware.Stack) error {
s.Finalize.Clear()
s.Initialize.Remove(`OperationInputValidation`)
return nil
},
},
EndpointResolver: EndpointResolverFunc(func(region string, options EndpointResolverOptions) (e aws.Endpoint, err error) {
e.URL = serverURL
e.SigningRegion = "us-west-2"
return e, err
}),
HTTPClient: awshttp.NewBuildableClient(),
IdempotencyTokenProvider: smithyrand.NewUUIDIdempotencyToken(&smithytesting.ByteLoop{}),
Region: "us-west-2",
})
result, err := client.HttpStringPayload(context.Background(), c.Params)
if err != nil {
t.Fatalf("expect nil err, got %v", err)
}
if result == nil {
t.Fatalf("expect not nil result")
}
if e, a := c.ExpectMethod, actualReq.Method; e != a {
t.Errorf("expect %v method, got %v", e, a)
}
if e, a := c.ExpectURIPath, actualReq.URL.RawPath; e != a {
t.Errorf("expect %v path, got %v", e, a)
}
queryItems := smithytesting.ParseRawQuery(actualReq.URL.RawQuery)
smithytesting.AssertHasQuery(t, c.ExpectQuery, queryItems)
smithytesting.AssertHasQueryKeys(t, c.RequireQuery, queryItems)
smithytesting.AssertNotHaveQueryKeys(t, c.ForbidQuery, queryItems)
smithytesting.AssertHasHeader(t, c.ExpectHeader, actualReq.Header)
smithytesting.AssertHasHeaderKeys(t, c.RequireHeader, actualReq.Header)
smithytesting.AssertNotHaveHeaderKeys(t, c.ForbidHeader, actualReq.Header)
if c.BodyAssert != nil {
if err := c.BodyAssert(actualReq.Body); err != nil {
t.Errorf("expect body equal, got %v", err)
}
}
})
}
}
func TestClient_HttpStringPayload_awsRestjson1Deserialize(t *testing.T) {
cases := map[string]struct {
StatusCode int
Header http.Header
BodyMediaType string
Body []byte
ExpectResult *HttpStringPayloadOutput
}{
"StringPayloadResponse": {
StatusCode: 200,
Body: []byte(`rawstring`),
ExpectResult: &HttpStringPayloadOutput{
Payload: ptr.String("rawstring"),
},
},
}
for name, c := range cases {
t.Run(name, func(t *testing.T) {
serverURL := "http://localhost:8888/"
client := New(Options{
HTTPClient: smithyhttp.ClientDoFunc(func(r *http.Request) (*http.Response, error) {
headers := http.Header{}
for k, vs := range c.Header {
for _, v := range vs {
headers.Add(k, v)
}
}
if len(c.BodyMediaType) != 0 && len(headers.Values("Content-Type")) == 0 {
headers.Set("Content-Type", c.BodyMediaType)
}
response := &http.Response{
StatusCode: c.StatusCode,
Header: headers,
Request: r,
}
if len(c.Body) != 0 {
response.ContentLength = int64(len(c.Body))
response.Body = ioutil.NopCloser(bytes.NewReader(c.Body))
} else {
response.Body = http.NoBody
}
return response, nil
}),
APIOptions: []func(*middleware.Stack) error{
func(s *middleware.Stack) error {
s.Finalize.Clear()
s.Initialize.Remove(`OperationInputValidation`)
return nil
},
},
EndpointResolver: EndpointResolverFunc(func(region string, options EndpointResolverOptions) (e aws.Endpoint, err error) {
e.URL = serverURL
e.SigningRegion = "us-west-2"
return e, err
}),
IdempotencyTokenProvider: smithyrand.NewUUIDIdempotencyToken(&smithytesting.ByteLoop{}),
Region: "us-west-2",
})
var params HttpStringPayloadInput
result, err := client.HttpStringPayload(context.Background(), ¶ms)
if err != nil {
t.Fatalf("expect nil err, got %v", err)
}
if result == nil {
t.Fatalf("expect not nil result")
}
opts := cmp.Options{
cmpopts.IgnoreUnexported(
middleware.Metadata{},
),
cmp.FilterValues(func(x, y float64) bool {
return math.IsNaN(x) && math.IsNaN(y)
}, cmp.Comparer(func(_, _ interface{}) bool { return true })),
cmp.FilterValues(func(x, y float32) bool {
return math.IsNaN(float64(x)) && math.IsNaN(float64(y))
}, cmp.Comparer(func(_, _ interface{}) bool { return true })),
cmpopts.IgnoreTypes(smithydocument.NoSerde{}),
}
if err := smithytesting.CompareValues(c.ExpectResult, result, opts...); err != nil {
t.Errorf("expect c.ExpectResult value match:\n%v", err)
}
})
}
}