lib/fog/azurerm/models/dns/zone.rb in fog-azure-rm-0.0.1 vs lib/fog/azurerm/models/dns/zone.rb in fog-azure-rm-0.0.2

@@ -1,20 +1,43 @@
 module Fog
   module DNS
     class AzureRM
+      # This class is giving implementation of create/save and
+      # delete/destroy for Zone.
       class Zone < Fog::Model
-        identity :name
         attribute :id
+        identity :name
         attribute :resource_group
+        attribute :location
+        attribute :type
+        attribute :tags
+        attribute :etag
+        attribute :number_of_record_sets
+        attribute :max_number_of_recordsets
 
+        def self.parse(zone)
+          hash = {}
+          hash['id'] = zone['id']
+          hash['name'] = zone['name']
+          hash['resource_group'] = zone['id'].split('/')[4]
+          hash['location'] = zone['location']
+          hash['type'] = zone['type']
+          hash['tags'] = zone['tags']
+          hash['etag'] = zone['etag']
+          hash['number_of_record_sets'] = zone['properties']['numberOfRecordSets']
+          hash['max_number_of_recordsets'] = zone['properties']['maxNumberOfRecordSets']
+          hash
+        end
+
         def save
           requires :name
           requires :resource_group
-          service.create_zone(resource_group, name)
+          zone = service.create_zone(resource_group, name)
+          merge_attributes(Fog::DNS::AzureRM::Zone.parse(zone))
         end
 
         def destroy
-          service.delete_zone(name, resource_group)
+          service.delete_zone(resource_group, name)
         end
       end
     end
   end
 end