fog-azure-rm-temp 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4b1b3d3e92b86d79df668e8ed1c864edd9b35263
-  data.tar.gz: e89d10ef333b8d8be25d6af5f876e422b93725f2
+  metadata.gz: a20d559af2750fa5accf144e58bfaed3066f527b
+  data.tar.gz: 697b8f3ccf1afe6692add47b3c95ea1fa48cf6de
 SHA512:
-  metadata.gz: 58f52397024c1fa8d4e4ccfde73b167405bcf2ef6349a50c3c4a2a19c6bf458828a1739e92506db2a066e14ac3a4815eaffd78cffeec877d2ea824e57a84c490
-  data.tar.gz: d39cba508579ae02029186ebf2a28a01bcfb182c0363325be614127131675973dd160208b982fb125baf5cfd710fbbca471218f62faec4f4083c7971cbc2618d
+  metadata.gz: 563047f630231789c7178dd53c74a71959db9d6ecfb919b1fe0d4291caa8af53f03af922364071ba1543135c88a146542002b19e26d2b565129a625d446f1b59
+  data.tar.gz: e7feaca9e8386910c7025a1abb25000acebbf5e4df79cd97a1d7548e9556e062fc88fe43f104d700407cf0210595cf4ebe0f4b296ebc209f7f3bf7de5fcf3004

data/CHANGELOG.md

@@ -1,3 +1,17 @@
+## 0.3.3
+
+**Added:**
+- Compute Service - Added support to create Managed Availability Set
+
+**Changed:**
+- Compute Service - Updated default values for fault and update domains
+- Compute Service - Provided option to configure fault and update domains with reasonable default values
+
+## 0.3.2
+
+**Added:**
+- Compute Service - Made remaining server(virtual machine) methods async
+
 ## 0.3.1
 
 **Changed:**

data/fog-azure-rm.gemspec

@@ -5,7 +5,7 @@ require 'fog/azurerm/version'
 Gem::Specification.new do |spec|
   spec.name          = 'fog-azure-rm-temp'
   spec.version       = Fog::AzureRM::VERSION
-  spec.authors       = ['Shaffan Chaudhry', 'Samawia Moin', 'Adnan Khalil', 'Zeeshan Arshad', 'Haider Ali', 'Waqar Haider', 'Bilal Naeem']
+  spec.authors       = ['Shaffan Chaudhry', 'Samawia Moin', 'Adnan Khalil', 'Zeeshan Arshad', 'Haider Ali', 'Waqar Haider', 'Bilal Naeem', 'Muhammad Asad', 'Azeem Sajid', 'Maham Nazir', 'Abbas Sheikh']
   spec.summary       = "Module for the 'fog' gem to support Azure Resource Manager cloud services."
   spec.description   = "This library can be used as a module for 'fog' or as standalone provider
                         to use the Azure Resource Manager cloud services in applications.."
@@ -32,6 +32,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'azure_mgmt_traffic_manager', '~> 0.9.0'
   spec.add_dependency 'azure_mgmt_sql', '~> 0.9.0'
   spec.add_dependency 'azure_mgmt_key_vault', '~> 0.9.0'
-  spec.add_dependency 'azure-storage', '>= 0.11.5.preview', '< 1.0'
+  spec.add_dependency 'azure-storage', '= 0.11.5.preview'
   spec.add_dependency 'vhd', '0.0.4'
 end

data/lib/fog/azurerm/constants.rb

@@ -7,7 +7,7 @@ STANDARD_STORAGE = 'Standard'.freeze
 PREMIUM_STORAGE = 'Premium'.freeze
 ALLOWED_STANDARD_REPLICATION = %w(LRS ZRS GRS RAGRS).freeze
 API_VERSION = '2016-06-01'.freeze
-FAULT_DOMAIN_COUNT = 3
+FAULT_DOMAIN_COUNT = 2
 UPDATE_DOMAIN_COUNT = 5
 WINDOWS = 'windows'.freeze
 VPN = 'Vpn'.freeze
