fog-azure-rm 0.4.5 → 0.4.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 285effdfb655079d88d06ecd58189657b38a9817cb3eed807cf5bb65c8c68d58
-  data.tar.gz: d0930b897fcfeb0f1407e8f8a980bd34f0c0222afae5c49d46b2ffe2dbe92a55
+  metadata.gz: 4b9efeefac7a938ff930c43843ac651fb93a38af3555b3241ec970f8aca0e412
+  data.tar.gz: 3dcbbb882dd381a9b52f174f2c62d48ea6247888033820a7b4084eeb05758cb5
 SHA512:
-  metadata.gz: d1aefc7178613404f4ef3638cb08c9fe7326b03b8d2ee165ed650bffe0020bf37e72820de71a872983f02f4deca2b3f92449a94ba07d0f52d790bbecbba29fa1
-  data.tar.gz: 9fe6cdd1730b11f9aa8235cfcbc55279401072ac46e6c694d15cfcf6d0a715320279ff6bc11e591f37b9ae034f3654004afb0bbd9029afad17cc45be93378696
+  metadata.gz: fba09be6a12ab9eb51298ed1ca9ac9a136004c0d1dec6606b4db9ca6125d1eb659c2471f7dc714178f5931f5ca70432d7dfdf2a1f17317ea877296d6f4fcb2f6
+  data.tar.gz: 1f6acc0b6182e755ad8d823bd7a9ff9ea91cc9cf92ad595d91df4258df6feb95518e83982cd5d7034547bab2dc81079b4539829d604e40a259ae5f74513d6b9a

data/CHANGELOG.md

@@ -1,4 +1,10 @@
+## 0.4.6
+
+**Added:**
+- Created Custom Fog Exceptions
+
 ## 0.4.5
+
 **Changed:**
 - Updated dependency of fog-json to (~> 1.0.2)
 

data/lib/fog/azurerm/custom_fog_errors.rb

@@ -0,0 +1,18 @@
+# This file contains any or all custom Fog errors that we create
+module Fog
+  module AzureRM
+    # This is a custom Fog exception inherited from MsRestAzure::AzureOperationError
+    class CustomAzureOperationError < MsRestAzure::AzureOperationError
+      def initialize(message, azure_exception)
+        super(azure_exception.request, azure_exception.response, azure_exception.body, "Exception in #{message}")
+      end
+    end
+
+    # This is a custom Fog exception inherited from Azure::Core::Http::HTTPError
+    class CustomAzureCoreHttpError < Azure::Core::Http::HTTPError
+      def initialize(azure_exception)
+        super(azure_exception.http_response)
+      end
+    end
+  end
+end

data/lib/fog/azurerm/utilities/general.rb

@@ -1,3 +1,5 @@
+require File.expand_path('../../custom_fog_errors.rb', __FILE__)
+
 # Pick Resource Group name from Azure Resource Id(String)
 def get_resource_group_from_id(id)
   id.split('/')[4]
@@ -47,11 +49,9 @@ def get_record_type(type)
 end
 
 def raise_azure_exception(exception, msg)
-  description = exception.is_a?(Azure::Core::Http::HTTPError) ? exception.description : exception.error_message
-  exception_message = "Exception in #{msg} #{description} Type: #{exception.class}\n#{exception.backtrace.join('\n')}"
-
-  Fog::Logger.debug exception.backtrace
-  raise exception_message
+  raise Fog::AzureRM::CustomAzureCoreHttpError.new(exception) if exception.is_a?(Azure::Core::Http::HTTPError)
+  raise Fog::AzureRM::CustomAzureOperationError.new(msg, exception) if exception.is_a?(MsRestAzure::AzureOperationError)
+  raise exception
 end
 
 # Make sure if input_params(Hash) contains all keys present in required_params(Array)

data/lib/fog/azurerm/version.rb

@@ -1,5 +1,5 @@
 module Fog
   module AzureRM
