lib/fog/compute/xen_server/models/pbd.rb in fog-xenserver-0.0.1.beta1 vs lib/fog/compute/xen_server/models/pbd.rb in fog-xenserver-0.0.1.beta2

@@ -1,29 +1,26 @@
 module Fog
   module Compute
     class XenServer
       module Models
-        class Pbd < Fog::Model
+        class Pbd < Model
           # API Reference here:
           # http://docs.vmd.citrix.com/XenServer/6.2.0/1.0/en_gb/api/?c=PBD
 
           provider_class :PBD
+          collection_name :pbds
 
           identity :reference
 
           attribute :currently_attached
           attribute :device_config
           attribute :other_config
           attribute :uuid
 
-          has_one :host,  :hosts
-          has_one :sr,    :storage_repositories,    :aliases => :SR
+          has_one_identity :host,  :hosts
+          has_one_identity :sr,    :storage_repositories,    :aliases => :SR,   :as => :SR
 
           alias_method :storage_repository, :sr
-
-          def unplug
-            service.unplug_pbd reference
-          end
         end
       end
     end
   end
 end
\ No newline at end of file