Support Rails 5.2 with #polymorphic_name

Author: mctaylorpants

lib/polymorphic_integer_type.rb

@@ -3,7 +3,6 @@
 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"
@@ -15,5 +14,8 @@
   require "polymorphic_integer_type/activerecord_5_0_0/association_query_handler_extension"
 end
 
+if ACTIVE_RECORD_VERSION < Gem::Version.new("5.2.0")
+  require "polymorphic_integer_type/activerecord_4/belongs_to_polymorphic_association_extension"
+end
 
 module PolymorphicIntegerType; end

…_to_polymorphic_association_extension.rb …_to_polymorphic_association_extension.rblib/polymorphic_integer_type/belongs_to_polymorphic_association_extension.rb renamed to lib/polymorphic_integer_type/activerecord_4/belongs_to_polymorphic_association_extension.rb lib/polymorphic_integer_type/belongs_to_polymorphic_association_extension.rb renamed to lib/polymorphic_integer_type/activerecord_4/belongs_to_polymorphic_association_extension.rb

@@ -24,7 +24,7 @@
     end
   end
 
-  context "when the source is a class that modifies the sti_name" do
+  context "when the source is a class that modifies the sti_name or polymorphic_name" do
     it "properly sets the source_type to the modified class name" do
       link = Link.new(source: Namespaced::Animal.new)
       expect(link.source_type).to eql "Animal"

spec/polymorphic_integer_type_spec.rb

@@ -3,5 +3,9 @@ class Animal < ActiveRecord::Base
 
     self.store_full_sti_class = false
     self.table_name = "animals"
+
+    def self.polymorphic_name
+      "Animal"
+    end
   end
-end
+end