far-gem 0.5.1 → 0.5.2

data/test/api_stub/models/compute/server.rb

@@ -21,6 +21,7 @@ module ApiStub
                 },
                 'osDisk' => {
                   'name' => 'fog-test-server_os_disk',
+                  'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/fog-test-server_os_disk',
                   'vhd' => {
                     'uri' => 'http://storageAccount.blob.core.windows.net/vhds/fog-test-server_os_disk.vhd'
                   }

data/test/api_stub/models/resources/resource.rb

@@ -37,6 +37,23 @@ module ApiStub
           result_mapper = Azure::ARM::Resources::Models::ResourceListResult.mapper
           client.deserialize(result_mapper, Fog::JSON.decode(resources), 'result.body').value
         end
+
+        def self.list_resources_in_resource_group_response(client)
+          resources = '{
+            "id": "/subscriptions/########-####-####-####-############/resourceGroups/{RESOURCE-GROUP}/providers/Microsoft.Network/{PROVIDER-NAME}/{RESOURCE-NAME}",
+            "name": "your-resource-name",
+                "type": "providernamespace/resourcetype",
+                "location": "westus",
+                "tags": {
+                "tag_name": "tag_value"
+            },
+                "plan": {
+                "name": "free"
+            }
+          }'
+          result_mapper = Azure::ARM::Resources::Models::GenericResource.mapper
+          client.deserialize(result_mapper, Fog::JSON.decode(resources), 'result.body')
+        end
       end
     end
   end

data/test/api_stub/requests/compute/virtual_machine.rb

@@ -243,6 +243,7 @@ module ApiStub
                 },
                 "osDisk": {
                   "name":"myosdisk1",
+                  "id":"/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1",
                   "vhd": {
                     "uri":"http://mystorage1.blob.core.windows.net/vhds/myosdisk1.vhd"
                   },
@@ -320,6 +321,7 @@ module ApiStub
                 },
                 "osDisk": {
                   "name":"myosdisk1",
+                  "id":"/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1",
                   "vhd": {
                     "uri":"http://mystorage1.blob.core.windows.net/vhds/myosdisk1.vhd"
                   },
@@ -397,6 +399,7 @@ module ApiStub
                 },
                 "osDisk": {
                   "name":"myosdisk1",
+                  "id":"/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1",
                   "vhd": {
                     "uri":"https://custimagestorage.blob.core.windows.net/customimage/trusty-server-cloudimg-amd64-disk1.vhd"
                   },
@@ -474,6 +477,7 @@ module ApiStub
                 },
                 "osDisk": {
                   "name":"myosdisk1",
+                  "id":"/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1",
                   "vhd": {
                     "uri":"http://mystorage1.blob.core.windows.net/vhds/myosdisk1.vhd"
                   },
@@ -546,6 +550,7 @@ module ApiStub
                     'osDisk' =>
                       {
                         'name' => 'myosdisk1',
+                        'id' => '/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1',
                         'vhd' =>
                           {
                             'uri' => 'http://mystorage1.blob.core.windows.net/vhds/myosdisk1.vhd'
@@ -651,6 +656,7 @@ module ApiStub
                     },
                     "osDisk": {
                       "name":"myosdisk1",
+                      "id":"/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1",
                       "vhd": {
                         "uri":"http://mystorage1.blob.core.windows.net/vhds/myosdisk1.vhd"
                       },
@@ -809,6 +815,7 @@ module ApiStub
                     },
                     "osDisk": {
                       "name":"myosdisk1",
+                      "id":"/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1",
                       "vhd": {
                         "uri":"http://mystorage1.blob.core.windows.net/vhds/myosdisk1.vhd"
                       },
@@ -896,6 +903,7 @@ module ApiStub
                     },
                     "osDisk": {
                       "name":"myosdisk1",
+                      "id":"/subscriptions/{subscription-id}/resourceGroups/fog-test-rg/providers/Microsoft.Compute/disks/myosdisk1",
                       "vhd": {
                         "uri":"http://mystorage1.blob.core.windows.net/vhds/myosdisk1.vhd"
                       },

data/test/api_stub/requests/network/network_security_group.rb

@@ -216,8 +216,8 @@ module ApiStub
              }
             ]
           }'
