From 72e63c71bb3f73870e280964def25b7578818b1b Mon Sep 17 00:00:00 2001 From: Gannon McGibbon Date: Thu, 29 Nov 2018 14:43:38 -0500 Subject: [PATCH] Allow aliased attributes in update Allow aliased attributes to be used in `#update_columns` and `#update`. --- activerecord/CHANGELOG.md | 4 ++++ .../lib/active_record/attribute_methods/dirty.rb | 10 +++++++--- activerecord/test/cases/attribute_methods_test.rb | 6 ++++++ 3 files changed, 17 insertions(+), 3 deletions(-) diff --git a/activerecord/CHANGELOG.md b/activerecord/CHANGELOG.md index d32c6c3c5d2c0..5b07d636847c5 100644 --- a/activerecord/CHANGELOG.md +++ b/activerecord/CHANGELOG.md @@ -1,3 +1,7 @@ +* Allow aliased attributes to be used in `#update_columns` and `#update`. + + *Gannon McGibbon* + * Allow spaces in postgres table names. Fixes issue where "user post" is misinterpreted as "\"user\".\"post\"" when quoting table names with the postgres adapter. diff --git a/activerecord/lib/active_record/attribute_methods/dirty.rb b/activerecord/lib/active_record/attribute_methods/dirty.rb index ebc2252c5029a..45e4b8adfaa76 100644 --- a/activerecord/lib/active_record/attribute_methods/dirty.rb +++ b/activerecord/lib/active_record/attribute_methods/dirty.rb @@ -158,9 +158,13 @@ def attributes_in_database end private - def write_attribute_without_type_cast(attr_name, _) - result = super - clear_attribute_change(attr_name) + def write_attribute_without_type_cast(attr_name, value) + name = attr_name.to_s + if self.class.attribute_alias?(name) + name = self.class.attribute_alias(name) + end + result = super(name, value) + clear_attribute_change(name) result end diff --git a/activerecord/test/cases/attribute_methods_test.rb b/activerecord/test/cases/attribute_methods_test.rb index 12a4027a4e6fb..d341dd0083b42 100644 --- a/activerecord/test/cases/attribute_methods_test.rb +++ b/activerecord/test/cases/attribute_methods_test.rb @@ -323,6 +323,12 @@ def setup assert_raises(ActiveModel::UnknownAttributeError) { topic.update(no_column_exists: "Hello!") } end + test "write_attribute allows writing to aliased attributes" do + topic = Topic.first + assert_nothing_raised { topic.update_columns(heading: "Hello!") } + assert_nothing_raised { topic.update(heading: "Hello!") } + end + test "read_attribute" do topic = Topic.new topic.title = "Don't change the topic"