diff --git a/lib/default_value_for.rb b/lib/default_value_for.rb index fc5b573..f4f4389 100644 --- a/lib/default_value_for.rb +++ b/lib/default_value_for.rb @@ -171,6 +171,17 @@ def set_default_values end next unless connection_default_value_defined || attribute_blank + is_present = -> (value) do + if [true, false].include?(value) + !value.nil? + else + value.present? + end + end + + # Do not overwrite if the attribute is present in @initialization_attributes + next if @initialization_attributes.is_a?(Hash) && is_present.call(@initialization_attributes.dig(attribute)) + # allow explicitly setting nil through allow nil option next if @initialization_attributes.is_a?(Hash) && ( diff --git a/test.rb b/test.rb index 361c6a3..4d73f35 100644 --- a/test.rb +++ b/test.rb @@ -90,6 +90,7 @@ class ToSql < (Arel::VERSION[/\A\d+/].to_i >= 6 ? Arel::Visitors::Reduce : Arel: t.string :type t.integer :number t.integer :count, :null => false, :default => 1 + t.boolean :flag2, :null => false, :default => false t.integer :user_id t.timestamp :timestamp t.text :stuff @@ -168,6 +169,16 @@ def test_overwrites_db_default assert_equal 1234, Book.new.count end + def test_does_not_overwrite_db_default_if_provided_by_mass_assignment + Book.default_value_for(:count, :allows_nil => false) { 1234 } + assert_equal Book.column_defaults['count'], Book.new(count: Book.column_defaults['count']).count + end + + def test_does_not_overwrite_db_default_boolean_value_if_provided_by_mass_assignment + Book.default_value_for(:flag2, :allows_nil => false) { true } + assert_equal Book.column_defaults['flag2'], Book.new(flag2: Book.column_defaults['flag2']).flag2 + end + def test_doesnt_overwrite_values_provided_by_mass_assignment Book.default_value_for :number, 1234 assert_equal 1, Book.new(:number => 1, :count => 2).number @@ -199,6 +210,11 @@ def test_overwrites_explicitly_provided_nil_values_in_mass_assignment assert_equal 1234, Book.new(:number => nil).number end + def test_overwrites_string_blank_value_in_mass_assignment + Book.default_value_for(:type, :allows_nil => false) { 'string' } + assert_equal 'string', Book.new(type: '').type + end + def test_default_values_are_inherited Book.default_value_for :number, 1234 assert_equal 1234, Novel.new.number