@@ -59,3 +59,5 @@ MAXIMUM_CHUNK_SIZE = 4 * 1024 * 1024
 # The hash value of 4MB empty content
 HASH_OF_4MB_EMPTY_CONTENT = 'b5cfa9d6c8febd618f91ac2843d50a1c'.freeze
 REST_CLIENT_API_VERSION = %w(2014-04-01-preview 2016-05-01).freeze
+AS_SKU_CLASSIC = 'Classic'.freeze
+AS_SKU_ALIGNED = 'Aligned'.freeze

data/lib/fog/azurerm/docs/compute.md

@@ -52,6 +52,7 @@ When using **managed_disk_storage_type** you should not pass the **vhd_path** as
         password: '<Password for VM>',             # Optional, if 'platform' partameter is 'Linux'.
         disable_password_authentication: false,
         network_interface_card_ids: ['/subscriptions/{Subscription-Id}/resourceGroups/{Resource-Group-Name}/providers/Microsoft.Network/networkInterfaces/{Network-Interface-Id}'],
+        availability_set_id: '<availability_set_id>', # Optional
         publisher: 'Canonical',                    # Not required if custom image is being used 
         offer: 'UbuntuServer',                     # Not required if custom image is being used
         sku: '14.04.2-LTS',                        # Not required if custom image is being used
@@ -77,6 +78,7 @@ Create a new windows server
         password: '<Password for VM>',
         disable_password_authentication: false,
         network_interface_card_ids: ['/subscriptions/{Subscription-Id}/resourceGroups/{Resource-Group-Name}/providers/Microsoft.Network/networkInterfaces/{Network-Interface-Id}'],
+        availability_set_id: '<availability_set_id>', # Optional
         publisher: 'MicrosoftWindowsServerEssentials',   # Not required if custom image is being used
         offer: 'WindowsServerEssentials',                # Not required if custom image is being used  
         sku: 'WindowsServerEssentials',                  # Not required if custom image is being used
@@ -103,6 +105,7 @@ Create a new linux server asynchronously
         password: '<Password for VM>',             # Optional, if 'platform' partameter is 'Linux'.
         disable_password_authentication: false,
         network_interface_card_ids: ['/subscriptions/{Subscription-Id}/resourceGroups/{Resource-Group-Name}/providers/Microsoft.Network/networkInterfaces/{Network-Interface-Id}'],
+        availability_set_id: '<availability_set_id>', # Optional
         publisher: 'Canonical',                    # Not required if custom image is being used 
         offer: 'UbuntuServer',                     # Not required if custom image is being used
         sku: '14.04.2-LTS',                        # Not required if custom image is being used