-          result_mapper = Azure::ARM::Network::Models::NetworkInterfaceListResult.mapper
-          network_client.deserialize(result_mapper, Fog::JSON.decode(nsg_list), 'result.body')
+          result_mapper = Azure::ARM::Network::Models::NetworkSecurityGroupListResult.mapper
+          network_client.deserialize(result_mapper, Fog::JSON.decode(nsg_list), 'result.body').value
         end
       end
     end

data/test/api_stub/requests/resources/resource.rb

@@ -39,6 +39,23 @@ module ApiStub
           result_mapper = Azure::ARM::Resources::Models::ResourceListResult.mapper
           client.deserialize(result_mapper, Fog::JSON.decode(body), 'result.body')
         end
+
+        def self.list_resources_in_resource_group(client)
+          body = '{
+            "id": "/subscriptions/########-####-####-####-############/resourceGroups/{RESOURCE-GROUP}/providers/Microsoft.Network/{PROVIDER-NAME}/{RESOURCE-NAME}",
+            "name": "your-resource-name",
+            "type": "providernamespace/resourcetype",
+            "location": "westus",
+            "tags": {
+            "tag_name": "tag_value"
+            },
+            "plan": {
+            "name": "free"
+            }
+          }'
+          result_mapper = Azure::ARM::Resources::Models::GenericResource.mapper
+          client.deserialize(result_mapper, Fog::JSON.decode(body), 'result.body')
+        end
       end
     end
   end

data/test/api_stub/requests/storage/file.rb

@@ -5,6 +5,15 @@ module ApiStub
       # Below data should be as same as those in Mock classes in lib/fog/azurerm/requests/storage/*.rb
       class File
         def self.blob
+          blob_data = blob_as_hash
+          blob = Azure::Storage::Blob::Blob.new
+          blob.name = blob_data['name']
+          blob.metadata = blob_data['metadata']
+          blob.properties = blob_data['properties'].map { |k, v| { k.to_sym => v } }.reduce({}, &:merge!)
+          blob
+        end
+
+        def self.blob_as_hash
           {
             'name' => 'test_blob',
             'metadata' => {},

data/test/integration/network_interface.rb

@@ -107,7 +107,8 @@ begin
     public_ip_address_id: "/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-NI/providers/Microsoft.Network/publicIPAddresses/mypubip",
     ip_configuration_name: 'testIpConfiguration',
     private_ip_allocation_method: 'Dynamic',
-    tags: { key: 'value' }
+    tags: { key: 'value' },
+    enable_accelerated_networking: true
   )
   puts "Created network interface: #{network_interface.name}"
 

data/test/integration/resource_group.rb

@@ -17,7 +17,6 @@ resource = Fog::Resources::AzureRM.new(
 resource_group_name = "RG-#{current_time}"
 
 begin
-
   ########################################################################################################################
   ######################                   Check Resource Group Exists?                             ######################
   ########################################################################################################################
@@ -51,6 +50,16 @@ begin
   resource_group = resource_groups.get(resource_group_name)
   puts "Get resource group: #{resource_group.name}"
 
+  ########################################################################################################################
+  ######################             Get All Resources in a Resource Group                          ######################
+  ########################################################################################################################
+
+  resources = resource.azure_resources.list_resources_in_resource_group(resource_group_name)
+  puts 'List resources in resource groups:'
+  resources.each do |a_resource|
+    puts a_resource.name
+  end
+
   ########################################################################################################################
   ######################                           Destroy Resource Group                           ######################
   ########################################################################################################################

data/test/models/compute/test_managed_disk.rb

@@ -50,13 +50,20 @@ class TestManagedDisk < Minitest::Test
 
   def test_destroy_method_true_response
     @service.stub :delete_managed_disk, true do
-      assert @managed_disk.destroy
+      assert @managed_disk.destroy(false)
     end
   end
 
   def test_destroy_method_false_response
     @service.stub :delete_managed_disk, false do
-      assert !@managed_disk.destroy
+      assert !@managed_disk.destroy(false)
+    end
+  end
+
+  def test_destroy_method_can_take_params_async
+    async_response = Concurrent::Promise.execute { 10 }
+    @service.stub :delete_managed_disk, async_response do
+      assert_instance_of Fog::AzureRM::AsyncResponse, @managed_disk.destroy(true)
     end
   end
 end

data/test/models/compute/test_server.rb

@@ -38,6 +38,7 @@ class TestServer < Minitest::Test
       :vm_size,
       :storage_account_name,
       :os_disk_name,
+      :os_disk_id,
       :os_disk_vhd_uri,
       :os_disk_caching,
       :publisher,

data/test/models/network/test_network_interface.rb

@@ -45,7 +45,8 @@ class TestNetworkInterface < Minitest::Test
       :applied_dns_servers,
       :internal_dns_name_label,
       :internal_fqd,
-      :tags
+      :tags,
+      :enable_accelerated_networking
     ]
     attributes.each do |attribute|
       assert_respond_to @network_interface, attribute

