test: add unknown error mixed header cases

aws · Feb 27, 2023 · 6697bc6 · 6697bc6
1 parent d087d7c
commit 6697bc6
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/private/protocol/restjson/unmarshal_error_test.go b/private/protocol/restjson/unmarshal_error_test.go
@@ -161,6 +161,32 @@ func TestUnmarshalTypedError(t *testing.T) {
 				respMeta.RequestID,
 			),
 		},
+		"unknown code header only": {
+			Response: &http.Response{
+				Header: http.Header{
+					errorTypeHeader: []string{"UnknownError"},
+				},
+				Body: ioutil.NopCloser(strings.NewReader(unknownErrJSON)),
+			},
+			Expect: awserr.NewRequestFailure(
+				awserr.New("UnknownError", "error message", nil),
+				respMeta.StatusCode,
+				respMeta.RequestID,
+			),
+		},
+		"unknown message header only": {
+			Response: &http.Response{
+				Header: http.Header{
+					errorMessageHeader: []string{"overwritten error message"},
+				},
+				Body: ioutil.NopCloser(strings.NewReader(unknownErrJSON)),
+			},
+			Expect: awserr.NewRequestFailure(
+				awserr.New("UnknownError", "error message", nil),
+				respMeta.StatusCode,
+				respMeta.RequestID,
+			),
+		},
 		"code from header": {
 			Response: &http.Response{
 				Header: http.Header{