Make it the responsibility of the connection to hold on to a visitor for generating SQL, rather than the TreeManager. (There is a related commit coming in Active Record.)

1 files changed, 3 insertions, 3 deletions

diff --git a/test/visitors/test_to_sql.rb b/test/visitors/test_to_sql.rb
index 3b58c71cd8..12af197596 100644
--- a/test/visitors/test_to_sql.rb
+++ b/test/visitors/test_to_sql.rb
@@ -8,7 +8,7 @@ module Arel
   module Visitors
     describe 'the to_sql visitor' do
       before do
-        @visitor = ToSql.new Table.engine
+        @visitor = ToSql.new Table.engine.connection_pool
         @table = Table.new(:users)
         @attr = @table[:id]
       end
@@ -220,7 +220,7 @@ module Arel
             end
           end
           in_node = Nodes::In.new @attr, %w{ a b c }
-          visitor = visitor.new(Table.engine)
+          visitor = visitor.new(Table.engine.connection_pool)
           visitor.expected = Table.engine.connection.columns(:users).find { |x|
             x.name == 'name'
           }
@@ -308,7 +308,7 @@ module Arel
             end
           end
           in_node = Nodes::NotIn.new @attr, %w{ a b c }
-          visitor = visitor.new(Table.engine)
+          visitor = visitor.new(Table.engine.connection_pool)
           visitor.expected = Table.engine.connection.columns(:users).find { |x|
             x.name == 'name'
           }