Revert "Temporary fix to E2E test to unblock release"

This reverts commit 1236a3c.
aws · Aug 9, 2024 · b205e91 · b205e91
1 parent 649e4eb
commit b205e91
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/build_tools/aws-sdk-code-generator/spec/protocol_tests_spec_helper.rb b/build_tools/aws-sdk-code-generator/spec/protocol_tests_spec_helper.rb
@@ -314,7 +314,13 @@ def match_resp_data(test_case, http_resp, it)
         expected_data =
           if error_case?(test_case)
             error_shape = http_resp.context.operation.errors.find do |err|
-              err.shape.name == test_case['errorCode']
+              shape =
+                if err.shape['error']
+                  err.shape['error']['code']
+                else
+                  err.shape.name
+                end
+              shape == test_case['errorCode']
             end
             raise "Unable to find #{test_case['errorCode']} in error shapes" if error_shape.nil?