data/test/models/resources/test_resources.rb

@@ -14,7 +14,8 @@ class TestResources < Minitest::Test
     methods = [
       :all,
       :get,
-      :check_azure_resource_exists
+      :check_azure_resource_exists,
+      :list_resources_in_resource_group
     ]
     methods.each do |method|
       assert_respond_to @resources, method
@@ -49,4 +50,16 @@ class TestResources < Minitest::Test
       assert !@resources.check_azure_resource_exists(@resource_id, '2016-09-01')
     end
   end
+
+  def test_list_resources_in_resource_group_method_response
+    client = @service.instance_variable_get(:@rmc)
+    response = [ApiStub::Models::Resources::Resource.list_resources_in_resource_group_response(client)]
+    @service.stub :list_resources_in_resource_group, response do
+      assert_instance_of Fog::Resources::AzureRM::AzureResources, @resources.list_resources_in_resource_group('fog-test-rg')
+      assert @resources.list_resources_in_resource_group('fog-test-rg').size >= 1
+      @resources.list_resources_in_resource_group('fog-test-rg').each do |s|
+        assert_instance_of Fog::Resources::AzureRM::AzureResource, s
+      end
+    end
+  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

@@ -10,80 +10,104 @@ class TestCreateVirtualMachine < Minitest::Test
     @custom_data_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_with_custom_data_response(compute_client)
     @linux_virtual_machine_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_params
     @windows_virtual_machine_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_params
+    @error_response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
   end
 
   def test_create_linux_virtual_machine_success
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash, true), @response
+      end
     end
   end
 
   def test_create_windows_virtual_machine_success
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash, true), @response
+      end
     end
   end
 
   def test_create_linux_virtual_machine_from_custom_image_success
     linux_virtual_machine_with_custom_image_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_with_custom_image_params
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash, true), @response
+      end
     end
   end
 
   def test_create_windows_virtual_machine_from_custom_image_success
     windows_virtual_machine_with_custom_image_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_with_custom_image_params
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash, true), @response
+      end
     end
   end
 
   def test_create_linux_virtual_machine_with_custom_data_success
     linux_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_with_custom_data_params
     @virtual_machines.stub :create_or_update, @custom_data_response do
-      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_data_hash), @custom_data_response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_data_hash), @custom_data_response
+      end
     end
   end
 
   def test_create_windows_virtual_machine_with_custom_data_success
     windows_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_with_custom_data_params
     @virtual_machines.stub :create_or_update, @custom_data_response do
-      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_data_hash), @custom_data_response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_data_hash), @custom_data_response
+      end
     end
   end
 
   def test_create_virtual_machine_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @virtual_machines.stub :create_or_update, response do
-      assert_raises RuntimeError do
-        @service.create_virtual_machine(@linux_virtual_machine_hash)
+    @virtual_machines.stub :create_or_update, @error_response do
+      @virtual_machines.stub :get, @error_response do
+        assert_raises MsRestAzure::AzureOperationError do
+          @service.create_virtual_machine(@linux_virtual_machine_hash)
+        end
       end
     end
 
     # Async
-    @virtual_machines.stub :create_or_update_async, response do
-      assert_raises RuntimeError do
-        @service.create_virtual_machine(@linux_virtual_machine_hash, true)
+    @virtual_machines.stub :create_or_update_async, @error_response do
+      @virtual_machines.stub :get, @error_response do
+        assert_raises MsRestAzure::AzureOperationError do
+          @service.create_virtual_machine(@linux_virtual_machine_hash, true)
+        end
       end
     end
   end