@@ -11,11 +11,13 @@ module ConstantizeAttribute module_eval <<-end_eval, __FILE__, __LINE__ def #{name}= value - write_attribute(:#{name}, value.to_s) + value = value.nil? ? nil : value.to_s + write_attribute(:#{name}, value) end def #{name} - read_attribute(:#{name}).constantize + value = read_attribute(:#{name}) + value.nil? ? nil : value.constantize end end_eval lib/constantize_attribute.rb @@ -53,4 +53,16 @@ class ConstantizeAttributeTest < Test::Unit::TestCase def test_constantize_attributes_is_the_same_as_constantize_attribute assert_equal ConstantizeAttribute.instance_method(:constantize_attribute), ConstantizeAttribute.instance_method(:constantize_attributes) end + + def test_writer_passes_nil_along_as_nil + m = SomeModel.new + m.foo = nil + assert_nil m.attributes[:foo] + end + + def test_reader_passes_nil_along_as_nil + m = SomeModel.new + m.attributes[:foo] = nil + assert_nil m.foo + end end test/constantize_attribute_test.rb 4069d16ee287391b2c070d1c31f029a7d1dacd13 Chris O'Meara chris@omeara.com http://github.com/duelinmarkers/constantize_attribute/commit/a2b13a10a043ec69ea9a56f214554e399ca6f0ba a2b13a10a043ec69ea9a56f214554e399ca6f0ba 2009-06-03T22:31:54-07:00 2009-06-03T22:31:54-07:00 Changed attr readers and writers to preserve nil values e901bff8011a29010cce89e0af30a8b1a6bdcadb Chris O'Meara chris@omeara.com