add expected signed retry header in unit tests

aws · Dec 28, 2023 · 1f15cc0 · 1f15cc0
1 parent 2382d13
commit 1f15cc0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/service/s3/internal/customizations/presign_test.go b/service/s3/internal/customizations/presign_test.go
@@ -41,6 +41,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
@@ -62,6 +63,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
@@ -84,6 +86,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
@@ -105,6 +108,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Host": []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
 			},
 		},
@@ -123,6 +127,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Host": []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
 			},
 		},
@@ -142,6 +147,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
 				"Content-Length": []string{"100"},
 			},
@@ -163,6 +169,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mfzwi23gnjvgw.mrap.accesspoint.s3-global.amazonaws.com"},
@@ -204,6 +211,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
@@ -228,6 +236,7 @@ func TestPutObject_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Host": []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
 			},
 		},
@@ -315,6 +324,7 @@ func TestUploadPart_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
@@ -340,6 +350,7 @@ func TestUploadPart_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
@@ -366,6 +377,7 @@ func TestUploadPart_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Content-Length": []string{"11"},
 				"Content-Type":   []string{"application/octet-stream"},
 				"Host":           []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
@@ -391,6 +403,7 @@ func TestUploadPart_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Host": []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
 			},
 		},
@@ -413,6 +426,7 @@ func TestUploadPart_PresignURL(t *testing.T) {
 			},
 			expectMethod: "PUT",
 			expectSignedHeader: http.Header{
+				"Amz-Sdk-Request": []string{"attempt=1; max=1"},
 				"Host": []string{"mock-bucket.s3.us-west-2.amazonaws.com"},
 			},
 		},