-    VERSION = '0.4.5'.freeze
+    VERSION = '0.4.6'.freeze
   end
 end

data/test/requests/application_gateway/test_check_ag_exists.rb

@@ -32,7 +32,7 @@ class TestCheckAGExists < Minitest::Test
   def test_check_app_gateway_exists_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @gateways.stub :get, response do
-      assert_raises(RuntimeError) { @service.check_ag_exists('fog-test-rg', 'fogRM-rg') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.check_ag_exists('fog-test-rg', 'fogRM-rg') }
     end
   end
 end

data/test/requests/application_gateway/test_create_or_update_application_gateway.rb

@@ -27,7 +27,7 @@ class TestCreateOrUpdateApplicationGateway < Minitest::Test
   def test_create_or_update_application_gateway_exception_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @gateways.stub :create_or_update, response do
-      assert_raises RuntimeError do
+      assert_raises MsRestAzure::AzureOperationError do
         @service.create_or_update_application_gateway(@gateway_params)
       end
     end

data/test/requests/application_gateway/test_delete_application_gateway.rb

@@ -17,7 +17,7 @@ class TestDeleteApplicationGateway < Minitest::Test
   def test_delete_application_gateway_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @gateways.stub :delete, response do
-      assert_raises(RuntimeError) { @service.delete_application_gateway('fogRM-rg', 'gateway') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.delete_application_gateway('fogRM-rg', 'gateway') }
     end
   end
 end

data/test/requests/application_gateway/test_get_application_gateway.rb

@@ -18,7 +18,7 @@ class TestGetApplicationGateway < Minitest::Test
   def test_get_application_gateway_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @gateways.stub :get, response do
-      assert_raises(RuntimeError) { @service.get_application_gateway('fog-test-rg', 'fogRM-rg') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.get_application_gateway('fog-test-rg', 'fogRM-rg') }
     end
   end
 end

data/test/requests/application_gateway/test_list_application_gateways.rb

@@ -18,7 +18,7 @@ class TestListApplicationGateways < Minitest::Test
   def test_list_application_gateways_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @gateways.stub :list_as_lazy, response do
-      assert_raises(RuntimeError) { @service.list_application_gateways('fogRM-rg') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.list_application_gateways('fogRM-rg') }
     end
   end
 end

data/test/requests/application_gateway/test_start_application_gateway.rb

@@ -17,7 +17,7 @@ class TestStartApplicationGateway < Minitest::Test
   def test_start_application_gateway_exception_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @gateways.stub :start, response do
-      assert_raises RuntimeError do
+      assert_raises MsRestAzure::AzureOperationError do
         @service.start_application_gateway('test-rg', 'test-ag')
       end
     end

data/test/requests/application_gateway/test_stop_application_gateway.rb

@@ -17,7 +17,7 @@ class TestStopApplicationGateway < Minitest::Test
   def test_stop_application_gateway_exception_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @gateways.stub :stop, response do
-      assert_raises RuntimeError do
+      assert_raises MsRestAzure::AzureOperationError do
         @service.stop_application_gateway('test-rg', 'test-ag')
       end
     end

data/test/requests/compute/test_attach_data_disk_to_vm.rb

@@ -62,7 +62,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     @virtual_machines.stub :get, @get_vm_response do
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, update_vm_response do
-          assert_raises RuntimeError do
+          assert_raises Azure::Core::Http::HTTPError do
             @service.attach_data_disk_to_vm(@input_params, false)
           end
         end
@@ -75,7 +75,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     @virtual_machines.stub :get, @get_vm_response do
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, update_vm_response do
-          assert_raises RuntimeError do
+          assert_raises Azure::Core::Http::HTTPError do
             @service.attach_data_disk_to_vm(@input_params, false)
           end
         end
@@ -86,7 +86,7 @@ class TestAttachDataDiskToVM < Minitest::Test
   def test_get_vm_failure
     get_vm_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :get, get_vm_response do
