From 7c48c8360e1b4e90f2a0c5ef15590355c4bf0856 Mon Sep 17 00:00:00 2001 From: Aaron Patterson Date: Tue, 14 Dec 2010 10:38:55 -0800 Subject: select core should visit the source node --- lib/arel/visitors/depth_first.rb | 3 ++- test/visitors/test_depth_first.rb | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/lib/arel/visitors/depth_first.rb b/lib/arel/visitors/depth_first.rb index 9979c8313e..5ebd1296dd 100644 --- a/lib/arel/visitors/depth_first.rb +++ b/lib/arel/visitors/depth_first.rb @@ -57,6 +57,7 @@ module Arel alias :visit_Arel_Nodes_GreaterThan :binary alias :visit_Arel_Nodes_GreaterThanOrEqual :binary alias :visit_Arel_Nodes_In :binary + alias :visit_Arel_Nodes_JoinSource :binary alias :visit_Arel_Nodes_InnerJoin :binary alias :visit_Arel_Nodes_LessThan :binary alias :visit_Arel_Nodes_LessThanOrEqual :binary @@ -116,7 +117,7 @@ module Arel def visit_Arel_Nodes_SelectCore o visit o.projections - visit o.froms + visit o.source visit o.wheres visit o.groups visit o.having diff --git a/test/visitors/test_depth_first.rb b/test/visitors/test_depth_first.rb index 23b011c2eb..34931c783e 100644 --- a/test/visitors/test_depth_first.rb +++ b/test/visitors/test_depth_first.rb @@ -97,6 +97,7 @@ module Arel Arel::Nodes::As, Arel::Nodes::DeleteStatement, Arel::Nodes::Ordering, + Arel::Nodes::JoinSource, ].each do |klass| define_method("test_#{klass.name.gsub('::', '_')}") do binary = klass.new(:a, :b) -- cgit v1.2.3