aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/cases/relations_test.rb
diff options
context:
space:
mode:
authorPratik Naik <pratiknaik@gmail.com>2010-04-02 17:34:48 +0100
committerPratik Naik <pratiknaik@gmail.com>2010-04-02 17:38:02 +0100
commitb77dd218ce845f01753d02fcbc2605c9a5ee93e1 (patch)
treeea07dbe6965bd31ea7d066e04f7dad0e67e62f31 /activerecord/test/cases/relations_test.rb
parentbc7da9b77d6347eeccefa2c735b2f236a08eea57 (diff)
downloadrails-b77dd218ce845f01753d02fcbc2605c9a5ee93e1.tar.gz
rails-b77dd218ce845f01753d02fcbc2605c9a5ee93e1.tar.bz2
rails-b77dd218ce845f01753d02fcbc2605c9a5ee93e1.zip
Add Relation extensions
Diffstat (limited to 'activerecord/test/cases/relations_test.rb')
-rw-r--r--activerecord/test/cases/relations_test.rb16
1 files changed, 16 insertions, 0 deletions
diff --git a/activerecord/test/cases/relations_test.rb b/activerecord/test/cases/relations_test.rb
index 1e345399f5..7b9e680c02 100644
--- a/activerecord/test/cases/relations_test.rb
+++ b/activerecord/test/cases/relations_test.rb
@@ -572,4 +572,20 @@ class RelationTest < ActiveRecord::TestCase
assert_equal Post.all, all_posts.all
end
+ def test_anonymous_extension
+ relation = Post.where(:author_id => 1).order('id ASC') do
+ def author
+ 'lifo'
+ end
+ end
+
+ assert_equal "lifo", relation.author
+ assert_equal "lifo", relation.limit(1).author
+ end
+
+ def test_named_extension
+ relation = Post.where(:author_id => 1).order('id ASC').extending(Post::NamedExtension)
+ assert_equal "lifo", relation.author
+ assert_equal "lifo", relation.limit(1).author
+ end
end