Fix tests

awslabs · Nov 13, 2024 · 21d8919 · 21d8919
1 parent 6896502
commit 21d8919
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...entRuntime/Tests/AWSClientRuntimeTests/Middlewares/FlexibleChecksumsMiddlewareTests.swift b/...entRuntime/Tests/AWSClientRuntimeTests/Middlewares/FlexibleChecksumsMiddlewareTests.swift
@@ -267,7 +267,7 @@ class FlexibleChecksumsMiddlewareTests: XCTestCase {
             "Stream is not 'chunked eligible'",
             file: file, line: line
         )
-        if let validatedChecksum = self.builtContext.attributes.get(key: AttributeKey<String>(name: "ChecksumHeaderValidated")), validatedChecksum == expectedHeader {
+        if let validatedChecksum = self.builtContext.get(key: AttributeKey<String>(name: "ChecksumHeaderValidated")), validatedChecksum == expectedHeader {
             isChecksumValidated = true
         }
         if expectedChecksum != "" {
@@ -304,7 +304,7 @@ class FlexibleChecksumsMiddlewareTests: XCTestCase {
             }
         }
 
-        if let validatedChecksum = self.builtContext.attributes.get(key: AttributeKey<String>(name: "ChecksumHeaderValidated")), validatedChecksum == expectedHeader {
+        if let validatedChecksum = self.builtContext.get(key: AttributeKey<String>(name: "ChecksumHeaderValidated")), validatedChecksum == expectedHeader {
             isChecksumValidated = true
         }
         if expectedChecksum != "" {
@@ -329,7 +329,7 @@ class FlexibleChecksumsMiddlewareTests: XCTestCase {
             }
         } else {
             try await handleMiddleware()
-            validatedChecksum = self.builtContext.attributes.get(key: AttributeKey<String>(name: "ChecksumHeaderValidated"))
+            validatedChecksum = self.builtContext.get(key: AttributeKey<String>(name: "ChecksumHeaderValidated"))
             if validatedChecksum != nil && validatedChecksum == expectedValidationHeader {
                 isChecksumValidated = true
             }