test/visitors/test_mssql.rb in arel-2.0.10 vs test/visitors/test_mssql.rb in arel-2.1.0

@@ -11,8 +11,17 @@
         stmt = Nodes::SelectStatement.new
         stmt.cores.last.top = Nodes::Top.new(1)
         sql = @visitor.accept(stmt)
         sql.must_be_like "SELECT TOP 1"
       end
+
+      it 'uses TOP in updates with a limit' do
+        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 TOP 1 'id' )"
+      end
+
     end
   end
 end