From e3246ef5b50d48aa86ee41a161b42611a936f4fb Mon Sep 17 00:00:00 2001 From: Emilio Tagua Date: Thu, 9 Sep 2010 15:27:10 -0300 Subject: [PATCH] Add tests for Relation#only. --- activerecord/test/cases/relations_test.rb | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/activerecord/test/cases/relations_test.rb b/activerecord/test/cases/relations_test.rb index b7112d0e37..9901f09f57 100644 --- a/activerecord/test/cases/relations_test.rb +++ b/activerecord/test/cases/relations_test.rb @@ -647,6 +647,17 @@ class RelationTest < ActiveRecord::TestCase assert_equal Post.all, all_posts.all end + def test_only + relation = Post.where(:author_id => 1).order('id ASC').limit(1) + assert_equal [posts(:welcome)], relation.all + + author_posts = relation.only(:where) + assert_equal Post.where(:author_id => 1).all, author_posts.all + + all_posts = relation.only(:limit) + assert_equal Post.limit(1).all.first, all_posts.first + end + def test_anonymous_extension relation = Post.where(:author_id => 1).order('id ASC').extending do def author