Adds visit_Arel_Nodes_InfixOperation to Arel::Visitors::DepthFirst

ratbeard · Mar 22, 2012 · c50b406 · c50b406
1 parent 660768f
commit c50b406
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/lib/arel/visitors/depth_first.rb b/lib/arel/visitors/depth_first.rb
@@ -70,6 +70,7 @@ def binary o
       alias :visit_Arel_Nodes_GreaterThan        :binary
       alias :visit_Arel_Nodes_GreaterThanOrEqual :binary
       alias :visit_Arel_Nodes_In                 :binary
+      alias :visit_Arel_Nodes_InfixOperation     :binary
       alias :visit_Arel_Nodes_JoinSource         :binary
       alias :visit_Arel_Nodes_InnerJoin          :binary
       alias :visit_Arel_Nodes_LessThan           :binary

diff --git a/test/visitors/test_depth_first.rb b/test/visitors/test_depth_first.rb
@@ -114,6 +114,12 @@ def test_outer_join
         end
       end
 
+      def test_Arel_Nodes_InfixOperation
+        binary = Arel::Nodes::InfixOperation.new(:o, :a, :b)
+        @visitor.accept binary
+        assert_equal [:a, :b, binary], @collector.calls
+      end
+
       # N-ary
       [
         Arel::Nodes::And,