adding select statement support

2 files changed, 27 insertions, 0 deletions

diff --git a/lib/arel/visitors/depth_first.rb b/lib/arel/visitors/depth_first.rb
index 5c89ef96b3..534f9edaf5 100644
--- a/lib/arel/visitors/depth_first.rb
+++ b/lib/arel/visitors/depth_first.rb
@@ -70,6 +70,15 @@ module Arel
         @block.call o
       end
 
+      def visit_Arel_Nodes_SelectStatement o
+        visit o.cores
+        visit o.orders
+        visit o.limit
+        visit o.lock
+        visit o.offset
+        @block.call o
+      end
+
       def visit_Arel_Nodes_UpdateStatement o
         visit o.relation
         visit o.values
diff --git a/test/visitors/test_depth_first.rb b/test/visitors/test_depth_first.rb
index a318583e26..04c5096f5b 100644
--- a/test/visitors/test_depth_first.rb
+++ b/test/visitors/test_depth_first.rb
@@ -102,6 +102,24 @@ module Arel
           :e,
           core], @collector.calls
       end
+
+      def test_select_statement
+        ss = Nodes::SelectStatement.new
+        ss.cores.replace [:a]
+        ss.orders << :b
+        ss.limit = :c
+        ss.lock = :d
+        ss.offset = :e
+
+        @visitor.accept ss
+        assert_equal [
+          :a, ss.cores,
+          :b, ss.orders,
+          :c,
+          :d,
+          :e,
+          ss], @collector.calls
+      end
     end
   end
 end