@@ -23,8 +23,8 @@ import (
2323 "github.com/aws/aws-sdk-go-v2/service/ec2"
2424 "github.com/aws/aws-sdk-go-v2/service/ec2/types"
2525 "github.com/aws/aws-sdk-go/aws"
26- "github.com/aws/aws-sdk-go/aws/awserr"
2726 "github.com/aws/aws-sdk-go/aws/request"
27+ "github.com/aws/smithy-go"
2828 "github.com/golang/mock/gomock"
2929 . "github.com/onsi/gomega"
3030 "github.com/pkg/errors"
@@ -33,7 +33,6 @@ import (
3333 "sigs.k8s.io/controller-runtime/pkg/client/fake"
3434
3535 infrav1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta2"
36- "sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/awserrors"
3736 "sigs.k8s.io/cluster-api-provider-aws/v2/pkg/cloud/scope"
3837 "sigs.k8s.io/cluster-api-provider-aws/v2/test/mocks"
3938 clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
@@ -414,7 +413,7 @@ func TestReconcileVPC(t *testing.T) {
414413 Values : []string {string (types .VpcStatePending ), string (types .VpcStateAvailable )},
415414 },
416415 },
417- })).Return (nil , awserr . New ( "404" , "http not found err" , errors . New ( "err" )) )
416+ })).Return (nil , & smithy. GenericAPIError { Code : "404" , Message : "http not found err" } )
418417 },
419418 },
420419 {
@@ -464,7 +463,7 @@ func TestReconcileVPC(t *testing.T) {
464463 },
465464 },
466465 }, nil )
467- m .DescribeVpcAttribute (context .TODO (), gomock .AssignableToTypeOf (& ec2.DescribeVpcAttributeInput {})).Return (nil , awserr . New ( "InvalidVpcID.NotFound" , "not found" , nil ) )
466+ m .DescribeVpcAttribute (context .TODO (), gomock .AssignableToTypeOf (& ec2.DescribeVpcAttributeInput {})).Return (nil , & smithy. GenericAPIError { Code : "InvalidVpcID.NotFound" , Message : "not found" } )
468467 m .DescribeVpcAttribute (context .TODO (), gomock .AssignableToTypeOf (& ec2.DescribeVpcAttributeInput {})).
469468 DoAndReturn (describeVpcAttributeTrue ).AnyTimes ()
470469 },
@@ -494,7 +493,7 @@ func TestReconcileVPC(t *testing.T) {
494493 },
495494 },
496495 }, nil )
497- m .DescribeVpcAttribute (context .TODO (), gomock .AssignableToTypeOf (& ec2.DescribeVpcAttributeInput {})).AnyTimes ().Return (nil , awserrors . NewFailedDependency ( " failed dependency") )
496+ m .DescribeVpcAttribute (context .TODO (), gomock .AssignableToTypeOf (& ec2.DescribeVpcAttributeInput {})).AnyTimes ().Return (nil , & smithy. GenericAPIError { Code : "FailedDependency" , Message : " failed dependency"} )
498497 },
499498 },
500499 {
@@ -510,7 +509,7 @@ func TestReconcileVPC(t *testing.T) {
510509 },
511510 },
512511 })).Return (& ec2.DescribeVpcsOutput {Vpcs : []types.Vpc {}}, nil )
513- m .CreateVpc (context .TODO (), gomock .AssignableToTypeOf (& ec2.CreateVpcInput {})).After (describeVPCByNameCall ).Return (nil , awserrors . NewFailedDependency ( " failed dependency") )
512+ m .CreateVpc (context .TODO (), gomock .AssignableToTypeOf (& ec2.CreateVpcInput {})).After (describeVPCByNameCall ).Return (nil , & smithy. GenericAPIError { Code : "FailedDependency" , Message : " failed dependency"} )
514513 },
515514 },
516515 {
@@ -610,7 +609,7 @@ func TestDeleteVPC(t *testing.T) {
610609 expect : func (m * mocks.MockEC2APIMockRecorder ) {
611610 m .DeleteVpc (context .TODO (), gomock .Eq (& ec2.DeleteVpcInput {
612611 VpcId : aws .String ("managed-vpc" ),
613- })).Return (nil , awserrors . NewFailedDependency ( " failed dependency") )
612+ })).Return (nil , & smithy. GenericAPIError { Code : "FailedDependency" , Message : " failed dependency"} )
614613 },
615614 },
616615 {
@@ -636,7 +635,7 @@ func TestDeleteVPC(t *testing.T) {
636635 expect : func (m * mocks.MockEC2APIMockRecorder ) {
637636 m .DeleteVpc (context .TODO (), gomock .Eq (& ec2.DeleteVpcInput {
638637 VpcId : aws .String ("managed-vpc" ),
639- })).Return (nil , awserr . New ( "InvalidVpcID.NotFound" , "not found" , nil ) )
638+ })).Return (nil , & smithy. GenericAPIError { Code : "InvalidVpcID.NotFound" , Message : "not found" } )
640639 },
641640 },
642641 }
0 commit comments