Updating parts list to remove warning on over riding map

mikel · Jun 15, 2010 · 9e26bfe · 9e26bfe
1 parent 75f2c02
commit 9e26bfe
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/mail/parts_list.rb b/lib/mail/parts_list.rb
@@ -14,8 +14,17 @@ def collect
         to_a
       end
     end
+
+    undef :map
+    alias_method :map, :collect
 
-    alias :collect :map 
+    def map!
+      raise NoMethodError, "#map! is not defined, please call #collect and create a new PartsList"
+    end
+
+    def collect!
+      raise NoMethodError, "#collect! is not defined, please call #collect and create a new PartsList"
+    end
 
     def sort!(order)
       sorted = self.sort do |a, b|