dev

Author: huiii99

src/azure-cli/azure/cli/command_modules/vm/custom.py

@@ -1665,8 +1665,9 @@ def resize_vm(cmd, resource_group_name, vm_name, size, no_wait=False):
     from .operations.vm import VMUpdate as _VMUpdate, VMShow as _VMShow
 
     class VMUpdate(_VMUpdate):
-        def __init__(self):
-            self._skip_update = None
+        def __init__(self, cli_ctx):
+            super().__init__(cli_ctx=cli_ctx)
+            self._skip_update = False
 
         def pre_instance_update(self, instance):
             current_size = instance.properties.hardware_profile.vm_size
@@ -1683,15 +1684,19 @@ def pre_instance_update(self, instance):
             if current_size_str == size:
                 logger.warning("VM is already %s", size)
                 self._skip_update = True
-                return
+                return instance
 
             instance.properties.hardware_profile.vm_size = size
             logger.info("Resizing VM from %s to %s", current_size_str, size)
+            return instance
 
         def post_operations(self, *args, **kwargs):
             if getattr(self, "_skip_update", False):
                 return None
 
+            if no_wait:
+                return None
+
             return _VMShow(cli_ctx=cmd.cli_ctx)(command_args={
                 'vm_name': vm_name,
                 'resource_group': resource_group_name,