| /* |
| Copyright 2018 The Kubernetes Authors. |
| |
| Licensed under the Apache License, Version 2.0 (the "License"); |
| you may not use this file except in compliance with the License. |
| You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| */ |
| |
| package azure |
| |
| import ( |
| "testing" |
| |
| "github.com/stretchr/testify/assert" |
| ) |
| |
| func TestExtractVmssVMName(t *testing.T) { |
| cases := []struct { |
| description string |
| vmName string |
| expectError bool |
| expectedScaleSet string |
| expectedInstanceID string |
| }{ |
| { |
| description: "wrong vmss VM name should report error", |
| vmName: "vm1234", |
| expectError: true, |
| }, |
| { |
| description: "wrong VM name separator should report error", |
| vmName: "vm-1234", |
| expectError: true, |
| }, |
| { |
| description: "correct vmss VM name should return correct scaleSet and instanceID", |
| vmName: "vm_1234", |
| expectedScaleSet: "vm", |
| expectedInstanceID: "1234", |
| }, |
| { |
| description: "correct vmss VM name with Extra Separator should return correct scaleSet and instanceID", |
| vmName: "vm_test_1234", |
| expectedScaleSet: "vm_test", |
| expectedInstanceID: "1234", |
| }, |
| } |
| |
| for _, c := range cases { |
| ssName, instanceID, err := extractVmssVMName(c.vmName) |
| if c.expectError { |
| assert.Error(t, err, c.description) |
| continue |
| } |
| |
| assert.Equal(t, c.expectedScaleSet, ssName, c.description) |
| assert.Equal(t, c.expectedInstanceID, instanceID, c.description) |
| } |
| } |