Merge pull request #37 from Drew-Goddyn/fix-type-to-ids-mapping-bug

fix bug with selecting correct class name from type mapping values

Author: Drew-Goddyn

lib/polymorphic_integer_type/activerecord_5_0_0/polymorphic_array_value_extension.rb

@@ -14,7 +14,7 @@ def type_to_ids_mapping
       name = association.name
       default_hash = Hash.new { |hsh, key| hsh[key] = [] }
       values.each_with_object(default_hash) do |value, hash|
-        klass = respond_to?(:klass) ? klass(value) : klass.class
+        klass = respond_to?(:klass, true) ? klass(value) : value.class
         if association.active_record.respond_to?("#{name}_type_mapping")
           mapping = association.active_record.send("#{name}_type_mapping")
           key ||= mapping.key(klass.polymorphic_name) if klass.respond_to?(:polymorphic_name)

lib/polymorphic_integer_type/belongs_to_polymorphic_association_extension.rb

@@ -3,7 +3,9 @@ module Associations
     class BelongsToPolymorphicAssociation < BelongsToAssociation
       private def replace_keys(record)
         super
-        owner[reflection.foreign_type] = record.class
+        unless record.nil?
+          owner[reflection.foreign_type] = record.class.base_class
+        end
       end
     end
   end

spec/polymorphic_integer_type_spec.rb

@@ -117,6 +117,15 @@
     it "properly finds the object with a find_by" do
       expect(Link.find_by(source: source, id: link.id)).to eql link
     end
+
+    context "when source and target are namespaced without modifying polymorphic_name" do
+      it "properly finds the object" do
+        plant = Namespaced::Plant.create(name: "Mighty", kind: "Oak", owner: owner)
+        activity = Namespaced::Activity.create(name: "swaying")
+        link = Link.create(source: plant, target: activity)
+        expect(Link.where(source: plant, id: link.id).first).to eql link
+      end
+    end
   end
 
   shared_examples "proper source" do