-      assert_raises RuntimeError do
+      assert_raises MsRestAzure::AzureOperationError do
         @service.attach_data_disk_to_vm(@input_params, false)
       end
     end
@@ -96,7 +96,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     storage_access_keys_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :get, @get_vm_response do
       @storage_accounts.stub :list_keys, storage_access_keys_response do
-        assert_raises RuntimeError do
+        assert_raises MsRestAzure::AzureOperationError do
           @service.attach_data_disk_to_vm(@input_params, false)
         end
       end

data/test/requests/compute/test_check_availability_set_exists.rb

@@ -32,7 +32,7 @@ class TestCheckAvailabilitySetExists < Minitest::Test
   def test_check_availability_set_exists_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @availability_sets.stub :get, response do
-      assert_raises(RuntimeError) { @service.check_availability_set_exists('myrg1', 'myavset1') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.check_availability_set_exists('myrg1', 'myavset1') }
     end
   end
 end

data/test/requests/compute/test_check_managed_disk_exists.rb

@@ -32,7 +32,7 @@ class TestCheckManagedDiskExists < Minitest::Test
   def test_check_managed_disk_exists_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @managed_disks.stub :get, response do
-      assert_raises(RuntimeError) { @service.check_managed_disk_exists('myrg1', 'mydisk1') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.check_managed_disk_exists('myrg1', 'mydisk1') }
     end
   end
 end

data/test/requests/compute/test_check_vm_exists.rb

@@ -32,7 +32,7 @@ class TestCheckVirtualMachineExists < Minitest::Test
   def test_check_vm_exists_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @virtual_machines.stub :get, response do
-      assert_raises(RuntimeError) { @service.check_vm_exists('fog-test-rg', 'fog-test-server', false) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.check_vm_exists('fog-test-rg', 'fog-test-server', false) }
     end
   end
 end

data/test/requests/compute/test_check_vm_extension_exists.rb

@@ -32,7 +32,7 @@ class TestCheckVMExtensionExists < Minitest::Test
   def test_check_vm_extension_exists_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @vm_extension.stub :get, response do
-      assert_raises(RuntimeError) { @service.check_vm_extension_exists('fog-test-rg', 'fog-test-vm', 'fog-test-extension') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.check_vm_extension_exists('fog-test-rg', 'fog-test-vm', 'fog-test-extension') }
     end
   end
 end

data/test/requests/compute/test_create_availability_set.rb

@@ -71,7 +71,7 @@ class TestCreateAvailabilitySet < Minitest::Test
 
     @availability_sets.stub :validate_params, true do
       @availability_sets.stub :create_or_update, response do
-        assert_raises(RuntimeError) { @service.create_availability_set(avail_set_params) }
+        assert_raises(MsRestAzure::AzureOperationError) { @service.create_availability_set(avail_set_params) }
       end
     end
   end
@@ -105,7 +105,7 @@ class TestCreateAvailabilitySet < Minitest::Test
 
     @availability_sets.stub :validate_params, true do
       @availability_sets.stub :create_or_update, response do
-        assert_raises(RuntimeError) { @service.create_availability_set(avail_set_params) }
+        assert_raises(MsRestAzure::AzureOperationError) { @service.create_availability_set(avail_set_params) }
       end
     end
   end

data/test/requests/compute/test_create_image.rb

@@ -19,7 +19,7 @@ class TestCreateImage < Minitest::Test
   def test_create_image_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @image.stub :create_or_update, response do
-      assert_raises RuntimeError do
+      assert_raises MsRestAzure::AzureOperationError do
         @service.create_image(@input_params)
       end
     end

data/test/requests/compute/test_create_or_update_managed_disk.rb

@@ -31,7 +31,7 @@ class TestCreateManagedDisk < Minitest::Test
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @managed_disks.stub :validate_params, true do
       @managed_disks.stub :create_or_update, response do
