fog-azure-rm 0.3.2 → 0.3.3

data/test/integration/deployment.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resources = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/express_route_circuit.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resources = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/external_load_balancer.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/internal_load_balancer.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/key_vault.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resources = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/local_network_gateway.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resource = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/managed_disk.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/network_interface.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/network_security_group.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/network_security_rule.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/public_ip.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/record_set.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resource = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/resource_group.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resource = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/resource_tag.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resources = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/server.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/server_custom_image.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],
@@ -92,7 +92,7 @@ begin
     username: 'testuser',
     password: 'Confiz=123',
     disable_password_authentication: false,
-    network_interface_card_id: "/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-CustomVM/providers/Microsoft.Network/networkInterfaces/NetInt",
+    network_interface_card_ids: ["/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-CustomVM/providers/Microsoft.Network/networkInterfaces/NetInt"],
     platform: 'linux',
     vhd_path: 'https://custimagestorage.blob.core.windows.net/newcustomvhd/trusty-server-cloudimg-amd64-disk1-zeeshan.vhd'
   )

data/test/integration/sql_server.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resources = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/storage_account.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/subnet.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/traffic_manager.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resources = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/virtual_machine_extension.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],
@@ -87,7 +87,7 @@ begin
     username: 'testuser',
     password: 'Confiz=123',
     disable_password_authentication: false,
-    network_interface_card_id: "/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-VME/providers/Microsoft.Network/networkInterfaces/NetInt",
+    network_interface_card_ids: ["/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-VME/providers/Microsoft.Network/networkInterfaces/NetInt"],
     publisher: 'MicrosoftWindowsServer',
     offer: 'WindowsServer',
     sku: '2008-R2-SP1',

data/test/integration/virtual_network.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/integration/virtual_network_gateway.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resource = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],
@@ -81,8 +81,8 @@ begin
       }
     ],
     resource_group: 'TestRG-VNG',
-    sku_name: 'Basic',
-    sku_tier: 'Basic',
+    sku_name: 'Standard',
+    sku_tier: 'Standard',
     sku_capacity: 2,
     gateway_type: 'ExpressRoute',
     enable_bgp: false,
@@ -111,12 +111,6 @@ begin
 
   puts "Deleted virtual network gateway: #{network_gateway.destroy}"
 
-  pubip = network.public_ips.get('TestRG-VNG', 'mypubip')
-  pubip.destroy
-
-  vnet = network.virtual_networks.get('TestRG-VNG', 'testVnet')
-  vnet.destroy
-
   rg = resource.resource_groups.get('TestRG-VNG')
   rg.destroy
 rescue

data/test/integration/{Virtual_network_gateway_connection.rb → virtual_network_gateway_connection.rb}

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resource = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],
@@ -214,7 +214,8 @@ begin
 
   resource_group = resource.resource_groups.get('TestRG-GC')
   resource_group.destroy
-rescue
+rescue Exception => e
+  raise e
   puts 'Integration Test for virtual network gateway connection is failing'
   resource_group.destroy unless resource_group.nil?
 end

data/test/integration/virtual_network_gateway_connection_to_express_route.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 resource = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],
@@ -70,8 +70,8 @@ begin
       }
     ],
     resource_group: 'TestRG-GCE',
-    sku_name: 'Basic',
-    sku_tier: 'Basic',
+    sku_name: 'Standard',
+    sku_tier: 'Standard',
     sku_capacity: 2,
     gateway_type: 'ExpressRoute',
     enable_bgp: false,

data/test/integration/zone.rb

@@ -6,7 +6,7 @@ require 'yaml'
 ######################                              Keep it Uncommented!                          ######################
 ########################################################################################################################
 
-azure_credentials = YAML.load_file('credentials/azure.yml')
+azure_credentials = YAML.load_file(File.expand_path('credentials/azure.yml', __dir__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],

data/test/models/compute/test_availability_set.rb

@@ -6,7 +6,7 @@ class TestAvailabilitySet < Minitest::Test
     @service = Fog::Compute::AzureRM.new(credentials)
     @availability_set = availability_set(@service)
     compute_client = @service.instance_variable_get(:@compute_mgmt_client)
-    @response = ApiStub::Models::Compute::AvailabilitySet.create_availability_set_response(compute_client)
+    @response = ApiStub::Models::Compute::AvailabilitySet.create_unmanaged_availability_set_response(compute_client)
   end
 
   def test_model_methods
@@ -27,7 +27,9 @@ class TestAvailabilitySet < Minitest::Test
       :location,
       :resource_group,
       :platform_update_domain_count,
-      :platform_fault_domain_count
+      :platform_fault_domain_count,
+      :use_managed_disk,
+      :sku_name
     ]
     attributes.each do |attribute|
       assert_respond_to @availability_set, attribute

