blob: b47c37db736be6786eab7276f290d5f458aa3782 (
plain) (
tree)
|
|
# frozen_string_literal: true
require 'helper'
require 'arel/collectors/bind'
require 'arel/collectors/composite'
require 'arel/collectors/sql_string'
module Arel
module Collectors
class TestComposite < Arel::Test
def setup
@conn = FakeRecord::Base.new
@visitor = Visitors::ToSql.new @conn.connection
super
end
def collect node
sql_collector = Collectors::SQLString.new
bind_collector = Collectors::Bind.new
collector = Collectors::Composite.new(sql_collector, bind_collector)
@visitor.accept(node, collector)
end
def compile node
collect(node).value
end
def ast_with_binds bvs
table = Table.new(:users)
manager = Arel::SelectManager.new table
manager.where(table[:age].eq(Nodes::BindParam.new(bvs.shift)))
manager.where(table[:name].eq(Nodes::BindParam.new(bvs.shift)))
manager.ast
end
def test_composite_collector_performs_multiple_collections_at_once
sql, binds = compile(ast_with_binds(["hello", "world"]))
assert_equal 'SELECT FROM "users" WHERE "users"."age" = ? AND "users"."name" = ?', sql
assert_equal ["hello", "world"], binds
sql, binds = compile(ast_with_binds(["hello2", "world3"]))
assert_equal 'SELECT FROM "users" WHERE "users"."age" = ? AND "users"."name" = ?', sql
assert_equal ["hello2", "world3"], binds
end
end
end
end
|