-        assert_raises(RuntimeError) { @service.create_or_update_managed_disk(@disk) }
+        assert_raises(MsRestAzure::AzureOperationError) { @service.create_or_update_managed_disk(@disk) }
       end
     end
   end

data/test/requests/compute/test_create_virtual_machine.rb

@@ -96,7 +96,7 @@ class TestCreateVirtualMachine < Minitest::Test
   def test_create_virtual_machine_failure
     @virtual_machines.stub :create_or_update, @error_response do
       @virtual_machines.stub :get, @error_response do
-        assert_raises RuntimeError do
+        assert_raises MsRestAzure::AzureOperationError do
           @service.create_virtual_machine(@linux_virtual_machine_hash)
         end
       end
@@ -105,7 +105,7 @@ class TestCreateVirtualMachine < Minitest::Test
     # Async
     @virtual_machines.stub :create_or_update_async, @error_response do
       @virtual_machines.stub :get, @error_response do
-        assert_raises RuntimeError do
+        assert_raises MsRestAzure::AzureOperationError do
           @service.create_virtual_machine(@linux_virtual_machine_hash, true)
         end
       end

data/test/requests/compute/test_create_vm_extension.rb

@@ -27,7 +27,7 @@ class TestCreateVMExtension < Minitest::Test
   def test_create_vm_extension_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @vm_extension.stub :create_or_update, response do
-      assert_raises RuntimeError do
+      assert_raises MsRestAzure::AzureOperationError do
         @service.create_or_update_vm_extension(@vm_extension_params)
       end
     end

data/test/requests/compute/test_deallocate_virtual_machine.rb

@@ -22,11 +22,11 @@ class TestDeallocateVirtualMachine < Minitest::Test
   def test_deallocate_virtual_machine_failure
     response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :deallocate, response do
-      assert_raises(RuntimeError) { @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server', false) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server', false) }
     end
 
     @virtual_machines.stub :deallocate_async, response do
-      assert_raises(RuntimeError) { @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server', true) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server', true) }
     end
   end
 end

data/test/requests/compute/test_delete_availability_set.rb

@@ -17,7 +17,7 @@ class TestDeleteAvailabilitySet < Minitest::Test
   def test_delete_availability_set_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @availability_sets.stub :delete, response do
-      assert_raises(RuntimeError) { @service.delete_availability_set('fog-test-rg', 'fog-test-as') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.delete_availability_set('fog-test-rg', 'fog-test-as') }
     end
   end
 end

data/test/requests/compute/test_delete_image.rb

@@ -17,7 +17,7 @@ class TestDeleteImage < Minitest::Test
   def test_delete_image_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @image.stub :delete, response do
-      assert_raises(RuntimeError) { @service.delete_image('fog-test-rg', 'fog-test-server-osImage') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.delete_image('fog-test-rg', 'fog-test-server-osImage') }
     end
   end
 end

data/test/requests/compute/test_delete_managed_disk.rb

@@ -17,7 +17,7 @@ class TestDeleteManagedDisk < Minitest::Test
   def test_delete_managed_disk_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @managed_disks.stub :delete, response do
-      assert_raises(RuntimeError) { @service.delete_managed_disk('fog-test-rg', 'test-disk') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.delete_managed_disk('fog-test-rg', 'test-disk') }
     end
   end
 end

data/test/requests/compute/test_delete_virtual_machine.rb

@@ -22,11 +22,11 @@ class TestDeleteVirtualMachine < Minitest::Test
   def test_delete_virtual_machine_failure
     response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :delete, response do
-      assert_raises(RuntimeError) { @service.delete_virtual_machine('fog-test-rg', 'fog-test-server', false) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.delete_virtual_machine('fog-test-rg', 'fog-test-server', false) }
     end
 
     @virtual_machines.stub :delete_async, response do
-      assert_raises(RuntimeError) { @service.delete_virtual_machine('fog-test-rg', 'fog-test-server', true) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.delete_virtual_machine('fog-test-rg', 'fog-test-server', true) }
     end
   end
 end

