# 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 = 'eastus' 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": "eastus", "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": "eastus", "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": "eastus", "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() @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @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": "eastus", "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": "eastus", "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() @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @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": "eastus", "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": "eastus", "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": "eastus", "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": "eastus", "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 # """) @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @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": "eastus", "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()