test/visitors/test_ibm_db.rb in arel-5.0.1.20140414130214 vs test/visitors/test_ibm_db.rb in arel-6.0.0.beta1

@@ -5,22 +5,28 @@
     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 = @visitor.accept(stmt)
+        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.limit = Nodes::Limit.new(1)
-        stmt.key = 'id'
-        sql = @visitor.accept(stmt)
-        sql.must_be_like "UPDATE NULL WHERE 'id' IN (SELECT 'id' FETCH FIRST 1 ROWS ONLY)"
+        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