2 files changed, 16 insertions, 1 deletions
diff --git a/test/support/fake_record.rb b/test/support/fake_record.rb
index 376ed40f2d..54f73489c9 100644
--- a/test/support/fake_record.rb
+++ b/test/support/fake_record.rb
@@ -3,7 +3,7 @@ module FakeRecord
   end
 
   class Connection
-    attr_reader :tables
+    attr_reader :tables, :columns_hash
 
     def initialize
       @tables = %w{ users photos developers }
@@ -15,6 +15,9 @@ module FakeRecord
           Column.new('created_at', :date),
         ]
       }
+      @columns_hash = {
+        'users' => Hash[@columns['users'].map { |x| [x.name, x] }]
+      }
       @primary_keys = {
         'users' => 'id'
       }
@@ -75,6 +78,14 @@ module FakeRecord
     def with_connection
       yield connection
     end
+
+    def table_exists? name
+      connection.tables.include? name.to_s
+    end
+
+    def columns_hash
+      connection.columns_hash
+    end
   end
 
   class Base
diff --git a/test/test_select_manager.rb b/test/test_select_manager.rb
index 3ffdbe2e23..3b5ded8389 100644
--- a/test/test_select_manager.rb
+++ b/test/test_select_manager.rb
@@ -30,6 +30,10 @@ module Arel
       @engine.connection.columns table, message
     end
 
+    def columns_hash
+      @engine.connection.columns_hash
+    end
+
     def table_exists? name
       @engine.connection.table_exists? name
     end