2012-05-30 18:09:13 -04:00
|
|
|
require "cases/helper"
|
2012-09-11 14:11:51 -04:00
|
|
|
require 'models/author'
|
|
|
|
require 'models/price_estimate'
|
|
|
|
require 'models/treasure'
|
2012-05-30 18:09:13 -04:00
|
|
|
require 'models/post'
|
2012-09-12 18:32:50 -04:00
|
|
|
require 'models/comment'
|
2012-07-05 07:24:59 -04:00
|
|
|
require 'models/edge'
|
2013-05-01 14:33:11 -04:00
|
|
|
require 'models/topic'
|
2012-05-30 18:09:13 -04:00
|
|
|
|
|
|
|
module ActiveRecord
|
|
|
|
class WhereTest < ActiveRecord::TestCase
|
2013-01-24 17:29:03 -05:00
|
|
|
fixtures :posts, :edges, :authors
|
|
|
|
|
|
|
|
def test_where_copies_bind_params
|
|
|
|
author = authors(:david)
|
|
|
|
posts = author.posts.where('posts.id != 1')
|
|
|
|
joined = Post.where(id: posts)
|
|
|
|
|
|
|
|
assert_operator joined.length, :>, 0
|
|
|
|
|
|
|
|
joined.each { |post|
|
|
|
|
assert_equal author, post.author
|
|
|
|
assert_not_equal 1, post.id
|
|
|
|
}
|
|
|
|
end
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2013-11-02 22:45:03 -04:00
|
|
|
def test_rewhere_on_root
|
|
|
|
assert_equal posts(:welcome), Post.rewhere(title: 'Welcome to the weblog').first
|
|
|
|
end
|
|
|
|
|
2012-09-11 14:11:51 -04:00
|
|
|
def test_belongs_to_shallow_where
|
2012-09-12 18:32:50 -04:00
|
|
|
author = Author.new
|
|
|
|
author.id = 1
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
assert_equal Post.where(author_id: 1).to_sql, Post.where(author: author).to_sql
|
2012-09-11 14:11:51 -04:00
|
|
|
end
|
|
|
|
|
Better support for `where()` conditions that use an association name.
Using the name of an association in `where` previously worked only
if the value was a single `ActiveRecrd::Base` object. e.g.
Post.where(author: Author.first)
Any other values, including `nil`, would cause invalid SQL to be
generated. This change supports arguments in the `where` query
conditions where the key is a `belongs_to` association name and the
value is `nil`, an `Array` of `ActiveRecord::Base` objects, or an
`ActiveRecord::Relation` object.
# Given the Post model
class Post < ActiveRecord::Base
belongs_to :author
end
# nil value finds records where the association is not set
Post.where(author: nil)
# SELECT "posts".* FROM "posts" WHERE "posts"."author_id" IS NULL
# Array values find records where the association foreign key
# matches the ids of the passed ActiveRecord models, resulting
# in the same query as Post.where(author_id: [1,2])
authors_array = [Author.find(1), Author.find(2)]
Post.where(author: authors_array)
# ActiveRecord::Relation values find records using the same
# query as Post.where(author_id: Author.where(last_name: "Emde"))
Post.where(author: Author.where(last_name: "Emde"))
Polymorphic `belongs_to` associations will continue to be handled
appropriately, with the polymorphic `association_type` field added
to the query to match the base class of the value. This feature
previously only worked when the value was a single `ActveRecord::Base`.
class Post < ActiveRecord::Base
belongs_to :author, polymorphic: true
end
Post.where(author: Author.where(last_name: "Emde"))
# Generates a query similar to:
Post.where(author_id: Author.where(last_name: "Emde"), author_type: "Author")
2013-12-16 17:16:15 -05:00
|
|
|
def test_belongs_to_nil_where
|
|
|
|
assert_equal Post.where(author_id: nil).to_sql, Post.where(author: nil).to_sql
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_belongs_to_array_value_where
|
|
|
|
assert_equal Post.where(author_id: [1,2]).to_sql, Post.where(author: [1,2]).to_sql
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_belongs_to_nested_relation_where
|
|
|
|
expected = Post.where(author_id: Author.where(id: [1,2])).to_sql
|
|
|
|
actual = Post.where(author: Author.where(id: [1,2])).to_sql
|
|
|
|
|
|
|
|
assert_equal expected, actual
|
|
|
|
end
|
|
|
|
|
2012-09-11 14:11:51 -04:00
|
|
|
def test_belongs_to_nested_where
|
2012-09-12 18:32:50 -04:00
|
|
|
parent = Comment.new
|
|
|
|
parent.id = 1
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
expected = Post.where(comments: { parent_id: 1 }).joins(:comments)
|
|
|
|
actual = Post.where(comments: { parent: parent }).joins(:comments)
|
|
|
|
|
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
2012-09-11 14:11:51 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_polymorphic_shallow_where
|
2012-09-12 18:32:50 -04:00
|
|
|
treasure = Treasure.new
|
|
|
|
treasure.id = 1
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
expected = PriceEstimate.where(estimate_of_type: 'Treasure', estimate_of_id: 1)
|
|
|
|
actual = PriceEstimate.where(estimate_of: treasure)
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
2012-09-11 14:11:51 -04:00
|
|
|
end
|
|
|
|
|
Better support for `where()` conditions that use an association name.
Using the name of an association in `where` previously worked only
if the value was a single `ActiveRecrd::Base` object. e.g.
Post.where(author: Author.first)
Any other values, including `nil`, would cause invalid SQL to be
generated. This change supports arguments in the `where` query
conditions where the key is a `belongs_to` association name and the
value is `nil`, an `Array` of `ActiveRecord::Base` objects, or an
`ActiveRecord::Relation` object.
# Given the Post model
class Post < ActiveRecord::Base
belongs_to :author
end
# nil value finds records where the association is not set
Post.where(author: nil)
# SELECT "posts".* FROM "posts" WHERE "posts"."author_id" IS NULL
# Array values find records where the association foreign key
# matches the ids of the passed ActiveRecord models, resulting
# in the same query as Post.where(author_id: [1,2])
authors_array = [Author.find(1), Author.find(2)]
Post.where(author: authors_array)
# ActiveRecord::Relation values find records using the same
# query as Post.where(author_id: Author.where(last_name: "Emde"))
Post.where(author: Author.where(last_name: "Emde"))
Polymorphic `belongs_to` associations will continue to be handled
appropriately, with the polymorphic `association_type` field added
to the query to match the base class of the value. This feature
previously only worked when the value was a single `ActveRecord::Base`.
class Post < ActiveRecord::Base
belongs_to :author, polymorphic: true
end
Post.where(author: Author.where(last_name: "Emde"))
# Generates a query similar to:
Post.where(author_id: Author.where(last_name: "Emde"), author_type: "Author")
2013-12-16 17:16:15 -05:00
|
|
|
def test_polymorphic_nested_array_where
|
|
|
|
treasure = Treasure.new
|
|
|
|
treasure.id = 1
|
|
|
|
hidden = HiddenTreasure.new
|
|
|
|
hidden.id = 2
|
|
|
|
|
|
|
|
expected = PriceEstimate.where(estimate_of_type: 'Treasure', estimate_of_id: [treasure, hidden])
|
|
|
|
actual = PriceEstimate.where(estimate_of: [treasure, hidden])
|
|
|
|
|
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_polymorphic_nested_relation_where
|
|
|
|
expected = PriceEstimate.where(estimate_of_type: 'Treasure', estimate_of_id: Treasure.where(id: [1,2]))
|
|
|
|
actual = PriceEstimate.where(estimate_of: Treasure.where(id: [1,2]))
|
|
|
|
|
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
|
|
|
end
|
|
|
|
|
2012-09-11 14:11:51 -04:00
|
|
|
def test_polymorphic_sti_shallow_where
|
2012-09-12 18:32:50 -04:00
|
|
|
treasure = HiddenTreasure.new
|
|
|
|
treasure.id = 1
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
expected = PriceEstimate.where(estimate_of_type: 'Treasure', estimate_of_id: 1)
|
|
|
|
actual = PriceEstimate.where(estimate_of: treasure)
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
2012-09-11 14:11:51 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_polymorphic_nested_where
|
2012-09-12 18:32:50 -04:00
|
|
|
thing = Post.new
|
|
|
|
thing.id = 1
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
expected = Treasure.where(price_estimates: { thing_type: 'Post', thing_id: 1 }).joins(:price_estimates)
|
|
|
|
actual = Treasure.where(price_estimates: { thing: thing }).joins(:price_estimates)
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
2012-09-11 14:11:51 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_polymorphic_sti_nested_where
|
2012-09-12 18:32:50 -04:00
|
|
|
treasure = HiddenTreasure.new
|
|
|
|
treasure.id = 1
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
expected = Treasure.where(price_estimates: { estimate_of_type: 'Treasure', estimate_of_id: 1 }).joins(:price_estimates)
|
|
|
|
actual = Treasure.where(price_estimates: { estimate_of: treasure }).joins(:price_estimates)
|
2012-09-11 14:11:51 -04:00
|
|
|
|
2012-09-12 18:32:50 -04:00
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
2012-09-11 14:11:51 -04:00
|
|
|
end
|
2013-08-20 07:21:23 -04:00
|
|
|
|
|
|
|
def test_decorated_polymorphic_where
|
|
|
|
treasure_decorator = Struct.new(:model) do
|
|
|
|
def self.method_missing(method, *args, &block)
|
|
|
|
Treasure.send(method, *args, &block)
|
|
|
|
end
|
|
|
|
|
|
|
|
def is_a?(klass)
|
|
|
|
model.is_a?(klass)
|
|
|
|
end
|
|
|
|
|
|
|
|
def method_missing(method, *args, &block)
|
|
|
|
model.send(method, *args, &block)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
treasure = Treasure.new
|
|
|
|
treasure.id = 1
|
|
|
|
decorated_treasure = treasure_decorator.new(treasure)
|
|
|
|
|
|
|
|
expected = PriceEstimate.where(estimate_of_type: 'Treasure', estimate_of_id: 1)
|
|
|
|
actual = PriceEstimate.where(estimate_of: decorated_treasure)
|
|
|
|
|
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
|
|
|
end
|
2012-05-30 18:09:13 -04:00
|
|
|
|
2013-05-01 14:33:11 -04:00
|
|
|
def test_aliased_attribute
|
|
|
|
expected = Topic.where(heading: 'The First Topic')
|
|
|
|
actual = Topic.where(title: 'The First Topic')
|
|
|
|
|
|
|
|
assert_equal expected.to_sql, actual.to_sql
|
|
|
|
end
|
|
|
|
|
2012-05-30 18:09:13 -04:00
|
|
|
def test_where_error
|
|
|
|
assert_raises(ActiveRecord::StatementInvalid) do
|
|
|
|
Post.where(:id => { 'posts.author_id' => 10 }).first
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_where_with_table_name
|
|
|
|
post = Post.first
|
|
|
|
assert_equal post, Post.where(:posts => { 'id' => post.id }).first
|
|
|
|
end
|
2012-07-05 07:24:59 -04:00
|
|
|
|
|
|
|
def test_where_with_table_name_and_empty_hash
|
2013-02-08 16:53:49 -05:00
|
|
|
assert_equal 0, Post.where(:posts => {}).count
|
2012-07-05 07:24:59 -04:00
|
|
|
end
|
|
|
|
|
2013-01-08 17:00:26 -05:00
|
|
|
def test_where_with_table_name_and_empty_array
|
2013-02-08 14:22:10 -05:00
|
|
|
assert_equal 0, Post.where(:id => []).count
|
2013-01-08 17:00:26 -05:00
|
|
|
end
|
|
|
|
|
2012-07-05 07:24:59 -04:00
|
|
|
def test_where_with_empty_hash_and_no_foreign_key
|
2013-02-08 16:53:49 -05:00
|
|
|
assert_equal 0, Edge.where(:sink => {}).count
|
2012-07-05 07:24:59 -04:00
|
|
|
end
|
2012-12-06 22:00:33 -05:00
|
|
|
|
|
|
|
def test_where_with_blank_conditions
|
|
|
|
[[], {}, nil, ""].each do |blank|
|
|
|
|
assert_equal 4, Edge.where(blank).order("sink_id").to_a.size
|
|
|
|
end
|
|
|
|
end
|
2012-05-30 18:09:13 -04:00
|
|
|
end
|
|
|
|
end
|