Updates the error code returned for Errors modeled with awsQueryError trait (#1333)

* java codegen change to override getErrorCode method for awsQuery protocol and use AwsQueryError trait

* generated clients and protocol test to use awsQueryError trait

* remove hard coded error code from web identity provider test

Author: skotambkar

codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsQuery.java

@@ -24,6 +24,7 @@
 import software.amazon.smithy.model.shapes.Shape;
 import software.amazon.smithy.model.shapes.ShapeId;
 import software.amazon.smithy.model.shapes.StructureShape;
+import software.amazon.smithy.model.traits.ErrorTrait;
 
 /**
  * Handles generating the aws query protocol for services.
@@ -174,4 +175,14 @@ public Map<String, ShapeId> getOperationErrors(GenerationContext context, Operat
 
         return errors;
     }
+
+    @Override
+    public String getErrorCode(ServiceShape service, StructureShape errorShape) {
+        Optional<AwsQueryErrorTrait> trait = errorShape.getTrait(AwsQueryErrorTrait.class);
+        if (trait.isPresent()) {
+            return trait.get().getCode();
+        }
+
+        return super.getErrorCode(service, errorShape);
+    }
 }

credentials/stscreds/web_identity_provider_test.go

@@ -85,8 +85,9 @@ func TestWebIdentityProviderRetrieve(t *testing.T) {
 					fn(&o)
 				}
 
-				if o.Retryer.IsErrorRetryable(mockErrorCode("InvalidIdentityTokenException")) != true {
-					return nil, fmt.Errorf("expected InvalidIdentityTokenException to be retryable")
+				errorCode := (&types.InvalidIdentityTokenException{}).ErrorCode()
+				if o.Retryer.IsErrorRetryable(mockErrorCode(errorCode)) != true {
+					return nil, fmt.Errorf("expected %v to be retryable", errorCode)
 				}
 
 				return &sts.AssumeRoleWithWebIdentityOutput{