1 files changed, 5 insertions, 1 deletions
diff --git a/test/support/fake_record.rb b/test/support/fake_record.rb
index 79182f86bd..58883e43cd 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, :columns_hash
+    attr_reader :tables
     attr_accessor :visitor
 
     def initialize(visitor = nil)
@@ -31,6 +31,10 @@ module FakeRecord
       @visitor = visitor
     end
 
+    def columns_hash table_name
+      @columns_hash[table_name]
+    end
+
     def primary_key name
       @primary_keys[name.to_s]
     end