forked from rails/arel
/
test_ibm_db.rb
32 lines (28 loc) · 948 Bytes
/
test_ibm_db.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
require 'helper'
module Arel
module Visitors
describe 'the ibm_db visitor' do
before do
@visitor = IBM_DB.new Table.engine.connection
end
def compile node
@visitor.accept(node, Collectors::SQLString.new).value
end
it 'uses FETCH FIRST n ROWS to limit results' do
stmt = Nodes::SelectStatement.new
stmt.limit = Nodes::Limit.new(1)
sql = compile(stmt)
sql.must_be_like "SELECT FETCH FIRST 1 ROWS ONLY"
end
it 'uses FETCH FIRST n ROWS in updates with a limit' do
table = Table.new(:users)
stmt = Nodes::UpdateStatement.new
stmt.relation = table
stmt.limit = Nodes::Limit.new(Nodes.build_quoted(1))
stmt.key = table[:id]
sql = compile(stmt)
sql.must_be_like "UPDATE \"users\" WHERE \"users\".\"id\" IN (SELECT \"users\".\"id\" FROM \"users\" FETCH FIRST 1 ROWS ONLY)"
end
end
end
end