use ARel factory methods for building AST nodes

This abstracts us from the actual construction of the nodes
This commit is contained in:
Aaron Patterson 2014-03-24 17:20:55 -07:00
parent e2097235b6
commit c7191e3195
2 changed files with 17 additions and 17 deletions

@ -5,10 +5,10 @@ module ActiveRecord
class PredicateBuilderTest < ActiveRecord::TestCase class PredicateBuilderTest < ActiveRecord::TestCase
def test_registering_new_handlers def test_registering_new_handlers
PredicateBuilder.register_handler(Regexp, proc do |column, value| PredicateBuilder.register_handler(Regexp, proc do |column, value|
Arel::Nodes::InfixOperation.new('~', column, value.source) Arel::Nodes::InfixOperation.new('~', column, Arel.sql(value.source))
end) end)
assert_match %r{["`]topics["`].["`]title["`] ~ 'rails'}i, Topic.where(title: /rails/).to_sql assert_match %r{["`]topics["`].["`]title["`] ~ rails}i, Topic.where(title: /rails/).to_sql
end end
end end
end end

@ -12,13 +12,13 @@ def setup
end end
def test_not_eq def test_not_eq
expected = Arel::Nodes::NotEqual.new(Post.arel_table[@name], 'hello') expected = Post.arel_table[@name].not_eq('hello')
relation = Post.where.not(title: 'hello') relation = Post.where.not(title: 'hello')
assert_equal([expected], relation.where_values) assert_equal([expected], relation.where_values)
end end
def test_not_null def test_not_null
expected = Arel::Nodes::NotEqual.new(Post.arel_table[@name], nil) expected = Post.arel_table[@name].not_eq(nil)
relation = Post.where.not(title: nil) relation = Post.where.not(title: nil)
assert_equal([expected], relation.where_values) assert_equal([expected], relation.where_values)
end end
@ -30,13 +30,13 @@ def test_not_with_nil
end end
def test_not_in def test_not_in
expected = Arel::Nodes::NotIn.new(Post.arel_table[@name], %w[hello goodbye]) expected = Post.arel_table[@name].not_in(%w[hello goodbye])
relation = Post.where.not(title: %w[hello goodbye]) relation = Post.where.not(title: %w[hello goodbye])
assert_equal([expected], relation.where_values) assert_equal([expected], relation.where_values)
end end
def test_association_not_eq def test_association_not_eq
expected = Arel::Nodes::NotEqual.new(Comment.arel_table[@name], 'hello') expected = Comment.arel_table[@name].not_eq('hello')
relation = Post.joins(:comments).where.not(comments: {title: 'hello'}) relation = Post.joins(:comments).where.not(comments: {title: 'hello'})
assert_equal(expected.to_sql, relation.where_values.first.to_sql) assert_equal(expected.to_sql, relation.where_values.first.to_sql)
end end
@ -44,20 +44,20 @@ def test_association_not_eq
def test_not_eq_with_preceding_where def test_not_eq_with_preceding_where
relation = Post.where(title: 'hello').where.not(title: 'world') relation = Post.where(title: 'hello').where.not(title: 'world')
expected = Arel::Nodes::Equality.new(Post.arel_table[@name], 'hello') expected = Post.arel_table[@name].eq('hello')
assert_equal(expected, relation.where_values.first) assert_equal(expected, relation.where_values.first)
expected = Arel::Nodes::NotEqual.new(Post.arel_table[@name], 'world') expected = Post.arel_table[@name].not_eq('world')
assert_equal(expected, relation.where_values.last) assert_equal(expected, relation.where_values.last)
end end
def test_not_eq_with_succeeding_where def test_not_eq_with_succeeding_where
relation = Post.where.not(title: 'hello').where(title: 'world') relation = Post.where.not(title: 'hello').where(title: 'world')
expected = Arel::Nodes::NotEqual.new(Post.arel_table[@name], 'hello') expected = Post.arel_table[@name].not_eq('hello')
assert_equal(expected, relation.where_values.first) assert_equal(expected, relation.where_values.first)
expected = Arel::Nodes::Equality.new(Post.arel_table[@name], 'world') expected = Post.arel_table[@name].eq('world')
assert_equal(expected, relation.where_values.last) assert_equal(expected, relation.where_values.last)
end end
@ -76,17 +76,17 @@ def test_not_eq_with_array_parameter
def test_chaining_multiple def test_chaining_multiple
relation = Post.where.not(author_id: [1, 2]).where.not(title: 'ruby on rails') relation = Post.where.not(author_id: [1, 2]).where.not(title: 'ruby on rails')
expected = Arel::Nodes::NotIn.new(Post.arel_table['author_id'], [1, 2]) expected = Post.arel_table['author_id'].not_in([1, 2])
assert_equal(expected, relation.where_values[0]) assert_equal(expected, relation.where_values[0])
expected = Arel::Nodes::NotEqual.new(Post.arel_table[@name], 'ruby on rails') expected = Post.arel_table[@name].not_eq('ruby on rails')
assert_equal(expected, relation.where_values[1]) assert_equal(expected, relation.where_values[1])
end end
def test_rewhere_with_one_condition def test_rewhere_with_one_condition
relation = Post.where(title: 'hello').where(title: 'world').rewhere(title: 'alone') relation = Post.where(title: 'hello').where(title: 'world').rewhere(title: 'alone')
expected = Arel::Nodes::Equality.new(Post.arel_table[@name], 'alone') expected = Post.arel_table[@name].eq('alone')
assert_equal 1, relation.where_values.size assert_equal 1, relation.where_values.size
assert_equal expected, relation.where_values.first assert_equal expected, relation.where_values.first
end end
@ -94,8 +94,8 @@ def test_rewhere_with_one_condition
def test_rewhere_with_multiple_overwriting_conditions def test_rewhere_with_multiple_overwriting_conditions
relation = Post.where(title: 'hello').where(body: 'world').rewhere(title: 'alone', body: 'again') relation = Post.where(title: 'hello').where(body: 'world').rewhere(title: 'alone', body: 'again')
title_expected = Arel::Nodes::Equality.new(Post.arel_table['title'], 'alone') title_expected = Post.arel_table['title'].eq('alone')
body_expected = Arel::Nodes::Equality.new(Post.arel_table['body'], 'again') body_expected = Post.arel_table['body'].eq('again')
assert_equal 2, relation.where_values.size assert_equal 2, relation.where_values.size
assert_equal title_expected, relation.where_values.first assert_equal title_expected, relation.where_values.first
@ -105,8 +105,8 @@ def test_rewhere_with_multiple_overwriting_conditions
def test_rewhere_with_one_overwriting_condition_and_one_unrelated def test_rewhere_with_one_overwriting_condition_and_one_unrelated
relation = Post.where(title: 'hello').where(body: 'world').rewhere(title: 'alone') relation = Post.where(title: 'hello').where(body: 'world').rewhere(title: 'alone')
title_expected = Arel::Nodes::Equality.new(Post.arel_table['title'], 'alone') title_expected = Post.arel_table['title'].eq('alone')
body_expected = Arel::Nodes::Equality.new(Post.arel_table['body'], 'world') body_expected = Post.arel_table['body'].eq('world')
assert_equal 2, relation.where_values.size assert_equal 2, relation.where_values.size
assert_equal body_expected, relation.where_values.first assert_equal body_expected, relation.where_values.first