lib/fog/azurerm/models/compute/availability_set.rb in fog-azure-rm-0.0.1 vs lib/fog/azurerm/models/compute/availability_set.rb in fog-azure-rm-0.0.2

@@ -8,19 +8,33 @@
       class AvailabilitySet < Fog::Model
         attribute :id
         identity  :name
         attribute :type
         attribute :location
-        attribute :tags
         attribute :resource_group
-        attribute :properties
+        attribute :platform_update_domain_count
+        attribute :platform_fault_domain_count
 
+        def self.parse(as)
+          hash = {}
+          hash['id'] = as['id']
+          hash['name'] = as['name']
+          hash['type'] = as['type']
+          hash['location'] = as['location']
+          hash['resource_group'] = as['id'].split('/')[4]
+          hash['platform_update_domain_count'] = as['properties']['platformUpdateDomainCount']
+          hash['platform_fault_domain_count'] = as['properties']['platformFaultDomainCount']
+          hash
+        end
+
         def save
           requires :name
           requires :location
           requires :resource_group
           # need to create the availability set
-          service.create_availability_set(resource_group, name, location)
+          as = service.create_availability_set(resource_group, name, location)
+          hash = Fog::Compute::AzureRM::AvailabilitySet.parse(as)
+          merge_attributes(hash)
         end
 
         def destroy
           service.delete_availability_set(resource_group, name)
         end