lib/mongo_mapper/plugins/associations.rb in mongo_mapper-0.10.1 vs lib/mongo_mapper/plugins/associations.rb in mongo_mapper-0.11.0

@@ -54,38 +54,36 @@
             associations[association.name] = association
             association.setup(self)
           end
       end
 
-      module InstanceMethods
-        def associations
-          self.class.associations
-        end
+      def associations
+        self.class.associations
+      end
 
-        def embedded_associations
-          associations.values.select { |assoc| assoc.embeddable? }
-        end
+      def embedded_associations
+        associations.values.select { |assoc| assoc.embeddable? }
+      end
 
-        def build_proxy(association)
-          proxy = association.proxy_class.new(self, association)
-          self.instance_variable_set(association.ivar, proxy)
+      def build_proxy(association)
+        proxy = association.proxy_class.new(self, association)
+        self.instance_variable_set(association.ivar, proxy)
 
-          proxy
-        end
+        proxy
+      end
 
-        def get_proxy(association)
-          unless proxy = self.instance_variable_get(association.ivar)
-            proxy = build_proxy(association)
-          end
-          proxy
+      def get_proxy(association)
+        unless proxy = self.instance_variable_get(association.ivar)
+          proxy = build_proxy(association)
         end
+        proxy
+      end
 
-        def save_to_collection(options={})
-          super if defined?(super)
-          associations.each do |association_name, association|
-            proxy = get_proxy(association)
-            proxy.save_to_collection(options) if proxy.proxy_respond_to?(:save_to_collection) && association.autosave?
-          end
+      def save_to_collection(options={})
+        super if defined?(super)
+        associations.each do |association_name, association|
+          proxy = get_proxy(association)
+          proxy.save_to_collection(options) if proxy.proxy_respond_to?(:save_to_collection) && association.autosave?
         end
       end
     end
   end
 end