data/test/models/key_vault/test_vault.rb

@@ -4,7 +4,7 @@ require File.expand_path '../../test_helper', __dir__
 class TestVault < Minitest::Test
   def setup
     @service = Fog::KeyVault::AzureRM.new(credentials)
-    @vault = Fog::KeyVault::AzureRM::Vault.new
+    @vault = key_vault(@service)
   end
 
   def test_model_attributes

data/test/requests/compute/test_attach_data_disk_to_vm.rb

@@ -20,7 +20,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_equal @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1'), @update_vm_response
+              assert_equal @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false), @update_vm_response
               blob_service.verify
             end
           end
@@ -35,7 +35,7 @@ class TestAttachDataDiskToVM < Minitest::Test
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, @update_vm_response do
           assert_raises RuntimeError do
-            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
           end
         end
       end
@@ -48,7 +48,7 @@ class TestAttachDataDiskToVM < Minitest::Test
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, update_vm_response do
           assert_raises RuntimeError do
-            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
           end
         end
       end
@@ -61,7 +61,7 @@ class TestAttachDataDiskToVM < Minitest::Test
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, update_vm_response do
           assert_raises RuntimeError do
-            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
           end
         end
       end
@@ -72,7 +72,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     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
-        @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+        @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
       end
     end
   end
@@ -82,7 +82,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     @virtual_machines.stub :get, @get_vm_response do
       @storage_accounts.stub :list_keys, storage_access_keys_response do
         assert_raises RuntimeError do
-          @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+          @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
         end
       end
     end

data/test/requests/compute/test_check_vm_exists.rb

@@ -11,21 +11,21 @@ class TestCheckVirtualMachineExists < Minitest::Test
   def test_check_vm_exists_success
     response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(@compute_client)
     @virtual_machines.stub :get, response do
-      assert @service.check_vm_exists('fog-test-rg', 'fog-test-server')
+      assert @service.check_vm_exists('fog-test-rg', 'fog-test-server', false)
     end
   end
 
   def test_check_vm_exists_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
     @virtual_machines.stub :get, response do
-      assert !@service.check_vm_exists('fog-test-rg', 'fog-test-server')
+      assert !@service.check_vm_exists('fog-test-rg', 'fog-test-server', false)
     end
   end
 
   def test_check_vm_exists_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @virtual_machines.stub :get, response do
-      assert_raises(RuntimeError) { @service.check_vm_exists('fog-test-rg', 'fog-test-server') }
+      assert_raises(RuntimeError) { @service.check_vm_exists('fog-test-rg', 'fog-test-server', false) }
     end
   end
 end

data/test/requests/compute/test_create_availability_set.rb

@@ -10,18 +10,102 @@ class TestCreateAvailabilitySet < Minitest::Test
 
   def test_create_availability_set_success
     mocked_response = ApiStub::Requests::Compute::AvailabilitySet.create_availability_set_response(@client)
+    avail_set_params =
+      {
+        name: 'myavset1',
+        location: 'west us',
+        resource_group: 'fog-test-rg'
+      }
+
+    @availability_sets.stub :validate_params, true do
+      @availability_sets.stub :create_or_update, mocked_response do
+        assert_equal @service.create_availability_set(avail_set_params), mocked_response
+      end
+    end
+  end
+
+  def test_create_custom_availability_set_success
+    mocked_response = ApiStub::Requests::Compute::AvailabilitySet.create_custom_availability_set_response(@client)
+    avail_set_params =
+      {
+        name: 'myavset1',
+        location: 'west us',
+        resource_group: 'fog-test-rg',
+        platform_fault_domain_count: 3,
+        platform_update_domain_count: 10
+      }
+
     @availability_sets.stub :validate_params, true do
       @availability_sets.stub :create_or_update, mocked_response do
-        assert_equal @service.create_availability_set('fog-test-rg', 'myavset1', 'west us'), mocked_response
+        assert_equal @service.create_availability_set(avail_set_params), mocked_response
       end
     end
   end
 
