Skip to content

Commit 0256f18

Browse files
fix e2e job
1 parent 86dd894 commit 0256f18

File tree

2 files changed

+8
-9
lines changed

2 files changed

+8
-9
lines changed

pkg/cloud/services/network/eips_test.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,9 @@ 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"
26+
"github.com/aws/smithy-go"
2727
"github.com/golang/mock/gomock"
2828
. "github.com/onsi/gomega"
29-
"github.com/pkg/errors"
3029
"k8s.io/apimachinery/pkg/runtime"
3130
"sigs.k8s.io/controller-runtime/pkg/client/fake"
3231

@@ -104,7 +103,7 @@ func TestServiceReleaseAddresses(t *testing.T) {
104103
},
105104
}, nil)
106105
m.DisassociateAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.DisassociateAddressInput{})).Return(nil, nil).Times(2)
107-
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, awserr.New(awserrors.AuthFailure, awserrors.AuthFailure, errors.Errorf(awserrors.AuthFailure)))
106+
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, &smithy.GenericAPIError{Code: awserrors.AuthFailure, Message: awserrors.AuthFailure})
108107
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, nil)
109108
},
110109
},
@@ -121,7 +120,7 @@ func TestServiceReleaseAddresses(t *testing.T) {
121120
},
122121
}, nil)
123122
m.DisassociateAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.DisassociateAddressInput{})).Return(nil, nil).Times(2)
124-
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, awserr.New(awserrors.InUseIPAddress, awserrors.InUseIPAddress, errors.Errorf(awserrors.InUseIPAddress)))
123+
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, &smithy.GenericAPIError{Code: awserrors.InUseIPAddress, Message: awserrors.InUseIPAddress})
125124
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, nil)
126125
},
127126
},
@@ -138,7 +137,7 @@ func TestServiceReleaseAddresses(t *testing.T) {
138137
},
139138
}, nil)
140139
m.DisassociateAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.DisassociateAddressInput{})).Return(nil, nil).Times(2)
141-
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, awserr.New("dependency-failure", "dependency-failure", errors.Errorf("dependency-failure")))
140+
m.ReleaseAddress(context.TODO(), gomock.AssignableToTypeOf(&ec2.ReleaseAddressInput{})).Return(nil, &smithy.GenericAPIError{Code: "dependency-failure", Message: "dependency-failure"})
142141
},
143142
wantErr: true,
144143
},

pkg/cloud/services/network/vpc_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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"
@@ -414,7 +414,7 @@ func TestReconcileVPC(t *testing.T) {
414414
Values: []string{string(types.VpcStatePending), string(types.VpcStateAvailable)},
415415
},
416416
},
417-
})).Return(nil, awserr.New("404", "http not found err", errors.New("err")))
417+
})).Return(nil, &smithy.GenericAPIError{Code: "404", Message: "http not found err"})
418418
},
419419
},
420420
{
@@ -464,7 +464,7 @@ func TestReconcileVPC(t *testing.T) {
464464
},
465465
},
466466
}, nil)
467-
m.DescribeVpcAttribute(context.TODO(), gomock.AssignableToTypeOf(&ec2.DescribeVpcAttributeInput{})).Return(nil, awserr.New("InvalidVpcID.NotFound", "not found", nil))
467+
m.DescribeVpcAttribute(context.TODO(), gomock.AssignableToTypeOf(&ec2.DescribeVpcAttributeInput{})).Return(nil, &smithy.GenericAPIError{Code: "InvalidVpcID.NotFound", Message: "not found"})
468468
m.DescribeVpcAttribute(context.TODO(), gomock.AssignableToTypeOf(&ec2.DescribeVpcAttributeInput{})).
469469
DoAndReturn(describeVpcAttributeTrue).AnyTimes()
470470
},
@@ -636,7 +636,7 @@ func TestDeleteVPC(t *testing.T) {
636636
expect: func(m *mocks.MockEC2APIMockRecorder) {
637637
m.DeleteVpc(context.TODO(), gomock.Eq(&ec2.DeleteVpcInput{
638638
VpcId: aws.String("managed-vpc"),
639-
})).Return(nil, awserr.New("InvalidVpcID.NotFound", "not found", nil))
639+
})).Return(nil, &smithy.GenericAPIError{Code: "InvalidVpcID.NotFound", Message: "not found"})
640640
},
641641
},
642642
}

0 commit comments

Comments
 (0)