diff options
Diffstat (limited to 'test/support/fake_record.rb')
-rw-r--r-- | test/support/fake_record.rb | 6 |
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 |