-  def test_create_availability_set_failure
+  def test_create_unmanaged_availability_set_success
+    mocked_response = ApiStub::Requests::Compute::AvailabilitySet.create_unmanaged_availability_set_response(@client)
+    avail_set_params =
+      {
+        name: 'myavset1',
+        location: 'west us',
+        resource_group: 'fog-test-rg',
+        use_managed_disk: false
+      }
+
+    @availability_sets.stub :validate_params, true do
+      @availability_sets.stub :create_or_update, mocked_response do
+        assert_equal @service.create_availability_set(avail_set_params), mocked_response
+      end
+    end
+  end
+
+  def test_create_unmanaged_availability_set_failure
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    avail_set_params =
+      {
+        name: 'myavset1',
+        location: 'west us',
+        resource_group: 'fog-test-rg',
+        use_managed_disk: false
+      }
+
+    @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) }
+      end
+    end
+  end
+
+  def test_create_managed_availability_set_success
+    mocked_response = ApiStub::Requests::Compute::AvailabilitySet.create_managed_availability_set_response(@client)
+    avail_set_params =
+      {
+        name: 'myavset1',
+        location: 'west us',
+        resource_group: 'fog-test-rg',
+        use_managed_disk: true
+      }
+
+    @availability_sets.stub :validate_params, true do
+      @availability_sets.stub :create_or_update, mocked_response do
+        assert_equal @service.create_availability_set(avail_set_params), mocked_response
+      end
+    end
+  end
+
+  def test_create_managed_availability_set_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    avail_set_params =
+      {
+        name: 'myavset1',
+        location: 'west us',
+        resource_group: 'fog-test-rg',
+        use_managed_disk: true
+      }
+
     @availability_sets.stub :validate_params, true do
       @availability_sets.stub :create_or_update, response do
-        assert_raises(RuntimeError) { @service.create_availability_set('fog-test-rg', 'fog-test-as', 'west us') }
+        assert_raises(RuntimeError) { @service.create_availability_set(avail_set_params) }
       end
     end
   end

data/test/requests/compute/test_detach_data_disk_from_vm.rb

@@ -13,7 +13,7 @@ class TestDetachDataDiskFromVM < Minitest::Test
     update_vm_response = ApiStub::Requests::Compute::VirtualMachine.detach_data_disk_from_vm_response(@compute_client)
     @virtual_machines.stub :get, get_vm_response do
       @virtual_machines.stub :create_or_update, update_vm_response do
-        assert_equal @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1'), update_vm_response
+        assert_equal @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1', false), update_vm_response
       end
     end
   end
@@ -24,7 +24,7 @@ class TestDetachDataDiskFromVM < Minitest::Test
     @virtual_machines.stub :get, get_vm_response do
       @virtual_machines.stub :create_or_update, update_vm_response do
         assert_raises RuntimeError do
-          @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1')
+          @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1', false)
         end
       end
     end

data/test/requests/compute/test_get_virtual_machine.rb

@@ -11,14 +11,14 @@ class TestGetVirtualMachine < Minitest::Test
   def test_get_virtual_machine_success
     response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(@compute_client)
     @virtual_machines.stub :get, response do
-      assert_equal @service.get_virtual_machine('fog-test-rg', 'fog-test-server'), response
+      assert_equal @service.get_virtual_machine('fog-test-rg', 'fog-test-server', false), response
     end
   end
 
   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') }
+      assert_raises(RuntimeError) { @service.get_virtual_machine('fog-test-rg', 'fog-test-server', false) }
     end
   end
 end

data/test/requests/compute/test_get_vm_status.rb

@@ -12,14 +12,14 @@ class TestGetVirtualMachineStatus < Minitest::Test
     response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_instance_view_response(@compute_client)
     compare_result = ApiStub::Requests::Compute::VirtualMachine.vm_status_response
     @virtual_machines.stub :get, response do
-      assert_equal @service.check_vm_status('fog-test-rg', 'fog-test-server'), compare_result
+      assert_equal @service.check_vm_status('fog-test-rg', 'fog-test-server', false), compare_result
     end
   end
 
   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') }
+      assert_raises(RuntimeError) { @service.check_vm_status('fog-test-rg', 'fog-test-server', false) }
     end
   end
 end

data/test/test_helper.rb

@@ -116,6 +116,8 @@ def availability_set(service)
     name: 'availability-set',
     location: 'West US',
     resource_group: 'fog-test-rg',
+    use_managed_disk: false,
+    sku_name: 'Classic',
     service: service
   )
 end
@@ -815,3 +817,24 @@ def managed_disk(service)
     service: service
   )
 end
+
+def key_vault(service)
+  Fog::KeyVault::AzureRM::Vault.new(
+    name: 'key-vault',
+    location: 'East US',
+    resource_group_name: 'fog-test-rg',
+    sku_family: 'A',
+    sku_name: 'Standard',
+    access_policies: [
+      {
+        tenant_id: 'XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX',
+        object_id: 'XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX',
+        permissions: {
+          keys: ['all'],
+          secrets: ['all']
+        }
+      }
+    ],
+    service: service
+  )
+end