lib/fog/azurerm/requests/compute/generalize_virtual_machine.rb in fog-azure-rm-0.0.4 vs lib/fog/azurerm/requests/compute/generalize_virtual_machine.rb in fog-azure-rm-0.0.5

@@ -1,26 +1,25 @@
 module Fog
   module Compute
     class AzureRM
-      # This class provides the actual implemention for service calls.
+      # This class provides the actual implementation for service calls.
       class Real
         def generalize_virtual_machine(resource_group, name)
-          Fog::Logger.debug "Generalizing Virtual Machine #{name} in Resource Group #{resource_group}."
+          msg = "Generalizing Virtual Machine #{name} in Resource Group #{resource_group}"
+          Fog::Logger.debug msg
           begin
-            promise = @compute_mgmt_client.virtual_machines.generalize(resource_group, name)
-            promise.value!
-            Fog::Logger.debug "Virtual Machine #{name} Generalized Successfully."
-            true
+            @compute_mgmt_client.virtual_machines.generalize(resource_group, name)
           rescue  MsRestAzure::AzureOperationError => e
-            msg = "Error Generalizing Virtual Machine '#{name}' in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
-            raise msg
+            raise_azure_exception(e, msg)
           end
+          Fog::Logger.debug "Virtual Machine #{name} Generalized Successfully."
+          true
         end
       end
       # This class provides the mock implementation for unit tests.
       class Mock
-        def generalize_virtual_machine(resource_group, name)
-          Fog::Logger.debug "Virtual Machine #{name} from Resource group #{resource_group} Generalized successfully."
+        def generalize_virtual_machine(*)
+          Fog::Logger.debug 'Virtual Machine fog-test-server from Resource group fog-test-rg Generalized successfully.'
           true
         end
       end
     end
   end