VM PATCH behavior

vm_test_advanced.py

import logging
logging.basicConfig(
    level=logging.DEBUG,
    format="%(asctime)s:%(levelname)s:%(name)s:%(message)s"
)

from devtools_testutils import mgmt_settings_real

sub_id = mgmt_settings_real.SUBSCRIPTION_ID
cred = mgmt_settings_real.get_azure_core_credentials()

from azure.mgmt.compute import ComputeManagementClient

client = ComputeManagementClient(cred, sub_id, logging_enable=True)

vm_rg = "lmazuel-vm"
vm_name = "LmazuelVMTest"

from azure.mgmt.core.polling.arm_polling import ARMPolling

class FastARMPolling(ARMPolling):
    def initialize(self, client, initial_response, deserialization_callback):
        super(FastARMPolling, self).initialize(client, initial_response, deserialization_callback)

        # Force the initial response to mark "start to poll one second after the initial call"
        initial_response.http_response.headers['retry-after'] = 1

    def request_status(self, status_link):
        response = super(FastARMPolling, self).request_status(status_link)

        # Force the intermediate response to mark "continue to poll one second after the previous call"
        response.http_response.headers['retry-after'] = 1
        return response


vm_parameters = {'storage_profile' : {'data_disks' : []}}
vm_op = client.virtual_machines.begin_update(vm_rg, vm_name, vm_parameters, polling=FastARMPolling())
vm_op.wait()

print("I went through")

vm_test_simple.py

import logging
logging.basicConfig(
    level=logging.DEBUG,
    format="%(asctime)s:%(levelname)s:%(name)s:%(message)s"
)

from devtools_testutils import mgmt_settings_real

sub_id = mgmt_settings_real.SUBSCRIPTION_ID
cred = mgmt_settings_real.get_azure_core_credentials()

from azure.mgmt.compute import ComputeManagementClient

client = ComputeManagementClient(cred, sub_id, logging_enable=True)

vm_rg = "lmazuel-vm"
vm_name = "LmazuelVMTest"

from azure.mgmt.core.polling.arm_polling import ARMPolling

vm_parameters = {'storage_profile' : {'data_disks' : []}}
vm_op = client.virtual_machines.begin_update(vm_rg, vm_name, vm_parameters, polling=ARMPolling(timeout=1))
vm_op.wait()

print("I went through")