mocked_response_test.go

package awsmocker

import (
	"testing"

	"github.com/stretchr/testify/require"
	"golang.org/x/exp/maps"
)

func TestMockedResponse_getResponse(t *testing.T) {

	fakePayload := map[string]any{
		"foo": map[string]any{
			"baz": "thing",
		},
		"bar": 1234,
		"b":   true,
		"bf":  false,
		"str": "some string",
	}

	tables := []struct {
		name string
		mr   *MockedResponse
		rr   *ReceivedRequest
		exp  func(*testing.T, *httpResponse)
	}{
		{
			name: "XMLWrapped",
			mr: &MockedResponse{
				Body:     maps.Clone(fakePayload),
				Encoding: ResponseEncodingXML,
			},
			rr: &ReceivedRequest{
				Service: "sts",
				Action:  "GetCallerIdentity",
			},
			exp: func(t *testing.T, hr *httpResponse) {
				require.Equal(t, ContentTypeXML, hr.contentType)
				bodyRaw := string(hr.bodyRaw)
				require.Contains(t, bodyRaw, "<GetCallerIdentityResult>")
				require.Contains(t, bodyRaw, "<GetCallerIdentityResponse>")
			},
		},
	}

	for _, table := range tables {
		t.Run(table.name, func(t *testing.T) {
			hr := table.mr.getResponse(table.rr)
			table.exp(t, hr)
		})
	}
}