top and limit can be visited

rails · Jan 10, 2011 · b258fdd · b258fdd
1 parent f64d067
commit b258fdd
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/lib/arel/visitors/depth_first.rb b/lib/arel/visitors/depth_first.rb
@@ -18,9 +18,11 @@ def unary o
       alias :visit_Arel_Nodes_Group             :unary
       alias :visit_Arel_Nodes_Grouping          :unary
       alias :visit_Arel_Nodes_Having            :unary
+      alias :visit_Arel_Nodes_Limit             :unary
       alias :visit_Arel_Nodes_Not               :unary
       alias :visit_Arel_Nodes_Offset            :unary
       alias :visit_Arel_Nodes_On                :unary
+      alias :visit_Arel_Nodes_Top               :unary
       alias :visit_Arel_Nodes_UnqualifiedColumn :unary
 
       def function o

diff --git a/lib/arel/visitors/dot.rb b/lib/arel/visitors/dot.rb
@@ -78,9 +78,11 @@ def unary o
       alias :visit_Arel_Nodes_Group             :unary
       alias :visit_Arel_Nodes_Grouping          :unary
       alias :visit_Arel_Nodes_Having            :unary
+      alias :visit_Arel_Nodes_Limit             :unary
       alias :visit_Arel_Nodes_Not               :unary
       alias :visit_Arel_Nodes_Offset            :unary
       alias :visit_Arel_Nodes_On                :unary
+      alias :visit_Arel_Nodes_Top               :unary
       alias :visit_Arel_Nodes_UnqualifiedColumn :unary
 
       def visit_Arel_Nodes_InsertStatement o

diff --git a/test/visitors/test_depth_first.rb b/test/visitors/test_depth_first.rb
@@ -30,6 +30,8 @@ def test_raises_with_object
         Arel::Nodes::Offset,
         Arel::Nodes::Having,
         Arel::Nodes::UnqualifiedColumn,
+        Arel::Nodes::Top,
+        Arel::Nodes::Limit,
       ].each do |klass|
         define_method("test_#{klass.name.gsub('::', '_')}") do
           op = klass.new(:a)

diff --git a/test/visitors/test_dot.rb b/test/visitors/test_dot.rb
@@ -16,6 +16,8 @@ def setup
         Arel::Nodes::Offset,
         Arel::Nodes::Having,
         Arel::Nodes::UnqualifiedColumn,
+        Arel::Nodes::Top,
+        Arel::Nodes::Limit,
       ].each do |klass|
         define_method("test_#{klass.name.gsub('::', '_')}") do
           op = klass.new(:a)