@@ -216,10 +219,13 @@ azure_compute_service.availability_sets.check_availability_set_exists(<Resource
 Create a new availability set
 
 ```ruby
-    azure_compute_service.availability_sets.create(
-         name: '<Availability Set name>',
-         location: 'West US',
-         resource_group: '<Resource Group name>'
+azure_compute_service.availability_sets.create(
+    name: '<Availability Set name>',
+    location: '<Location>',
+    resource_group: '<Resource Group name>'
+    platform_fault_domain_count: <No of Fault Domains>,     # [Optional] Default => { Managed: 2, Unmanaged: 3 }
+    platform_update_domain_count: <No of Update Domains>,   # [Optional] Default => { Managed: 5, Unmanaged: 5 }
+    is_managed: true                                        # [Optional] Possible values true or false
 )
 ```
 ## List Availability Sets 

data/lib/fog/azurerm/models/compute/availability_set.rb

@@ -11,6 +11,8 @@ module Fog
         attribute :resource_group
         attribute :platform_update_domain_count
         attribute :platform_fault_domain_count
+        attribute :is_managed
+        attribute :sku_name
 
         def self.parse(availability_set)
           hash = {}
@@ -21,6 +23,12 @@ module Fog
           hash['resource_group'] = get_resource_group_from_id(availability_set.id)
           hash['platform_update_domain_count'] = availability_set.platform_update_domain_count
           hash['platform_fault_domain_count'] = availability_set.platform_fault_domain_count
+
+          unless availability_set.sku.nil?
+            hash['sku_name'] = availability_set.sku.name
+            hash['is_managed'] = availability_set.sku.name.eql?(AS_SKU_ALIGNED)
+          end
+
           hash
         end
 
@@ -29,7 +37,7 @@ module Fog
           requires :location
           requires :resource_group
           # need to create the availability set
-          as = service.create_availability_set(resource_group, name, location)
+          as = service.create_availability_set(avail_set_params(platform_fault_domain_count, platform_update_domain_count, is_managed))
           hash = Fog::Compute::AzureRM::AvailabilitySet.parse(as)
           merge_attributes(hash)
         end
@@ -37,6 +45,19 @@ module Fog
         def destroy
           service.delete_availability_set(resource_group, name)
         end
+
+        private
+
+        def avail_set_params(platform_fault_domain_count, platform_update_domain_count, is_managed)
+          {
+            name: name,
+            location: location,
+            resource_group: resource_group,
+            platform_fault_domain_count: platform_fault_domain_count,
+            platform_update_domain_count: platform_update_domain_count,
+            is_managed: is_managed
+          }
+        end
       end
     end
   end

data/lib/fog/azurerm/models/key_vault/access_policy_entry.rb

@@ -21,10 +21,6 @@ module Fog
 
           access_policy_entry_hash
         end
-
-        def save
-          requires :object_id, :tenant_id, :keys, :secrets
-        end
       end
     end
   end

data/lib/fog/azurerm/models/network/network_interface.rb

@@ -22,7 +22,6 @@ module Fog
         attribute :applied_dns_servers
         attribute :internal_dns_name_label
         attribute :internal_fqd
-        attribute :type
 
         def self.parse(nic)
           hash = {}
@@ -34,7 +33,6 @@ module Fog
           hash['mac_address'] = nic.mac_address unless nic.mac_address.nil?
           hash['network_security_group_id'] = nil
           hash['network_security_group_id'] = nic.network_security_group.id unless nic.network_security_group.nil?
-          hash['type'] = nic.type
           ip_configuration = nic.ip_configurations[0] unless nic.ip_configurations.nil?
           unless ip_configuration.nil?
             hash['ip_configuration_name'] = ip_configuration.name

data/lib/fog/azurerm/models/traffic_manager/traffic_manager_profile.rb

@@ -55,7 +55,7 @@ module Fog
 
         def update(profile_params)
           validate_input(profile_params)
-          profile_params[:endpoints] = endpoints.map { |endpoint| get_hash_from_object(endpoint)['attributes'] }
+          profile_params[:endpoints] = endpoints.map { |endpoint| get_hash_from_object(endpoint)['attributes'] } unless endpoints.nil?
           merge_attributes(profile_params)
           profile = service.create_or_update_traffic_manager_profile(traffic_manager_profile_hash)
           merge_attributes(Fog::TrafficManager::AzureRM::TrafficManagerProfile.parse(profile))

data/lib/fog/azurerm/requests/compute/create_availability_set.rb

@@ -3,10 +3,18 @@ module Fog
     class AzureRM
       # This class provides the actual implementation for service calls.
       class Real
-        def create_availability_set(resource_group, name, location)
+        def create_availability_set(availability_set_params)
+          name = availability_set_params[:name]
+          location = availability_set_params[:location]
+          resource_group = availability_set_params[:resource_group]
+          fault_domain_count = availability_set_params[:platform_fault_domain_count]
+          update_domain_count = availability_set_params[:platform_update_domain_count]
+          is_managed = availability_set_params[:is_managed].nil? ? false : availability_set_params[:is_managed]
+
           msg = "Creating Availability Set '#{name}' in #{location} region."
           Fog::Logger.debug msg
-          avail_set_params = get_availability_set_properties(location)
+          avail_set_params = get_availability_set_properties(location, fault_domain_count, update_domain_count, is_managed)
+
           begin
             availability_set = @compute_mgmt_client.availability_sets.create_or_update(resource_group, name, avail_set_params)
           rescue MsRestAzure::AzureOperationError => e
@@ -17,23 +25,29 @@ module Fog
         end
 
         # create the properties object for creating availability sets
-        def get_availability_set_properties(location)
+        def get_availability_set_properties(location, fault_domain_count, update_domain_count, is_managed)
           avail_set = Azure::ARM::Compute::Models::AvailabilitySet.new
-          avail_set.platform_fault_domain_count = FAULT_DOMAIN_COUNT
-          avail_set.platform_update_domain_count = UPDATE_DOMAIN_COUNT
-          avail_set.virtual_machines = []
-          avail_set.statuses = []
           avail_set.location = location
+          avail_set.sku = create_availability_set_sku(is_managed)
+          avail_set.platform_fault_domain_count = fault_domain_count.nil? ? FAULT_DOMAIN_COUNT : fault_domain_count
+          avail_set.platform_update_domain_count = update_domain_count.nil? ? UPDATE_DOMAIN_COUNT : update_domain_count
           avail_set
         end
+
+        def create_availability_set_sku(is_managed)
+          sku = Azure::ARM::Compute::Models::Sku.new
+          sku.name = is_managed ? AS_SKU_ALIGNED : AS_SKU_CLASSIC
+          sku
+        end
       end
+
       # This class provides the mock implementation for unit tests.
       class Mock
-        def create_availability_set(resource_group, name, location)
+        def create_availability_set(availability_set_params)
           {
-            'location' => location,
-            'id' => "/subscriptions/########-####-####-####-############/resourceGroups/#{resource_group}/providers/Microsoft.Compute/availabilitySets/#{name}",
-            'name' => name,
+            'location' => availability_set_params[:location],
+            'id' => "/subscriptions/########-####-####-####-############/resourceGroups/#{availability_set_params[:resource_group]}/providers/Microsoft.Compute/availabilitySets/#{availability_set_params[:name]}",
+            'name' => availability_set_params[:name],
             'type' => 'Microsoft.Compute/availabilitySets',
             'properties' =>
             {

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

@@ -55,7 +55,7 @@ def raise_azure_exception(exception, msg)
               end
     exception_message = "Exception in #{msg} #{message} Type: #{exception.class}\n#{exception.backtrace.join("\n")}"
   else
-    "#{exception.inspect}\n#{exception.backtrace.join("\n")}"
+    exception_message = "#{exception.inspect}\n#{exception.backtrace.join("\n")}"
   end
 
   Fog::Logger.debug exception.backtrace

data/lib/fog/azurerm/version.rb

@@ -1,5 +1,5 @@
 module Fog
   module AzureRM
-    VERSION = '0.0.1'.freeze
+    VERSION = '0.0.2'.freeze
   end
 end

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

@@ -3,15 +3,33 @@ module ApiStub
     module Compute
       # Mock class for Availability Set Model
       class AvailabilitySet
-        def self.create_availability_set_response(sdk_compute_client)
+        def self.create_unmanaged_availability_set_response(sdk_compute_client)
           avail_set = {
             'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/availabilitySets/fog-test-availability-set',
             'name' => 'fog-test-availability-set',
             'type' => 'Microsoft.Compute/availabilitySets',
             'location' => 'westus',
             'platformUpdateDomainCount' => UPDATE_DOMAIN_COUNT,
-            'platformFaultDomainCount' => FAULT_DOMAIN_COUNT
+            'platformFaultDomainCount' => FAULT_DOMAIN_COUNT,
+            'sku' => {
+              'name' => 'Classic'
+            }
+          }
+          result_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
+          sdk_compute_client.deserialize(result_mapper, avail_set, 'result.body')
+        end
 
+        def self.create_managed_availability_set_response(sdk_compute_client)
+          avail_set = {
+            'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/availabilitySets/fog-test-availability-set',
+            'name' => 'fog-test-availability-set',
+            'type' => 'Microsoft.Compute/availabilitySets',
+            'location' => 'westus',
+            'platformUpdateDomainCount' => UPDATE_DOMAIN_COUNT,
+            'platformFaultDomainCount' => FAULT_DOMAIN_COUNT,
+            'sku' => {
+              'name' => 'Aligned'
+            }
           }
           result_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
           sdk_compute_client.deserialize(result_mapper, avail_set, 'result.body')
@@ -25,7 +43,6 @@ module ApiStub
               'location' => 'westus',
               'platformUpdateDomainCount' => UPDATE_DOMAIN_COUNT,
               'platformFaultDomainCount' => FAULT_DOMAIN_COUNT
-
           }
           result_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
           sdk_compute_client.deserialize(result_mapper, avail_set, 'result.body')
@@ -38,8 +55,10 @@ module ApiStub
               'type' => 'Microsoft.Compute/availabilitySets',
               'location' => 'westus',
               'platformUpdateDomainCount' => UPDATE_DOMAIN_COUNT,
-              'platformFaultDomainCount' => FAULT_DOMAIN_COUNT
-
+              'platformFaultDomainCount' => FAULT_DOMAIN_COUNT,
+              'sku' => {
+                'name' => 'Classic'
+              }
           }
           result_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
           sdk_compute_client.deserialize(result_mapper, avail_set, 'result.body')

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

@@ -3,6 +3,42 @@ module ApiStub
     module Compute
       # Mock class for Availability Set Requests
       class AvailabilitySet
+        def self.create_unmanaged_availability_set_response(compute_client)
+          body = '{
+                    "id":"/subscriptions/{subscription-id}/resourceGroups/myrg1/providers/Microsoft.Compute/availabilitySets/avset1",
+                    "name":"myavset1",
+                    "type":"Microsoft.Compute/availabilitySets",
+                    "location":"westus",
+                    "tags": {},
+                    "platformUpdateDomainCount": 5,
+                    "platformFaultDomainCount": 3,
+                    "virtualMachines":[],
+                    "sku":{
+                      "name":"Classic"
+                    }
+                  }'
+          availability_set_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
+          compute_client.deserialize(availability_set_mapper, Fog::JSON.decode(body), 'result.body')
+        end
+
+        def self.create_managed_availability_set_response(compute_client)
+          body = '{
+                    "id":"/subscriptions/{subscription-id}/resourceGroups/myrg1/providers/Microsoft.Compute/availabilitySets/avset1",
+                    "name":"myavset1",
+                    "type":"Microsoft.Compute/availabilitySets",
+                    "location":"westus",
+                    "tags": {},
+                    "platformUpdateDomainCount": 5,
+                    "platformFaultDomainCount": 2,
+                    "virtualMachines":[],
+                    "sku":{
+                      "name":"Aligned"
+                    }
+                  }'
+          availability_set_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
+          compute_client.deserialize(availability_set_mapper, Fog::JSON.decode(body), 'result.body')
+        end
+
         def self.create_availability_set_response(compute_client)
           body = '{
                     "id":"/subscriptions/{subscription-id}/resourceGroups/myrg1/providers/Microsoft.Compute/availabilitySets/avset1",