data/test/requests/compute/test_detach_data_disk_from_vm.rb

@@ -23,7 +23,7 @@ class TestDetachDataDiskFromVM < Minitest::Test
     update_vm_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :get, get_vm_response do
       @virtual_machines.stub :create_or_update, update_vm_response do
-        assert_raises RuntimeError do
+        assert_raises MsRestAzure::AzureOperationError do
           @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1', false)
         end
       end

data/test/requests/compute/test_generalize_virtual_machine.rb

@@ -22,11 +22,11 @@ class TestGeneralizeVirtualMachine < Minitest::Test
   def test_generalize_virtual_machine_failure
     response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :generalize, response do
-      assert_raises(RuntimeError) { @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server', false) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server', false) }
     end
 
     @virtual_machines.stub :generalize_async, response do
-      assert_raises(RuntimeError) { @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server', true) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server', true) }
     end
   end
 end

data/test/requests/compute/test_get_availability_set.rb

@@ -18,7 +18,7 @@ class TestGetAvailabilitySet < Minitest::Test
   def test_list_availability_set_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @availability_sets.stub :get, response do
-      assert_raises(RuntimeError) { @service.get_availability_set('myrg1', 'myavset1') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.get_availability_set('myrg1', 'myavset1') }
     end
   end
 end

data/test/requests/compute/test_get_image.rb

@@ -17,7 +17,7 @@ class TestGetImage < Minitest::Test
   def test_get_image_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @image.stub :get, response do
-      assert_raises(RuntimeError) { @service.get_image('fog-test-rg', 'TestImage') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.get_image('fog-test-rg', 'TestImage') }
     end
   end
 end

data/test/requests/compute/test_get_managed_disk.rb

@@ -18,7 +18,7 @@ class TestGetManagedDisk < Minitest::Test
   def test_get_managed_disk_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @managed_disks.stub :get, response do
-      assert_raises(RuntimeError) { @service.get_managed_disk('myrg1', 'disk1') }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.get_managed_disk('myrg1', 'disk1') }
     end
   end
 end

data/test/requests/compute/test_get_virtual_machine.rb

@@ -18,7 +18,7 @@ class TestGetVirtualMachine < Minitest::Test
   def test_get_virtual_machine_failure
     response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :get, response do
-      assert_raises(RuntimeError) { @service.get_virtual_machine('fog-test-rg', 'fog-test-server', false) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.get_virtual_machine('fog-test-rg', 'fog-test-server', false) }
     end
   end
 end

data/test/requests/compute/test_get_vm_extension.rb

@@ -18,7 +18,7 @@ class TestGetVMExtension < Minitest::Test
   def test_update_vm_extension_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @vm_extension.stub :get, response do
-      assert_raises RuntimeError do
+      assert_raises MsRestAzure::AzureOperationError do
         @service.get_vm_extension('fog-test-rg', 'fog-test-vm', 'fog-test-extension')
       end
     end

data/test/requests/compute/test_get_vm_status.rb

@@ -19,7 +19,7 @@ class TestGetVirtualMachineStatus < Minitest::Test
   def test_vm_status_failure
     response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :get, response do
-      assert_raises(RuntimeError) { @service.check_vm_status('fog-test-rg', 'fog-test-server', false) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.check_vm_status('fog-test-rg', 'fog-test-server', false) }
     end
   end
 end

data/test/requests/compute/test_grant_access_to_managed_disk.rb

@@ -20,7 +20,7 @@ class TestGrantAccessToManagedDisk < Minitest::Test
   def test_grant_access_to_managed_disk_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @managed_disks.stub :grant_access, response do
-      assert_raises(RuntimeError) { @service.grant_access_to_managed_disk('myrg1', 'disk1', 'Read', 100) }
+      assert_raises(MsRestAzure::AzureOperationError) { @service.grant_access_to_managed_disk('myrg1', 'disk1', 'Read', 100) }
     end
   end
 end