From 09dd29e4f775ef398bb9031dddebe6b84a7155b6 Mon Sep 17 00:00:00 2001
From: Sean Griffin <sean@thoughtbot.com>
Date: Wed, 11 Jun 2014 07:54:39 -0600
Subject: Ensure `column_types` returns a type object, and not a column

---
 activerecord/lib/active_record/model_schema.rb | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

(limited to 'activerecord')

diff --git a/activerecord/lib/active_record/model_schema.rb b/activerecord/lib/active_record/model_schema.rb
index f96f77f696..9e1afd32e6 100644
--- a/activerecord/lib/active_record/model_schema.rb
+++ b/activerecord/lib/active_record/model_schema.rb
@@ -220,25 +220,25 @@ module ActiveRecord
       end
 
       def column_types # :nodoc:
-        @column_types ||= decorate_columns(columns_hash.dup)
+        @column_types ||= decorate_types(build_types_hash)
       end
 
       def type_for_attribute(attr_name) # :nodoc:
-        column_types.fetch(attr_name) { column_for_attribute(attr_name) }
+        column_types.fetch(attr_name) { Type::Value.new }
       end
 
-      def decorate_columns(columns_hash) # :nodoc:
-        return if columns_hash.empty?
+      def decorate_types(types) # :nodoc:
+        return if types.empty?
 
         @time_zone_column_names ||= self.columns_hash.find_all do |name, col|
           create_time_zone_conversion_attribute?(name, col)
         end.map!(&:first)
 
         @time_zone_column_names.each do |name|
-          columns_hash[name] = AttributeMethods::TimeZoneConversion::Type.new(columns_hash[name])
+          types[name] = AttributeMethods::TimeZoneConversion::Type.new(types[name])
         end
 
-        columns_hash
+        types
       end
 
       # Returns a hash where the keys are column names and the values are
@@ -335,6 +335,10 @@ module ActiveRecord
           base.table_name
         end
       end
+
+      def build_types_hash
+        Hash[columns.map { |column| [column.name, column.cast_type] }]
+      end
     end
   end
 end
-- 
cgit v1.2.3