lib/fog/azurerm/requests/compute/restart_virtual_machine.rb in fog-azure-rm-0.0.1 vs lib/fog/azurerm/requests/compute/restart_virtual_machine.rb in fog-azure-rm-0.0.2

@@ -5,21 +5,23 @@
       class Real
         def restart_virtual_machine(resource_group, name)
           Fog::Logger.debug "Restarting Virtual Machine #{name} in Resource Group #{resource_group}."
           begin
             promise = @compute_mgmt_client.virtual_machines.restart(resource_group, name)
-            result = promise.value!
+            promise.value!
             Fog::Logger.debug "Virtual Machine #{name} Restarted Successfully."
-            result
+            true
           rescue  MsRestAzure::AzureOperationError => e
             msg = "Error Restarting Virtual Machine '#{name}' in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
             raise msg
           end
         end
       end
       # This class provides the mock implementation for unit tests.
       class Mock
         def restart_virtual_machine(resource_group, name)
+          Fog::Logger.debug "Virtual Machine #{name} from Resource group #{resource_group} Restarted successfully."
+          return true
         end
       end
     end
   end
 end