Merge pull request #43 from spemmons/master

fix for issue #37 where reloading a model doesn't refresh the ValueProxy objects

lib/bitmask_attributes.rb

@@ -43,6 +43,12 @@ def base_class_bitmasks
       @bitmasks ||= {}
     end
   end
+
+  def reload(*)
+    super
+    self.class.bitmasks.keys.each{|attribute| self.send("reload_#{attribute}")}
+    self
+  end
 end
 
 ActiveRecord::Base.send :include, BitmaskAttributes

lib/bitmask_attributes/definition.rb

@@ -60,6 +60,9 @@ def override_getter_on(model)
           def #{attribute}
             @#{attribute} ||= BitmaskAttributes::ValueProxy.new(self, :#{attribute}, &self.class.bitmask_definitions[:#{attribute}].extension)
           end
+          def reload_#{attribute}
+            @#{attribute} = nil
+          end
         )
       end
 

test/bitmask_attributes_test.rb

@@ -139,6 +139,19 @@ def self.context_with_classes(label, campaign_class, company_class)
         assert_stored campaign, :web, :print
       end
 
+      should "update bitmask values currently in the database with reload" do
+        instance1 = @campaign_class.create(:medium => [:web, :print])
+        instance2 = @campaign_class.find(instance1.id)
+        assert instance1.id == instance2.id
+        assert instance1.object_id != instance2.object_id
+        assert instance1.update_attributes(:medium => [:email])
+
+        assert_equal [:web, :print],instance2.medium
+
+        assert_equal @campaign_class,instance2.reload.class
+        assert_equal [:email],instance2.medium
+      end
+
       context "checking" do
         setup { @campaign = @campaign_class.new(:medium => [:web, :print]) }