# coding: utf-8 # ------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for # license information. # -------------------------------------------------------------------------- # covered ops: # virtual_machine_scale_sets: 20/22 # virtual_machine_scale_set_vms: 14/14 # virtual_machine_scale_set_vm_extensions: 0/5 # virtual_machine_scale_set_rolling_upgrades: 4/4 # virtual_machine_scale_set_extensions: 5/5 import os import time import unittest import pytest import azure.mgmt.compute from azure.core.exceptions import HttpResponseError, ResourceExistsError from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy AZURE_LOCATION = "eastus2" @pytest.mark.live_test_only class TestMgmtCompute(AzureMgmtRecordedTestCase): def setup_method(self, method): self.mgmt_client = self.create_mgmt_client(azure.mgmt.compute.ComputeManagementClient) if self.is_live: from azure.mgmt.network import NetworkManagementClient self.network_client = self.create_mgmt_client(NetworkManagementClient) def create_virtual_network(self, group_name, location, network_name, subnet_name): azure_operation_poller = self.network_client.virtual_networks.begin_create_or_update( group_name, network_name, {"location": location, "address_space": {"address_prefixes": ["10.0.0.0/16"]}}, ) result_create = azure_operation_poller.result() async_subnet_creation = self.network_client.subnets.begin_create_or_update( group_name, network_name, subnet_name, {"address_prefix": "10.0.0.0/24"} ) subnet_info = async_subnet_creation.result() return subnet_info def create_public_ip_address(self, group_name, location, public_ip_address_name): # Create public IP address defaults[put] BODY = { "public_ip_allocation_method": "Static", "idle_timeout_in_minutes": 10, "public_ip_address_version": "IPv4", "location": location, "sku": {"name": "Standard"}, } result = self.network_client.public_ip_addresses.begin_create_or_update( group_name, public_ip_address_name, BODY ) result = result.result() def create_load_balance_probe(self, group_name, location): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = group_name PUBLIC_IP_ADDRESS_NAME = "public_ip_address_name" LOAD_BALANCER_NAME = "myLoadBalancer" INBOUND_NAT_RULE_NAME = "myInboundNatRule" FRONTEND_IPCONFIGURATION_NAME = "myFrontendIpconfiguration" BACKEND_ADDRESS_POOL_NAME = "myBackendAddressPool" LOAD_BALANCING_RULE_NAME = "myLoadBalancingRule" OUTBOUND_RULE_NAME = "myOutboundRule" PROBE_NAME = "myProbe" self.create_public_ip_address(RESOURCE_GROUP, location, PUBLIC_IP_ADDRESS_NAME) # Create load balancer BODY = { "location": location, "sku": {"name": "Standard"}, "frontendIPConfigurations": [ { "name": FRONTEND_IPCONFIGURATION_NAME, # "subnet": { # "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + VIRTUAL_NETWORK_NAME + "/subnets/" + SUBNET_NAME # } "public_ip_address": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/publicIPAddresses/" + PUBLIC_IP_ADDRESS_NAME }, } ], "backend_address_pools": [{"name": BACKEND_ADDRESS_POOL_NAME}], "load_balancing_rules": [ { "name": LOAD_BALANCING_RULE_NAME, "frontend_ip_configuration": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/loadBalancers/" + LOAD_BALANCER_NAME + "/frontendIPConfigurations/" + FRONTEND_IPCONFIGURATION_NAME }, "frontend_port": "80", "backend_port": "80", "enable_floating_ip": True, "idle_timeout_in_minutes": "15", "protocol": "Tcp", "load_distribution": "Default", "disable_outbound_snat": True, "backend_address_pool": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/loadBalancers/" + LOAD_BALANCER_NAME + "/backendAddressPools/" + BACKEND_ADDRESS_POOL_NAME }, "probe": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/loadBalancers/" + LOAD_BALANCER_NAME + "/probes/" + PROBE_NAME }, } ], "probes": [ { "name": PROBE_NAME, "protocol": "Http", "port": "80", "request_path": "healthcheck.aspx", "interval_in_seconds": "15", "number_of_probes": "2", } ], "outbound_rules": [ { "name": OUTBOUND_RULE_NAME, "backend_address_pool": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/loadBalancers/" + LOAD_BALANCER_NAME + "/backendAddressPools/" + BACKEND_ADDRESS_POOL_NAME }, "frontend_ip_configurations": [ { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/loadBalancers/" + LOAD_BALANCER_NAME + "/frontendIPConfigurations/" + FRONTEND_IPCONFIGURATION_NAME } ], "protocol": "All", } ], } result = self.network_client.load_balancers.begin_create_or_update( resource_group_name=RESOURCE_GROUP, load_balancer_name=LOAD_BALANCER_NAME, parameters=BODY ) result = result.result() return ( "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/loadBalancers/" + LOAD_BALANCER_NAME + "/probes/myProbe", "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/loadBalancers/" + LOAD_BALANCER_NAME + "/backendAddressPools/" + BACKEND_ADDRESS_POOL_NAME, ) @unittest.skip("skip temporary") @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_rolling_upgrades(self, resource_group): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = resource_group.name VIRTUAL_MACHINE_SCALE_SET_NAME = self.get_resource_name("virtualmachinescaleset") NETWORK_NAME = self.get_resource_name("networknamex") SUBNET_NAME = self.get_resource_name("subnetnamex") if self.is_live: SUBNET = self.create_virtual_network(RESOURCE_GROUP, AZURE_LOCATION, NETWORK_NAME, SUBNET_NAME) else: SUBNET = "subneturi" BODY = { "sku": {"tier": "Standard", "capacity": "1", "name": "Standard_D1_v2"}, "location": "eastus2", "overprovision": True, "virtual_machine_profile": { "storage_profile": { "image_reference": { "sku": "2016-Datacenter", "publisher": "MicrosoftWindowsServer", "version": "latest", "offer": "WindowsServer", }, "os_disk": { "caching": "ReadWrite", "managed_disk": {"storage_account_type": "Standard_LRS"}, "create_option": "FromImage", "disk_size_gb": "512", }, }, "os_profile": { "computer_name_prefix": "testPC", "admin_username": "testuser", "admin_password": "Aa!1()-xyz", }, "network_profile": { "network_interface_configurations": [ { "name": "testPC", "primary": True, "enable_ipforwarding": True, "ip_configurations": [ { "name": "testPC", "properties": { "subnet": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + NETWORK_NAME + "/subnets/" + SUBNET_NAME + "" } }, } ], } ] }, }, "upgrade_policy": { "mode": "Manual", "rolling_upgrade_policy": { "max_unhealthy_upgraded_instance_percent": 100, "max_unhealthy_instance_percent": 100, }, }, "upgrade_mode": "Manual", } result = self.mgmt_client.virtual_machine_scale_sets.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # # Start an extension rolling upgrade.[post] result = self.mgmt_client.virtual_machine_scale_set_rolling_upgrades.begin_start_extension_upgrade( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # Message: Rolling Upgrade failed after exceeding the MaxUnhealthyInstancePercent value defined in the RollingUpgradePolicy. 100% of instances are in an unhealthy state after being upgraded - more than the threshold of 20% configured in the RollingUpgradePolicy. The most impactful error is: Instance found to be unhealthy or unreachable. For details on rolling upgrades, use http://aka.ms/AzureVMSSRollingUpgrade # Start vmss os upgrade (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_rolling_upgrades.begin_start_os_upgrade( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) # result = result.result() # Don't need result here # Get virtual machine scale set latest rolling upgrade (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_rolling_upgrades.get_latest( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) # Cancel vmss upgrade (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_rolling_upgrades.begin_cancel( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # Delete virtual machine set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() @unittest.skip("The entity was not found in this Azure location.") @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_extension(self, resource_group): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = resource_group.name VIRTUAL_MACHINE_SCALE_SET_NAME = self.get_resource_name("virtualmachinescaleset") VIRTUAL_MACHINE_EXTENSION_NAME = self.get_resource_name("vmssextensionx") NETWORK_NAME = self.get_resource_name("networknamex") SUBNET_NAME = self.get_resource_name("subnetnamex") if self.is_live: SUBNET = self.create_virtual_network(RESOURCE_GROUP, AZURE_LOCATION, NETWORK_NAME, SUBNET_NAME) # Create a scale set with empty data disks on each vm.[put] BODY = { "sku": {"tier": "Standard", "capacity": "1", "name": "Standard_D1_v2"}, "location": "eastus2", "overprovision": True, "virtual_machine_profile": { "storage_profile": { "image_reference": { "sku": "2016-Datacenter", "publisher": "MicrosoftWindowsServer", "version": "latest", "offer": "WindowsServer", }, "os_disk": { "caching": "ReadWrite", "managed_disk": {"storage_account_type": "Standard_LRS"}, "create_option": "FromImage", "disk_size_gb": "512", }, }, "os_profile": { "computer_name_prefix": "testPC", "admin_username": "testuser", "admin_password": "Aa!1()-xyz", }, "network_profile": { "network_interface_configurations": [ { "name": "testPC", "primary": True, "enable_ipforwarding": True, "ip_configurations": [ { "name": "testPC", "properties": { "subnet": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + NETWORK_NAME + "/subnets/" + SUBNET_NAME + "" } }, } ], } ] }, }, "upgrade_policy": {"mode": "Manual"}, "upgrade_mode": "Manual", } result = self.mgmt_client.virtual_machine_scale_sets.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # Get virtual machine scale set vm instance view (TODO: need swagger file) if self.is_live: time.sleep(180) for i in range(4): instance_id = i try: result = self.mgmt_client.virtual_machine_scale_set_vms.get_instance_view( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, instance_id ) except HttpResponseError: if instance_id >= 3: raise Exception("Can not get instance_id") else: break INSTANCE_ID = instance_id # Create VMSS vm extension (TODO: need swagger file) BODY = { "location": "eastus2", "auto_upgrade_minor_version": False, "publisher": "Microsoft.Azure.NetworkWatcher", "virtual_machine_extension_type": "NetworkWatcherAgentWindows", "type_handler_version": "1.4", } result = self.mgmt_client.virtual_machine_scale_set_vm_extensions.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID, VIRTUAL_MACHINE_EXTENSION_NAME, BODY ) try: result = result.result() except HttpResponseError: pass for i in range(3): try: # Get VMSS vm extension (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vm_extensions.get( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID, VIRTUAL_MACHINE_EXTENSION_NAME ) except HttpResponseError: if i >= 2: raise Exception("can not get extension.") else: break # List VMSS vm extensions (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vm_extensions.list( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) # Update VMSS vm extension (TODO: need swagger file) BODY = { "auto_upgrade_minor_version": False, "publisher": "Microsoft.Azure.NetworkWatcher", "virtual_machine_extension_type": "NetworkWatcherAgentWindows", "type_handler_version": "1.4", } result = self.mgmt_client.virtual_machine_scale_set_vm_extensions.update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID, VIRTUAL_MACHINE_EXTENSION_NAME, BODY ) # Delete VMSS vm exnteison (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vm_extensions.delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID, VIRTUAL_MACHINE_EXTENSION_NAME ) # Delete virtual machine set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_vm(self, resource_group): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = resource_group.name VIRTUAL_MACHINE_SCALE_SET_NAME = self.get_resource_name("virtualmachinescaleset") NETWORK_NAME = self.get_resource_name("networknamex") SUBNET_NAME = self.get_resource_name("subnetnamex") if self.is_live: SUBNET = self.create_virtual_network(RESOURCE_GROUP, AZURE_LOCATION, NETWORK_NAME, SUBNET_NAME) BODY = { "sku": {"tier": "Standard", "capacity": "1", "name": "Standard_D1_v2"}, "location": "eastus2", "overprovision": True, "virtual_machine_profile": { "storage_profile": { "image_reference": { "sku": "2016-Datacenter", "publisher": "MicrosoftWindowsServer", "version": "latest", "offer": "WindowsServer", }, "os_disk": { "caching": "ReadWrite", "managed_disk": {"storage_account_type": "Standard_LRS"}, "create_option": "FromImage", "disk_size_gb": "512", }, }, "os_profile": { "computer_name_prefix": "testPC", "admin_username": "testuser", "admin_password": "Aa!1()-xyz", }, "network_profile": { "network_interface_configurations": [ { "name": "testPC", "primary": True, "enable_ipforwarding": True, "ip_configurations": [ { "name": "testPC", "properties": { "subnet": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + NETWORK_NAME + "/subnets/" + SUBNET_NAME + "" } }, } ], } ] }, }, "upgrade_policy": {"mode": "Manual"}, "upgrade_mode": "Manual", } result = self.mgmt_client.virtual_machine_scale_sets.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # TODO: it has a bug that doesn't send request and always returns []. # List vitual machine scale set vms (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.list( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) # INSTANCE_ID_1 = result.current_page[0].instance_id # INSTANCE_ID_2 = result.current_page[1].instance_id # INSTANCE_ID = INSTANCE_ID_1 # Get virtual machine scale set vm instance view (TODO: need swagger file) if self.is_live: time.sleep(180) for i in range(4): instance_id = i try: result = self.mgmt_client.virtual_machine_scale_set_vms.get_instance_view( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, instance_id ) except HttpResponseError: if instance_id >= 3: raise Exception("Can not get instance_id") else: break INSTANCE_ID = instance_id # Get virtual machine scale set vm (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.get( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) INSTANCE_VM_1 = result # Update a virtual machine scale set vm (TODO: need a swagger file) BODY = {"location": "eastus2", "tags": {"department": "HR"}} result = self.mgmt_client.virtual_machine_scale_set_vms.begin_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID, BODY ) result = result.result() # VirtualMachineScaleSet vm restart (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.begin_restart( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # VirtualMachineScaleSet vm power off (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.begin_power_off( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # VirtualMachineScaleSet vm start (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.begin_start( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # VMSSVMRunCommand[post] BODY = {"command_id": "RunPowerShellScript"} result = self.mgmt_client.virtual_machine_scale_set_vms.begin_run_command( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID, BODY ) result = result.result() # Update instances of machine scale set (TODO: need swagger file) BODY = {"instance_ids": [INSTANCE_ID]} result = self.mgmt_client.virtual_machine_scale_sets.begin_update_instances( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # Deallocate virtual machine scale set vm (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.begin_deallocate( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # Delete instances of virtual machine scale sets (TODO: need swagger file) BODY = {"instance_ids": [INSTANCE_ID]} result = self.mgmt_client.virtual_machine_scale_sets.begin_delete_instances( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # Delete virtual machine set vm (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # Delete virtual machine set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_vm_2(self, resource_group): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = resource_group.name VIRTUAL_MACHINE_SCALE_SET_NAME = self.get_resource_name("virtualmachinescaleset") NETWORK_NAME = self.get_resource_name("networknamex") SUBNET_NAME = self.get_resource_name("subnetnamex") if self.is_live: SUBNET = self.create_virtual_network(RESOURCE_GROUP, AZURE_LOCATION, NETWORK_NAME, SUBNET_NAME) BODY = { "sku": {"tier": "Standard", "capacity": "1", "name": "Standard_D1_v2"}, "location": "eastus2", "overprovision": True, "virtual_machine_profile": { "storage_profile": { "image_reference": { "sku": "2016-Datacenter", "publisher": "MicrosoftWindowsServer", "version": "latest", "offer": "WindowsServer", }, "os_disk": { "caching": "ReadWrite", "managed_disk": {"storage_account_type": "Standard_LRS"}, "create_option": "FromImage", "disk_size_gb": "512", }, }, "os_profile": { "computer_name_prefix": "testPC", "admin_username": "testuser", "admin_password": "Aa!1()-xyz", }, "network_profile": { "network_interface_configurations": [ { "name": "testPC", "primary": True, "enable_ipforwarding": True, "ip_configurations": [ { "name": "testPC", "properties": { "subnet": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + NETWORK_NAME + "/subnets/" + SUBNET_NAME + "" } }, } ], } ] }, }, "upgrade_policy": {"mode": "Manual"}, "upgrade_mode": "Manual", } result = self.mgmt_client.virtual_machine_scale_sets.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # Get virtual machine scale set vm instance view (TODO: need swagger file) if self.is_live: time.sleep(180) for i in range(4): instance_id = i try: result = self.mgmt_client.virtual_machine_scale_set_vms.get_instance_view( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, instance_id ) except HttpResponseError: if instance_id >= 3: raise Exception("Can not get instance_id") else: break INSTANCE_ID = instance_id # it isn't in vmss vm # # Simulate eviction # self.mgmt_client.virtual_machine_scale_set_vms.simulate_eviction(resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID) # Redeploy virtual machine scale set vm (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_vms.begin_redeploy( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # Reimage a virtual machine scale set vm (TODO: need swagger file) BODY = {"temp_disk": True} result = self.mgmt_client.virtual_machine_scale_set_vms.begin_reimage( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # Reimage all virtual machine scale sets vm (TODO: need swagger file) BODY = {"temp_disk": True} result = self.mgmt_client.virtual_machine_scale_set_vms.begin_reimage_all( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() # Delete virtual machine set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() @unittest.skip("The (VMRedeployment) need artificially generated,skip for now") @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute(self, resource_group): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = resource_group.name VIRTUAL_MACHINE_SCALE_SET_NAME = self.get_resource_name("virtualmachinescaleset") VMSS_EXTENSION_NAME = self.get_resource_name("vmssextensionx") NETWORK_NAME = self.get_resource_name("networknamex") SUBNET_NAME = self.get_resource_name("subnetnamex") if self.is_live: SUBNET = self.create_virtual_network(RESOURCE_GROUP, AZURE_LOCATION, NETWORK_NAME, SUBNET_NAME) # Create a scale set with empty data disks on each vm.[put] BODY = { "sku": {"tier": "Standard", "capacity": "2", "name": "Standard_D1_v2"}, "location": "eastus2", "overprovision": True, "virtual_machine_profile": { "storage_profile": { "image_reference": { "sku": "2016-Datacenter", "publisher": "MicrosoftWindowsServer", "version": "latest", "offer": "WindowsServer", }, # "image_reference": { # "offer": "UbuntuServer", # "publisher": "Canonical", # "sku": "18.04-LTS", # # "urn": "Canonical:UbuntuServer:18.04-LTS:latest", # # "urnAlias": "UbuntuLTS", # "version": "latest" # }, "os_disk": { "caching": "ReadWrite", "managed_disk": {"storage_account_type": "Standard_LRS"}, "create_option": "FromImage", "disk_size_gb": "512", }, # "data_disks": [ # { # "disk_size_gb": "1023", # "create_option": "Empty", # "lun": "0" # }, # { # "disk_size_gb": "1023", # "create_option": "Empty", # "lun": "1" # } # ] }, "os_profile": { "computer_name_prefix": "testPC", "admin_username": "testuser", "admin_password": "Aa!1()-xyz", }, "network_profile": { "network_interface_configurations": [ { "name": "testPC", "primary": True, "enable_ipforwarding": True, "ip_configurations": [ { "name": "testPC", "properties": { "subnet": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + NETWORK_NAME + "/subnets/" + SUBNET_NAME + "" } }, } ], } ] }, }, "upgrade_policy": {"mode": "Manual"}, "upgrade_mode": "Manual", } result = self.mgmt_client.virtual_machine_scale_sets.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # Create virtual machine sacle set extension (TODO: need swagger file) BODY = { "location": "eastus2", "auto_upgrade_minor_version": True, "publisher": "Microsoft.Azure.NetworkWatcher", "type_properties_type": "NetworkWatcherAgentWindows", "type_handler_version": "1.4", } result = self.mgmt_client.virtual_machine_scale_set_extensions.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, VMSS_EXTENSION_NAME, BODY ) result = result.result() # Get a virtual machine scale sets (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.get(resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME) # Get virtual machine scale set extension (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_extensions.get( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, VMSS_EXTENSION_NAME ) # Get virtual machine scale set os upgrade history (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.get_os_upgrade_history( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) # Get instance view of virtual machine scale set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.get_instance_view( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) # List virtual machine scale sets in a resource group (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.list(resource_group.name) # List virtual machine scale set extension (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_extensions.list( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) # List all virtual machine scale sets (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.list_all() # List virtual machine scale sets skus (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.list_skus( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) # Start an extension rolling upgrade.[post] result = self.mgmt_client.virtual_machine_scale_set_rolling_upgrades.begin_start_extension_upgrade( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # Update a virtual machine scale set (TODO: need a swagger file) BODY = { "sku": {"tier": "Standard", "capacity": "2", "name": "Standard_D1_v2"}, "upgrade_policy": {"mode": "Manual"}, } result = self.mgmt_client.virtual_machine_scale_sets.begin_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # VirtualMachineScaleSet restart (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_restart( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # VirtualMachineScaleSet power off (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_power_off( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # VirtualMachineScaleSet start (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_start( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # VirtualMachineScaleSet stop againe. result = self.mgmt_client.virtual_machine_scale_sets.begin_power_off( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # Update virtual machine sacle set extension (TODO: need swagger file) BODY = { "auto_upgrade_minor_version": True, } result = self.mgmt_client.virtual_machine_scale_set_extensions.begin_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, VMSS_EXTENSION_NAME, BODY ) result = result.result() # Delete virtua machine scale set extension (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_set_extensions.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, VMSS_EXTENSION_NAME ) result = result.result() # Redeploy virtual machine scale set (TODO: need swagger file) try: result = self.mgmt_client.virtual_machine_scale_sets.begin_redeploy( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() except HttpResponseError as e: if not str(e).startswith("(VMRedeployment)"): raise e # Deallocate virtual machine scale set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_deallocate( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # Delete virtual machine set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() @unittest.skip("hard to test") @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_base_2(self, resource_group): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = resource_group.name VIRTUAL_MACHINE_SCALE_SET_NAME = self.get_resource_name("virtualmachinescaleset") VMSS_EXTENSION_NAME = self.get_resource_name("vmssextension") NETWORK_NAME = self.get_resource_name("networknamex") SUBNET_NAME = self.get_resource_name("subnetnamex") if self.is_live: SUBNET = self.create_virtual_network(RESOURCE_GROUP, AZURE_LOCATION, NETWORK_NAME, SUBNET_NAME) probe_uri, backed_pools_uri = self.create_load_balance_probe(RESOURCE_GROUP, AZURE_LOCATION) else: SUBNET = "subneturi" probe_uri = "probe_uri" backed_pools_uri = "backed_pools_uri" # Create a scale set with empty data disks on each vm.[put] BODY = { "sku": {"tier": "Standard", "capacity": "2", "name": "Standard_D1_v2"}, "location": "eastus2", "overprovision": True, "virtual_machine_profile": { "extension_profile": {}, "storage_profile": { # "image_reference": { # "offer": "UbuntuServer", # "publisher": "Canonical", # "sku": "18.04-LTS", # "version": "latest" # }, "image_reference": { "sku": "2016-Datacenter", "publisher": "MicrosoftWindowsServer", "version": "latest", "offer": "WindowsServer", }, "os_disk": { "caching": "ReadWrite", "managed_disk": {"storage_account_type": "Standard_LRS"}, "create_option": "FromImage", "disk_size_gb": "512", }, }, "os_profile": { "computer_name_prefix": "testPC", "admin_username": "testuser", "admin_password": "Aa!1()-xyz", }, "network_profile": { "network_interface_configurations": [ { "name": "testPC", "primary": True, "enable_ipforwarding": True, "ip_configurations": [ { "name": "testPC", "subnet": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + NETWORK_NAME + "/subnets/" + SUBNET_NAME + "" }, "load_balancer_backend_address_pools": [{"id": backed_pools_uri}], } ], } ], "health_probe": {"id": probe_uri}, }, }, "upgrade_policy": {"mode": "Manual"}, "upgrade_mode": "Manual", "automatic_repairs_policy": {"enabled": True, "grace_period": "PT30M"}, } result = self.mgmt_client.virtual_machine_scale_sets.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # Set orchestration service state (TODO: need example) BODY = {"action": "Suspend", "service_name": "AutomaticRepairs"} self.mgmt_client.virtual_machine_scale_sets.begin_set_orchestration_service_state( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) # Reimage a virtual machine scale set (TODO: need swagger file) BODY = {"temp_disk": True} result = self.mgmt_client.virtual_machine_scale_sets.begin_reimage( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # Reimage all virtual machine scale sets (TODO: need swagger file) BODY = {"temp_disk": True} result = self.mgmt_client.virtual_machine_scale_sets.begin_reimage_all( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # Force recovery service fabric platform update domain walk (TODO: need swagger file) # result = self.mgmt_client.virtual_machine_scale_sets.force_recovery_service_fabric_platform_update_domain_walk(resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, 1) # Convert to single placement virtual machine scale sets (TODO: need swagger file) # result = self.mgmt_client.virtual_machine_scale_sets.convert_to_single_placement_group(resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME) # Delete virtual machine set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() # @unittest.skip(""" # can not test it, see: # https://docs.microsoft.com/en-us/azure/virtual-machine-scale-sets/virtual-machine-scale-sets-maintenance-notifications # """) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_perform_maintenance(self, resource_group): SUBSCRIPTION_ID = self.get_settings_value("SUBSCRIPTION_ID") RESOURCE_GROUP = resource_group.name VIRTUAL_MACHINE_SCALE_SET_NAME = self.get_resource_name("virtualmachinescaleset") NETWORK_NAME = self.get_resource_name("networknamex") SUBNET_NAME = self.get_resource_name("subnetnamex") if self.is_live: SUBNET = self.create_virtual_network(RESOURCE_GROUP, AZURE_LOCATION, NETWORK_NAME, SUBNET_NAME) BODY = { "sku": {"tier": "Standard", "capacity": "1", "name": "Standard_D1_v2"}, "location": "eastus2", "overprovision": True, "virtual_machine_profile": { "storage_profile": { "image_reference": { "sku": "2016-Datacenter", "publisher": "MicrosoftWindowsServer", "version": "latest", "offer": "WindowsServer", }, "os_disk": { "caching": "ReadWrite", "managed_disk": {"storage_account_type": "Standard_LRS"}, "create_option": "FromImage", "disk_size_gb": "512", }, }, "os_profile": { "computer_name_prefix": "testPC", "admin_username": "testuser", "admin_password": "Aa!1()-xyz", }, "network_profile": { "network_interface_configurations": [ { "name": "testPC", "primary": True, "enable_ipforwarding": True, "ip_configurations": [ { "name": "testPC", "properties": { "subnet": { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + NETWORK_NAME + "/subnets/" + SUBNET_NAME + "" } }, } ], } ] }, }, "upgrade_policy": {"mode": "Manual"}, "upgrade_mode": "Manual", } result = self.mgmt_client.virtual_machine_scale_sets.begin_create_or_update( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, BODY ) result = result.result() # Get virtual machine scale set vm instance view (TODO: need swagger file) if self.is_live: time.sleep(180) for i in range(4): instance_id = i try: result = self.mgmt_client.virtual_machine_scale_set_vms.get_instance_view( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, instance_id ) except HttpResponseError: if instance_id >= 3: raise Exception("Can not get instance_id") else: break INSTANCE_ID = instance_id # TODO: Operation 'performMaintenance' is not allowed on VM 'virtualmachinescalesetname_2' since the Subscription of this VM is not eligible. # Perform maintenance virtual machine scale set (TODO: need swagger file) try: result = self.mgmt_client.virtual_machine_scale_sets.begin_perform_maintenance( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result() except ResourceExistsError as e: assert str(e).startswith("(OperationNotAllowed) Operation 'performMaintenance' is not allowed on") # TODO: Operation 'performMaintenance' is not allowed on VM 'virtualmachinescalesetname_2' since the Subscription of this VM is not eligible. # Perform maintenance virtual machine scale set vms (TODO: need swagger file) try: result = self.mgmt_client.virtual_machine_scale_set_vms.begin_perform_maintenance( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME, INSTANCE_ID ) result = result.result() except ResourceExistsError as e: assert str(e).startswith("(OperationNotAllowed) Operation 'performMaintenance' is not allowed on") # Delete virtual machine set (TODO: need swagger file) result = self.mgmt_client.virtual_machine_scale_sets.begin_delete( resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME ) result = result.result()