@@ -11,7 +47,22 @@ module ApiStub
                     "location":"westus",
                     "tags": {},
                     "platformUpdateDomainCount": 5,
-                    "platformFaultDomainCount":3,
+                    "platformFaultDomainCount": 3,
+                    "virtualMachines":[]
+                  }'
+          availability_set_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
+          compute_client.deserialize(availability_set_mapper, Fog::JSON.decode(body), 'result.body')
+        end
+
+        def self.create_custom_availability_set_response(compute_client)
+          body = '{
+                    "id":"/subscriptions/{subscription-id}/resourceGroups/myrg1/providers/Microsoft.Compute/availabilitySets/avset1",
+                    "name":"myavset1",
+                    "type":"Microsoft.Compute/availabilitySets",
+                    "location":"westus",
+                    "tags": {},
+                    "platformUpdateDomainCount": 10,
+                    "platformFaultDomainCount": 3,
                     "virtualMachines":[]
                   }'
           availability_set_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
@@ -28,7 +79,10 @@ module ApiStub
                     "tags": {},
                     "platformUpdateDomainCount":5,
                     "platformFaultDomainCount":3,
-                    "virtualMachines":[]
+                    "virtualMachines":[],
+                    "sku":{
+                      "name":"Classic"
+                    }
               } ]
           }'
           availability_set_mapper = Azure::ARM::Compute::Models::AvailabilitySetListResult.mapper
