test/models/compute/test_availability_set.rb in fog-azure-rm-0.3.2 vs test/models/compute/test_availability_set.rb in fog-azure-rm-0.3.3

@@ -4,11 +4,11 @@
 class TestAvailabilitySet < Minitest::Test
   def setup
     @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
     methods = [
       :save,
@@ -25,10 +25,12 @@
       :name,
       :type,
       :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
     end
   end