diff options
-rw-r--r-- | lib/arel/visitors/to_sql.rb | 6 | ||||
-rw-r--r-- | test/support/fake_record.rb | 6 |
2 files changed, 8 insertions, 4 deletions
diff --git a/lib/arel/visitors/to_sql.rb b/lib/arel/visitors/to_sql.rb index 050a9b2e63..d0a05fac7e 100644 --- a/lib/arel/visitors/to_sql.rb +++ b/lib/arel/visitors/to_sql.rb @@ -147,11 +147,11 @@ key on UpdateManager using UpdateManager#key= return nil unless table_exists? table - column_cache[table][name] + column_cache(table)[name] end - def column_cache - @schema_cache.columns_hash + def column_cache(table) + @schema_cache.columns_hash(table) end def visit_Arel_Nodes_Values o 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 |