From 21d9c0fc7f79f486fb2c67b3593e9496a1c57663 Mon Sep 17 00:00:00 2001
From: kennyj <kennyj@gmail.com>
Date: Wed, 14 Mar 2012 02:34:12 +0900
Subject: [PATCH] [3-2-stable] Fix GH #5399. connection_pools's keys are
 ActiveRecord::Base::ConnectionSpecification objects.

---
 activerecord/lib/active_record/base.rb      | 2 +-
 activerecord/test/cases/multiple_db_test.rb | 8 +++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/activerecord/lib/active_record/base.rb b/activerecord/lib/active_record/base.rb
index 425cd8cbab2d0..9746efc7d14fb 100644
--- a/activerecord/lib/active_record/base.rb
+++ b/activerecord/lib/active_record/base.rb
@@ -442,7 +442,7 @@ def arel_engine
           if self == ActiveRecord::Base
             ActiveRecord::Base
           else
-            connection_handler.connection_pools[name] ? self : superclass.arel_engine
+            connection_handler.retrieve_connection_pool(self) ? self : superclass.arel_engine
           end
         end
       end
diff --git a/activerecord/test/cases/multiple_db_test.rb b/activerecord/test/cases/multiple_db_test.rb
index e704322b5d4c3..b31d971309063 100644
--- a/activerecord/test/cases/multiple_db_test.rb
+++ b/activerecord/test/cases/multiple_db_test.rb
@@ -85,6 +85,12 @@ def test_transactions_across_databases
   end
 
   def test_arel_table_engines
-    assert_equal Entrant.arel_engine, Bird.arel_engine
+    assert_not_equal Entrant.arel_engine, Bird.arel_engine
+    assert_not_equal Entrant.arel_engine, Course.arel_engine
+  end
+
+  def test_connection
+    assert_equal Entrant.arel_engine.connection, Bird.arel_engine.connection
+    assert_not_equal Entrant.arel_engine.connection, Course.arel_engine.connection
   end
 end