Skip to content

Commit ef4f451

Browse files
committed
test: Add test of DescribeDhcpOptions API call that returns an error
1 parent 175074f commit ef4f451

File tree

1 file changed

+32
-0
lines changed

1 file changed

+32
-0
lines changed

pkg/cloud/services/ec2/instances_test.go

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6359,6 +6359,19 @@ func TestGetDHCPOptionSetDomainName(t *testing.T) {
63596359
expectedPrivateDNSName: nil,
63606360
mockCalls: mockedGetPrivateDNSDomainNameFromDHCPOptionsEmptyCalls,
63616361
},
6362+
{
6363+
name: "DescribeDhcpOptions call returns error, e.g. because it is denied by IAM policy",
6364+
vpcID: "vpc-exists",
6365+
dhcpOpt: &types.DhcpOptions{
6366+
DhcpConfigurations: []types.DhcpConfiguration{
6367+
{
6368+
Key: aws.String("domain-name"),
6369+
},
6370+
},
6371+
},
6372+
expectedPrivateDNSName: nil,
6373+
mockCalls: mockedGetPrivateDNSDomainNameFromDHCPOptionsErrorCalls,
6374+
},
63626375
}
63636376
for _, tc := range testsCases {
63646377
t.Run(tc.name, func(t *testing.T) {
@@ -6464,6 +6477,25 @@ func mockedGetPrivateDNSDomainNameFromDHCPOptionsEmptyCalls(m *mocks.MockEC2APIM
64646477
}, nil)
64656478
}
64666479

6480+
func mockedGetPrivateDNSDomainNameFromDHCPOptionsErrorCalls(m *mocks.MockEC2APIMockRecorder) {
6481+
m.DescribeVpcs(context.TODO(), &ec2.DescribeVpcsInput{
6482+
VpcIds: []string{"vpc-exists"},
6483+
}).Return(&ec2.DescribeVpcsOutput{
6484+
Vpcs: []types.Vpc{
6485+
{
6486+
VpcId: aws.String("vpc-exists"),
6487+
CidrBlock: aws.String("10.0.0.0/16"),
6488+
IsDefault: aws.Bool(false),
6489+
State: types.VpcStateAvailable,
6490+
DhcpOptionsId: aws.String("dopt-12345678"),
6491+
},
6492+
},
6493+
}, nil)
6494+
m.DescribeDhcpOptions(context.TODO(), &ec2.DescribeDhcpOptionsInput{
6495+
DhcpOptionsIds: []string{"dopt-12345678"},
6496+
}).Return(nil, errors.New("some error"))
6497+
}
6498+
64676499
func TestGetCapacityReservationSpecification(t *testing.T) {
64686500
mockCapacityReservationID := "cr-123"
64696501
mockCapacityReservationIDPtr := &mockCapacityReservationID

0 commit comments

Comments
 (0)