Need to add a []= method too

Author: Kyle d'Oliveira

lib/polymorphic_integer_type/extensions.rb

@@ -82,6 +82,14 @@ def [](value)
       end
     end
 
+    def []=(attr_name, value)
+      if _polymorphic_foreign_types.include?(attr_name)
+        send("#{attr_name}=", value)
+      else
+        super(attr_name, value)
+      end
+    end
+
   end
 
 end

spec/polymorphic_integer_type_spec.rb

@@ -37,6 +37,14 @@
     let(:link) { source.source_links.create }
     let(:source) { cat }
     include_examples "proper source"
+
+    context "and the link is accessed through the associations" do
+      before { link }
+      
+      it "should have the proper source" do
+        expect(source.source_links[0].source).to eql source
+      end
+    end
 
   end
   context "When a link is given polymorphic record" do