Skip to content

Commit 06dc935

Browse files
authored
Merge pull request #7281 from jackfrancis/azure-vms-pool-ut
azure: increase UT coverage in azure_vms_pool
2 parents a652aef + 0fd973a commit 06dc935

File tree

4 files changed

+265
-20
lines changed

4 files changed

+265
-20
lines changed

cluster-autoscaler/cloudprovider/azure/azure_agent_pool_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ func TestDeleteOutdatedDeployments(t *testing.T) {
172172
}
173173
}
174174

175-
func TestGetVMsFromCache(t *testing.T) {
175+
func TestAgentPoolGetVMsFromCache(t *testing.T) {
176176
ctrl := gomock.NewController(t)
177177
defer ctrl.Finish()
178178

@@ -309,7 +309,7 @@ func TestAgentPoolIncreaseSize(t *testing.T) {
309309
assert.NoError(t, err)
310310
}
311311

312-
func TestDecreaseTargetSize(t *testing.T) {
312+
func TestAgentPoolDecreaseTargetSize(t *testing.T) {
313313
ctrl := gomock.NewController(t)
314314
defer ctrl.Finish()
315315

cluster-autoscaler/cloudprovider/azure/azure_scale_set_test.go

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ func newApiNode(orchmode compute.OrchestrationMode, vmID int64) *apiv1.Node {
151151
}
152152
return node
153153
}
154-
func TestMaxSize(t *testing.T) {
154+
func TestScaleSetMaxSize(t *testing.T) {
155155
provider := newTestProvider(t)
156156
registered := provider.azureManager.RegisterNodeGroup(
157157
newTestScaleSet(provider.azureManager, "test-asg"))
@@ -160,7 +160,7 @@ func TestMaxSize(t *testing.T) {
160160
assert.Equal(t, provider.NodeGroups()[0].MaxSize(), 5)
161161
}
162162

163-
func TestMinSize(t *testing.T) {
163+
func TestScaleSetMinSize(t *testing.T) {
164164
provider := newTestProvider(t)
165165
registered := provider.azureManager.RegisterNodeGroup(
166166
newTestScaleSet(provider.azureManager, "test-asg"))
@@ -169,7 +169,7 @@ func TestMinSize(t *testing.T) {
169169
assert.Equal(t, provider.NodeGroups()[0].MinSize(), 1)
170170
}
171171

172-
func TestMinSizeZero(t *testing.T) {
172+
func TestScaleSetMinSizeZero(t *testing.T) {
173173
provider := newTestProvider(t)
174174
registered := provider.azureManager.RegisterNodeGroup(
175175
newTestScaleSetMinSizeZero(provider.azureManager, testASG))
@@ -178,7 +178,7 @@ func TestMinSizeZero(t *testing.T) {
178178
assert.Equal(t, provider.NodeGroups()[0].MinSize(), 0)
179179
}
180180

181-
func TestTargetSize(t *testing.T) {
181+
func TestScaleSetTargetSize(t *testing.T) {
182182
ctrl := gomock.NewController(t)
183183
defer ctrl.Finish()
184184

@@ -248,7 +248,7 @@ func TestTargetSize(t *testing.T) {
248248
}
249249
}
250250

251-
func TestIncreaseSize(t *testing.T) {
251+
func TestScaleSetIncreaseSize(t *testing.T) {
252252
ctrl := gomock.NewController(t)
253253
defer ctrl.Finish()
254254

@@ -365,7 +365,7 @@ func TestIncreaseSize(t *testing.T) {
365365

366366
// TestIncreaseSizeOnVMProvisioningFailed has been tweeked only for Uniform Orchestration mode.
367367
// If ProvisioningState == failed and power state is not running, Status.State == InstanceCreating with errorInfo populated.
368-
func TestIncreaseSizeOnVMProvisioningFailed(t *testing.T) {
368+
func TestScaleSetIncreaseSizeOnVMProvisioningFailed(t *testing.T) {
369369
testCases := map[string]struct {
370370
expectInstanceRunning bool
371371
isMissingInstanceView bool
@@ -543,7 +543,7 @@ func TestIncreaseSizeOnVMProvisioningFailedWithFastDelete(t *testing.T) {
543543
}
544544
}
545545

546-
func TestIncreaseSizeOnVMSSUpdating(t *testing.T) {
546+
func TestScaleSetIncreaseSizeOnVMSSUpdating(t *testing.T) {
547547
ctrl := gomock.NewController(t)
548548
defer ctrl.Finish()
549549

@@ -593,7 +593,7 @@ func TestIncreaseSizeOnVMSSUpdating(t *testing.T) {
593593
assert.NoError(t, err)
594594
}
595595

596-
func TestBelongs(t *testing.T) {
596+
func TestScaleSetBelongs(t *testing.T) {
597597
ctrl := gomock.NewController(t)
598598
defer ctrl.Finish()
599599

@@ -645,7 +645,7 @@ func TestBelongs(t *testing.T) {
645645
}
646646
}
647647

648-
func TestDeleteNodes(t *testing.T) {
648+
func TestScaleSetDeleteNodes(t *testing.T) {
649649
ctrl := gomock.NewController(t)
650650
defer ctrl.Finish()
651651

@@ -783,7 +783,7 @@ func TestDeleteNodes(t *testing.T) {
783783
}
784784
}
785785

786-
func TestDeleteNodeUnregistered(t *testing.T) {
786+
func TestScaleSetDeleteNodeUnregistered(t *testing.T) {
787787
ctrl := gomock.NewController(t)
788788
defer ctrl.Finish()
789789

@@ -896,7 +896,7 @@ func TestDeleteNodeUnregistered(t *testing.T) {
896896
}
897897
}
898898

899-
func TestDeleteInstancesWithForceDeleteEnabled(t *testing.T) {
899+
func TestScaleSetDeleteInstancesWithForceDeleteEnabled(t *testing.T) {
900900
ctrl := gomock.NewController(t)
901901
defer ctrl.Finish()
902902
manager := newTestAzureManager(t)
@@ -1010,7 +1010,7 @@ func TestDeleteInstancesWithForceDeleteEnabled(t *testing.T) {
10101010

10111011
}
10121012

1013-
func TestDeleteNoConflictRequest(t *testing.T) {
1013+
func TestScaleSetDeleteNoConflictRequest(t *testing.T) {
10141014
ctrl := gomock.NewController(t)
10151015
defer ctrl.Finish()
10161016

@@ -1071,7 +1071,7 @@ func TestDeleteNoConflictRequest(t *testing.T) {
10711071
err = scaleSet.DeleteNodes([]*apiv1.Node{node})
10721072
}
10731073

1074-
func TestId(t *testing.T) {
1074+
func TestScaleSetId(t *testing.T) {
10751075
provider := newTestProvider(t)
10761076
registered := provider.azureManager.RegisterNodeGroup(
10771077
newTestScaleSet(provider.azureManager, "test-asg"))
@@ -1080,7 +1080,7 @@ func TestId(t *testing.T) {
10801080
assert.Equal(t, provider.NodeGroups()[0].Id(), "test-asg")
10811081
}
10821082

1083-
func TestDebug(t *testing.T) {
1083+
func TestAgentPoolDebug(t *testing.T) {
10841084
asg := ScaleSet{
10851085
manager: newTestAzureManager(t),
10861086
minSize: 5,
@@ -1159,7 +1159,7 @@ func TestScaleSetNodes(t *testing.T) {
11591159

11601160
}
11611161

1162-
func TestEnableVmssFlexNodesFlag(t *testing.T) {
1162+
func TestScaleSetEnableVmssFlexNodesFlag(t *testing.T) {
11631163

11641164
// flag set to false
11651165
ctrl := gomock.NewController(t)
@@ -1191,7 +1191,7 @@ func TestEnableVmssFlexNodesFlag(t *testing.T) {
11911191
assert.NoError(t, err)
11921192
}
11931193

1194-
func TestTemplateNodeInfo(t *testing.T) {
1194+
func TestScaleSetTemplateNodeInfo(t *testing.T) {
11951195
ctrl := gomock.NewController(t)
11961196
defer ctrl.Finish()
11971197

@@ -1317,7 +1317,7 @@ func TestTemplateNodeInfo(t *testing.T) {
13171317
})
13181318

13191319
}
1320-
func TestCseErrors(t *testing.T) {
1320+
func TestScaleSetCseErrors(t *testing.T) {
13211321
errorMessage := to.StringPtr("Error Message Test")
13221322
vmssVMs := compute.VirtualMachineScaleSetVM{
13231323
Name: to.StringPtr("vmTest"),

cluster-autoscaler/cloudprovider/azure/azure_vms_pool.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ func (agentPool *VMsPool) Nodes() ([]cloudprovider.Instance, error) {
160160

161161
nodes := make([]cloudprovider.Instance, 0, len(vms))
162162
for _, vm := range vms {
163-
if len(*vm.ID) == 0 {
163+
if vm.ID == nil || len(*vm.ID) == 0 {
164164
continue
165165
}
166166
resourceID, err := convertResourceGroupNameToLower("azure://" + *vm.ID)

0 commit comments

Comments
 (0)