attribute_sanitizer 0.0.2 → 0.0.3

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    attribute_sanitizer (0.0.1)
+    attribute_sanitizer (0.0.2)
       activesupport
       i18n
 

data/lib/attribute_sanitizer_helpers.rb

@@ -45,17 +45,14 @@ class AttributeSanitizer
     # end
     #
     # Foo.sanitize_attributes("1,2,3") # => ["1", "2", "3"]
-    def sanitize_attributes(attrs=nil, &block)
-      if attrs
-        return attrs unless @_attribute_sanitizer
-        @_attribute_sanitizer.sanitize(attrs)
-      elsif block_given?
+    def sanitize_attributes(*args, &block)
+      if block_given?
         @_attribute_sanitizer = AttributeSanitizer.new(self)
         @_attribute_sanitizer.instance_eval(&block)
       else
-        raise "sanitize_attributes needs an argument"
+        retval = @_attribute_sanitizer ? args.map { |val| @_attribute_sanitizer.sanitize(val) } : args
+        retval.size < 2 ? retval.first : retval
       end
     end
-
   end
 end

data/test/attribute_sanitizer_test.rb

@@ -92,6 +92,15 @@ class TestAttributeSanitizer < Test::Unit::TestCase
       end
       assert_equal({bar: 1}, Bagel.sanitize_attributes(foo: 1))
     end
+
+    should "let us sanitize nil" do
+      Bagel.sanitize_attributes do
+        add_step do |val|
+          "pass"
+        end
+      end
+      assert_equal "pass", Bagel.sanitize_attributes(nil)
+    end
   end
 
   def stringify(val)

metadata

@@ -2,7 +2,7 @@
 name: attribute_sanitizer
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors: 
 - Jim Garvin
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-07-20 00:00:00 Z
+date: 2011-07-21 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: activesupport