Fix bug with reading dirty attributes on namespaced classes

When a class' name does not match its STI name, ActiveRecord
will reload an associated record from the database even if it's
already in memory. This bug exists in Rails 4 and above.

Author: mctaylorpants

lib/polymorphic_integer_type.rb

@@ -3,6 +3,7 @@
 require "polymorphic_integer_type/version"
 require "polymorphic_integer_type/extensions"
 require "polymorphic_integer_type/mapping"
+require "polymorphic_integer_type/belongs_to_polymorphic_association_extension"
 
 if ACTIVE_RECORD_VERSION < Gem::Version.new("5")
   require "polymorphic_integer_type/activerecord_4/predicate_builder_extension"
@@ -14,4 +15,5 @@
   require "polymorphic_integer_type/activerecord_5_0_0/association_query_handler_extension"
 end
 
+
 module PolymorphicIntegerType; end

lib/polymorphic_integer_type/belongs_to_polymorphic_association_extension.rb

@@ -0,0 +1,10 @@
+module ActiveRecord
+  module Associations
+    class BelongsToPolymorphicAssociation < BelongsToAssociation
+      private def replace_keys(record)
+        super
+        owner[reflection.foreign_type] = record.class.base_class
+      end
+    end
+  end
+end

spec/polymorphic_integer_type_spec.rb

@@ -29,6 +29,14 @@
       link = Link.new(source: Namespaced::Animal.new)
       expect(link.source_type).to eql "Animal"
     end
+
+    it "can read dirty attributes from an associated object" do
+      animal = Namespaced::Animal.create!(name: "Oldie")
+      animal.name = "Newton"
+      link = Link.create!(source: animal)
+
+      expect(link.source.name).to eq("Newton")
+    end
   end
 
   context "when querying the associations" do
@@ -109,7 +117,7 @@
 
   end
 
-  context "When using a relation to the links with eagar loading" do
+  context "When using a relation to the links with eager loading" do
     let!(:links){
       [Link.create(source: source, target: kibble),
         Link.create(source: source, target: water)]
@@ -124,7 +132,7 @@
 
   end
 
-  context "When using a through relation to the links with eagar loading" do
+  context "When using a through relation to the links with eager loading" do
     let!(:links){
       [Link.create(source: source, target: kibble),
         Link.create(source: source, target: water)]
@@ -139,7 +147,7 @@
 
   end
 
-  context "When eagar loading the polymorphic association" do
+  context "When eager loading the polymorphic association" do
     let(:link) { Link.create(source_id: source.id, source_type: source.class.to_s) }
     let(:source) { cat }
 
@@ -160,6 +168,7 @@
     end
 
 
+
   end
 
   context "when the association is an STI table" do