Creating the record through an association should work

Author: Kyle d'Oliveira

lib/polymorphic_integer_type/extensions.rb

@@ -17,7 +17,10 @@ def belongs_to(name, options = {})
           end
 
           define_method "#{foreign_type}=" do |klass|
-            enum = mapping.key(klass.to_s)
+            case klass
+            when Integer then enum = klass
+            when String then enum = mapping.key(klass)
+            end
             super(enum)
           end
 

spec/polymorphic_integer_type_spec.rb

@@ -33,6 +33,12 @@
     end
   end
 
+  context "When a link is created through an association" do
+    let(:link) { source.source_links.create }
+    let(:source) { cat }
+    include_examples "proper source"
+    
+  end
   context "When a link is given polymorphic record" do
     let(:link) { Link.create(:source => source) }
     let(:source) { cat }
@@ -116,5 +122,6 @@
 
 
   end
+  
 
 end