fix trailer checksum unit tests

aws · Dec 28, 2023 · 2382d13 · 2382d13
1 parent 2e45110
commit 2382d13
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/service/internal/checksum/middleware_compute_input_checksum_test.go b/service/internal/checksum/middleware_compute_input_checksum_test.go
@@ -773,9 +773,16 @@ func TestComputeInputPayloadChecksum(t *testing.T) {
 						EnableComputePayloadHash:         true,
 						EnableDecodedContentLengthHeader: true,
 					}
+
 					if c.optionsFn != nil {
 						c.optionsFn(m)
 					}
+					trailerMiddleware := &addInputChecksumTrailer{
+						EnableTrailingChecksum:           m.EnableTrailingChecksum,
+						RequireChecksum:                  m.RequireChecksum,
+						EnableComputePayloadHash:         m.EnableComputePayloadHash,
+						EnableDecodedContentLengthHeader: m.EnableDecodedContentLengthHeader,
+					}
 
 					ctx := context.Background()
 					var logged bytes.Buffer
@@ -809,6 +816,7 @@ func TestComputeInputPayloadChecksum(t *testing.T) {
 
 					// Build middleware
 					stack.Finalize.Add(m, middleware.After)
+					stack.Finalize.Add(trailerMiddleware, middleware.After)
 
 					// Validate defer to finalize was performed as expected
 					stack.Finalize.Add(middleware.FinalizeMiddlewareFunc(