test/ruby_test.rb in data_cleansing-0.2.0 vs test/ruby_test.rb in data_cleansing-0.3.0

@@ -19,10 +19,42 @@
 
   # Define a once off cleaner
   cleanse :address1, :address2, :cleaner => Proc.new {|string| "<< #{string.strip!} >>"}
 end
 
+# Another global cleaner, used by RubyUser2
+DataCleansing.register_cleaner(:upcase) {|string| string.upcase!}
+
+class RubyUser2
+  include DataCleansing::Cleanse
+
+  attr_accessor :first_name, :last_name, :title, :address1, :address2, :gender
+
+  # Use a global cleaner
+  cleanse :first_name, :last_name, :cleaner => :strip
+
+  # Define a once off cleaner
+  cleanse :address1, :address2, :cleaner => Proc.new {|string| string.strip!}
+
+  # Use multiple cleaners, and a custom block
+  cleanse :title, :cleaner => [:strip, :upcase, Proc.new {|string| "#{string}." unless string.end_with?('.')}]
+
+  # Change the cleansing rule based on the value of other attributes in that instance of user
+  # The 'title' is retrieved from the current instance of the user
+  cleanse :gender, :cleaner => [
+    :strip,
+    :upcase,
+    Proc.new do |gender|
+      if (gender == "UNKNOWN") && (title == "MR.")
+        "Male"
+      else
+        "Female"
+      end
+    end
+  ]
+end
+
 class RubyTest < Test::Unit::TestCase
   context "Ruby Models" do
 
     should 'have globally registered cleaner' do
       assert DataCleansing.cleaner(:strip)
@@ -49,9 +81,37 @@
 
       should 'cleanse_attributes! not cleanse nil attributes' do
         @user.first_name = nil
         @user.cleanse_attributes!
         assert_equal nil, @user.first_name
+      end
+    end
+
+    context "with ruby user2" do
+      setup do
+        @user = RubyUser2.new
+        @user.first_name = '    joe   '
+        @user.last_name  = "\n  black\n"
+        @user.address1   = "2632 Brown St   \n"
+        @user.title      = "   \nmr   \n"
+        @user.gender     = " Unknown  "
+      end
+
+      should 'cleanse_attributes!' do
+        @user.cleanse_attributes!
+        assert_equal 'joe', @user.first_name
+        assert_equal 'black', @user.last_name
+        assert_equal '2632 Brown St', @user.address1
+      end
+
+      should 'cleanse_attributes! with multiple cleaners' do
+        @user.cleanse_attributes!
+        assert_equal 'MR.', @user.title
+      end
+
+      should 'cleanse_attributes! referencing other attributes' do
+        @user.cleanse_attributes!
+        assert_equal 'Male', @user.gender
       end
     end
 
   end
 end
\ No newline at end of file