@@ -45,7 +99,10 @@ module ApiStub
                     "tags": {},
                     "platformUpdateDomainCount":5,
                     "platformFaultDomainCount":3,
-                    "virtualMachines":[]
+                    "virtualMachines":[],
+                    "sku":{
+                      "name":"Classic"
+                    }
               } ]
           }'
           availability_set_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper

data/test/integration/application_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/availability_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__))
 
 rs = Fog::Resources::AzureRM.new(
   tenant_id: azure_credentials['tenant_id'],
@@ -29,41 +29,114 @@ compute = Fog::Compute::AzureRM.new(
 ########################################################################################################################
 
 time = current_time
-resource_group_name = "AS-RG-#{time}"
-availability_set_name = "AS#{time}asetunique"
+resource_group_name       = "AS-RG-#{time}"
+unmanaged_as_name_default = "ASUnmanagedDefault#{time}"
+unmanaged_as_name_custom  = "ASUnmanagedCustom#{time}"
+managed_as_name_default   = "ASManagedDefault#{time}"
+managed_as_name_custom    = "ASManagedCustom#{time}"
 
 ########################################################################################################################
 ######################                                 Prerequisites                              ######################
 ########################################################################################################################
 
 begin
+  puts 'Running integration test for availability set...'
+
+  puts "Create resource group (#{resource_group_name}):"
   resource_group = rs.resource_groups.create(
     name: resource_group_name,
     location: LOCATION
   )
-
+  puts "Created resource group! [#{resource_group.name}]"
 
   ########################################################################################################################
-  ######################                            Check for Availability set                       ######################
+  ######################                            Check for Availability set                      ######################
   ########################################################################################################################
 
-  flag = compute.availability_sets.check_availability_set_exists(resource_group_name, availability_set_name)
-  puts "Availability set doesn't exist." unless flag
+  puts 'Check for existing availability sets:'
+
+  is_exists = compute.availability_sets.check_availability_set_exists(resource_group_name, unmanaged_as_name_default)
+  puts "Availability set does NOT exist! [#{unmanaged_as_name_default}] " unless is_exists
+
+  is_exists = compute.availability_sets.check_availability_set_exists(resource_group_name, unmanaged_as_name_custom)
+  puts "Availability set does NOT exist! [#{unmanaged_as_name_custom}] " unless is_exists
+
+  is_exists = compute.availability_sets.check_availability_set_exists(resource_group_name, managed_as_name_default)
+  puts "Availability set does NOT exist! [#{managed_as_name_default}] " unless is_exists
+
+  is_exists = compute.availability_sets.check_availability_set_exists(resource_group_name, managed_as_name_custom)
+  puts "Availability set does NOT exist! [#{managed_as_name_default}] " unless is_exists
 
   ########################################################################################################################
-  ######################                             Create Availability Set                        ######################
+  ######################                    Create Unmanaged Availability Set (Default)             ######################
   ########################################################################################################################
 
+  puts "Create unmanaged default availability set (#{unmanaged_as_name_default}):"
   avail_set = compute.availability_sets.create(
-    name: availability_set_name,
+    name: unmanaged_as_name_default,
     location: LOCATION,
     resource_group: resource_group_name
   )
-  puts "Created availability set: #{avail_set.name}"
+  name = avail_set.name
+  fault_domains = avail_set.platform_fault_domain_count
+  update_domains = avail_set.platform_update_domain_count
+  puts "Created availability set! [#{name}] => { fd: #{fault_domains}, ud: #{update_domains} }"
+
+  ########################################################################################################################
+  ######################                    Create Unmanaged Availability Set (Custom)              ######################
+  ########################################################################################################################
+
+  puts "Create unmanaged custom availability set (#{unmanaged_as_name_custom}):"
+  avail_set = compute.availability_sets.create(
+    name: unmanaged_as_name_custom,
+    location: LOCATION,
+    resource_group: resource_group_name,
+    platform_fault_domain_count: 3,
+    platform_update_domain_count: 10
+  )
+  name = avail_set.name
+  fault_domains = avail_set.platform_fault_domain_count
+  update_domains = avail_set.platform_update_domain_count
+  puts "Created availability set! [#{name}] => { fd: #{fault_domains}, ud: #{update_domains} }"
+
+  ########################################################################################################################
+  ######################                    Create Managed Availability Set (Default)               ######################
+  ########################################################################################################################
+
+  puts "Create managed default availability set (#{managed_as_name_default}):"
+  avail_set = compute.availability_sets.create(
+    name: managed_as_name_default,
+    location: LOCATION,
+    resource_group: resource_group_name,
+    is_managed: true
+  )
+  name = avail_set.name
+  fault_domains = avail_set.platform_fault_domain_count
+  update_domains = avail_set.platform_update_domain_count
+  puts "Created availability set! [#{name}] => { fd: #{fault_domains}, ud: #{update_domains} }"
+
+  ########################################################################################################################
+  ######################                     Create Managed Availability Set (Custom)               ######################
+  ########################################################################################################################
+
+  puts "Create managed custom availability set (#{managed_as_name_custom}):"
+  avail_set = compute.availability_sets.create(
+    name: managed_as_name_custom,
+    location: LOCATION,
+    resource_group: resource_group_name,
+    platform_fault_domain_count: 2,
+    platform_update_domain_count: 10,
+    is_managed: true
+  )
+  name = avail_set.name
+  fault_domains = avail_set.platform_fault_domain_count
+  update_domains = avail_set.platform_update_domain_count
+  puts "Created availability set! [#{name}] => { fd: #{fault_domains}, ud: #{update_domains} }"
 
   ########################################################################################################################
   ######################                       List Availability Sets                               ######################
   ########################################################################################################################
+
   puts 'List availability sets:'
   compute.availability_sets(resource_group: resource_group_name).each do |availability_set|
     puts availability_set.name
@@ -73,18 +146,32 @@ begin
   ######################                       Get and Delete Availability Set                      ######################
   ########################################################################################################################
 
-  avail_set = compute.availability_sets.get(resource_group_name, availability_set_name)
-  puts "Get availability set: #{avail_set.name}"
-  puts "Deleted availability set: #{avail_set.destroy}"
+  puts 'Get and delete availability sets:'
+
+  avail_set = compute.availability_sets.get(resource_group_name, unmanaged_as_name_default)
+  puts "Get availability set     : #{avail_set.name}"
+  puts "Deleted availability set : #{avail_set.destroy}"
+
+  avail_set = compute.availability_sets.get(resource_group_name, unmanaged_as_name_custom)
+  puts "Get availability set     : #{avail_set.name}"
+  puts "Deleted availability set : #{avail_set.destroy}"
+
+  avail_set = compute.availability_sets.get(resource_group_name, managed_as_name_default)
+  puts "Get availability set     : #{avail_set.name}"
+  puts "Deleted availability set : #{avail_set.destroy}"
+
+  avail_set = compute.availability_sets.get(resource_group_name, managed_as_name_custom)
+  puts "Get availability set     : #{avail_set.name}"
+  puts "Deleted availability set : #{avail_set.destroy}"
 
   ########################################################################################################################
-  ######################                                   CleanUp                                  ######################
+  ######################                                   Clean Up                                 ######################
   ########################################################################################################################
 
   rg = rs.resource_groups.get(resource_group_name)
   rg.destroy
-  puts 'Integration Test for availability set ran successfully'
+  puts 'Integration test for availability set ran successfully!'
 rescue
-  puts 'Integration Test for availability set is failing'
+  puts 'Integration Test for availability set is failing!!!'
   resource_group.destroy unless resource_group.nil?
 end