Merge pull request #23 from doliveirakn/support-sti-names

Use sti_name when writing the enum

Author: Kyle d'Oliveira

lib/polymorphic_integer_type/extensions.rb

@@ -31,7 +31,10 @@ def belongs_to(name, scope = nil, **options)
 
           define_method "#{foreign_type}=" do |klass|
             enum = mapping.key(klass.to_s)
-            enum ||= mapping.key(klass.base_class.to_s) if klass.kind_of?(Class) && klass <= ActiveRecord::Base
+            if klass.kind_of?(Class) && klass <= ActiveRecord::Base
+              enum ||= mapping.key(klass.base_class.to_s)
+              enum ||= mapping.key(klass.base_class.sti_name)
+            end
             enum ||= klass if klass != NilClass
             super(enum)
           end

spec/polymorphic_integer_type_spec.rb

@@ -24,6 +24,13 @@
     end
   end
 
+  context "when the source is a class that modifies the sti_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"
+    end
+  end
+
   context "when querying the associations" do
     let(:source) { cat }
     let(:target) { nil }

spec/spec_helper.rb

@@ -4,6 +4,7 @@
 require 'support/configuration'
 require 'support/link'
 require 'support/animal'
+require 'support/namespaced_animal'
 require 'support/dog'
 require 'support/person'
 require 'support/food'

spec/support/animal.rb

@@ -3,4 +3,5 @@ class Animal < ActiveRecord::Base
 
   belongs_to :owner, class_name: "Person"
   has_many :source_links, as: :source, integer_type: true, class_name: "Link"
-end
+
+end

spec/support/namespaced_animal.rb

@@ -0,0 +1,7 @@
+module Namespaced
+    class Animal < ActiveRecord::Base
+
+    self.store_full_sti_class = false
+    self.table_name = "animals"
+  end
+end