api fixes

Author: alicelyy

azure/azure.go

@@ -347,7 +347,7 @@ func (a *azureOps) SetInstanceGroupSize(instanceGroupID string,
 	instanceGroupSize := int32(count)
 	agentPoolProperties := containerservice.ManagedClusterAgentPoolProfileProperties{
 		Count:  &instanceGroupSize,
-		OsType: containerservice.Linux,
+		OsType: containerservice.OSTypeLinux,
 	}
 
 	agentPool := containerservice.AgentPool{
@@ -410,7 +410,7 @@ func (a *azureOps) Create(
 			Sku:      d.Sku,
 			DiskProperties: &compute.DiskProperties{
 				CreationData: &compute.CreationData{
-					CreateOption: compute.Empty,
+					CreateOption: compute.DiskCreateOptionEmpty,
 				},
 				DiskSizeGB:                   d.DiskProperties.DiskSizeGB,
 				DiskIOPSReadWrite:            d.DiskProperties.DiskIOPSReadWrite,
@@ -850,7 +850,7 @@ func (a *azureOps) Snapshot(diskName string, readonly bool, options map[string]s
 			Location: disk.Location,
 			SnapshotProperties: &compute.SnapshotProperties{
 				CreationData: &compute.CreationData{
-					CreateOption:     compute.Copy,
+					CreateOption:     compute.DiskCreateOptionCopy,
 					SourceResourceID: disk.ID,
 				},
 			},

azure/azure_test.go

@@ -43,7 +43,7 @@ func initAzure(t *testing.T) (cloudops.Ops, map[string]interface{}) {
 			DiskMBpsReadWrite: to.Int64Ptr(550),
 		},
 		Sku: &compute.DiskSku{
-			Name: compute.PremiumLRS,
+			Name: compute.DiskStorageAccountTypesPremiumLRS,
 		},
 	}
 

azure/base_vmsclient.go

@@ -90,6 +90,6 @@ func (b *baseVMsClient) describeInstance(
 		context.Background(),
 		b.resourceGroupName,
 		instanceName,
-		compute.InstanceView,
+		compute.InstanceViewTypesInstanceView,
 	)
 }

azure/scaleset_vmsclient.go

@@ -3,7 +3,7 @@ package azure
 import (
 	"context"
 
-	"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2019-12-01/compute"
+	"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2021-07-01/compute"
 	"github.com/Azure/go-autorest/autorest"
 )
 
@@ -92,7 +92,7 @@ func (s *scaleSetVMsClient) describeInstance(
 		s.resourceGroupName,
 		s.scaleSetName,
 		instanceID,
-		compute.InstanceView,
+		compute.InstanceViewTypesInstanceView,
 	)
 }
 

azure/scaleset_vmsclient_test.go

@@ -3,7 +3,7 @@ package azure
 import (
 	"testing"
 
-	"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2019-12-01/compute"
+	"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2021-07-01/compute"
 	"github.com/Azure/go-autorest/autorest/to"
 	"github.com/stretchr/testify/require"
 )
@@ -25,8 +25,8 @@ func TestRetrieveDataDisks(t *testing.T) {
 		expectedRes []compute.DataDisk
 	}{
 		{
-			name:  "nil vm properties",
-			input: compute.VirtualMachineScaleSetVM{},
+			name:        "nil vm properties",
+			input:       compute.VirtualMachineScaleSetVM{},
 			expectedRes: []compute.DataDisk{},
 		},
 		{

azure/vmsclient.go

@@ -1,7 +1,7 @@
 package azure
 
 import (
-	"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2019-12-01/compute"
+	"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2021-07-01/compute"
 	"github.com/Azure/go-autorest/autorest"
 )