Implement :required option.

freegenie · Apr 8, 2010 · 4393e95 · 4393e95
1 parent c2f2389
commit 4393e95
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/lib/joint.rb b/lib/joint.rb
@@ -11,7 +11,10 @@ def self.configure(model)
   end
 
   module ClassMethods
-    def attachment(name)
+    def attachment(name, options = {})
+      options.symbolize_keys!
+      name = name.to_sym
+
       self.attachment_names << name
 
       after_save     :save_attachments
@@ -23,6 +26,8 @@ def attachment(name)
       key "#{name}_size".to_sym, Integer
       key "#{name}_type".to_sym, String
 
+      validates_presence_of(name) if options[:required]
+
       class_eval <<-EOC
         def #{name}
           @#{name} ||= AttachmentProxy.new(self, :#{name})

diff --git a/test/test_joint.rb b/test/test_joint.rb
@@ -306,8 +306,7 @@ def @image.original_filename
       @model_class = Class.new do
         include MongoMapper::Document
         plugin Joint
-        attachment :file
-        validates_presence_of :file
+        attachment :file, :required => true
       end
     end