lib/fog/azurerm/requests/compute/restart_virtual_machine.rb in fog-azure-rm-0.3.1 vs lib/fog/azurerm/requests/compute/restart_virtual_machine.rb in fog-azure-rm-0.3.2

@@ -1,19 +1,27 @@
 module Fog
   module Compute
     class AzureRM
       # This class provides the actual implementation for service calls.
       class Real
-        def restart_virtual_machine(resource_group, name)
+        def restart_virtual_machine(resource_group, name, async)
           msg = "Restarting Virtual Machine #{name} in Resource Group #{resource_group}"
           Fog::Logger.debug msg
           begin
-            @compute_mgmt_client.virtual_machines.restart(resource_group, name)
+            if async
+              response = @compute_mgmt_client.virtual_machines.restart_async(resource_group, name)
+            else
+              @compute_mgmt_client.virtual_machines.restart(resource_group, name)
+            end
           rescue  MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
-          Fog::Logger.debug "Virtual Machine #{name} Restarted Successfully."
-          true
+          if async
+            response
+          else
+            Fog::Logger.debug "Virtual Machine #{name} Restarted Successfully."
+            true
+          end
         end
       end
       # This class provides the mock implementation for unit tests.
       class Mock
         def restart_virtual_machine(*)