polymorphic_integer_type 1.0.2 → 1.0.3

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    ZWZhZjhkYzJiNTgxNTBlZTUwYzYwOTY3YjE0MThlNzAyYWUxNTJlZA==
+    Y2I3YWE0OWMwZDNkZWM0MzkyZjY0NGQyNTJmMGUyZjdmY2UxYzIzNQ==
   data.tar.gz: !binary |-
-    M2JjYWIxZDU3Nzc3NjViMjI2MDc3ZTllNWRiMGNkZjEzNmUyNjQ0Zg==
+    OTFkNWRmNjhiMTkwNWJmZGVhMjFkZmJiZWU5ZDQzZTRmY2ZhZGRlNg==
 SHA512:
   metadata.gz: !binary |-
-    ODU0ZmI0NjhiYzM5NTcwYWM3NjU0YzIyYjg0ODJlMGMyOWMxNWYwNWMyZGMz
-    NWNjMzBhNjhjYzhiMmU1ZGU0YTUxNjhhZjU5ZDliNTQ5ZDZhYTgzMjg2MjU3
-    MzVmNzEwNTc5Y2IzYjI0YTE0NTM4NjYxYTY2MDk1M2RjMDI5NmE=
+    NGVkMWNlN2Y5MmRlNzNlMmM2NTkwMTEzNTcyNGNmYTMzOTdlYmRlYjU1YTAy
+    MDQyOWI4YTEyNmQ4MzIzNDFhMGIzZjAxMTNkMGI1MzMzY2ZkZDUzMzNhODg5
+    NmE0MTBkYWJkOGE5NTdmODEwY2YwMGFjMjkwY2JiNWNhZDk4MGI=
   data.tar.gz: !binary |-
-    NTE0NWY2MWJmNWRiMTE0MzdlZDcyOTcwOGI0ZGFlN2I3OTZjNmJkOTZlNWY2
-    MDE0NjYwMmViMzVhM2M1MmJkYWU1N2UxYzZkNmEzZWM0MGJiNjMzNmZmYTAz
-    MzQ5ODFiZmFjNTM1ZGVlYTRkYmIyYmEwYjAwZDA2MGZiMmNmNjA=
+    YzM1NDAzOTg5YjY5MjQyNGQ0MGU2ZDFhZDZjYWEyMmNjMjhjZTEyZGM4OTgw
+    N2NiZTk5Y2ZmODE0N2NjOGU2ZWJkY2JkMGRmMzc5ZWRhNzc0MjBhMTgzMjQy
+    Y2NlMzY5NDEzZGM0ZGNkNGYwYWQ4Y2U3Y2NlZTk1ZGQ3OTM1MjI=

data/lib/polymorphic_integer_type/extensions.rb

@@ -17,6 +17,7 @@ module PolymorphicIntegerType
           end
 
           define_method "#{foreign_type}=" do |klass|
+            klass = klass.base_class if klass.kind_of?(Class)
             enum = mapping.key(klass.to_s) || klass
             super(enum)
           end

data/lib/polymorphic_integer_type/version.rb

@@ -1,3 +1,3 @@
 module PolymorphicIntegerType
-  VERSION = "1.0.2"
+  VERSION = "1.0.3"
 end

data/spec/polymorphic_integer_type_spec.rb

@@ -131,5 +131,15 @@ describe PolymorphicIntegerType do
 
   end
   
+  context "when the association is an STI table" do
+    let(:link) { Link.create(:source => source, :target => whiskey) }
+    let(:source) { Dog.create(:name => "Bela", :kind => "Dog", :owner => owner) }
+    it "should have the proper id, type and object for the source" do
+      expect(link.source_id).to eql source.id
+      expect(link.source_type).to eql "Animal"
+      expect(link.source).to eql source
+    end
+  end
+  
 
 end

data/spec/spec_helper.rb

@@ -3,6 +3,7 @@ require 'polymorphic_integer_type'
 require 'support/configuration'
 require 'support/link'
 require 'support/animal'
+require 'support/dog'
 require 'support/person'
 require 'support/food'
 require 'support/drink'

data/spec/support/dog.rb

@@ -0,0 +1,4 @@
+class Dog < Animal
+
+
+end

data/spec/support/migrations/2_create_animal_table.rb

@@ -3,6 +3,7 @@ class CreateAnimalTable < ActiveRecord::Migration
   def up
     create_table :animals do |t|
       t.string :name
+      t.string :type
       t.string :kind
       t.integer :owner_id
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polymorphic_integer_type
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - Kyle d'Oliveira
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-02-25 00:00:00.000000000 Z
+date: 2014-07-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -116,6 +116,7 @@ files:
 - spec/support/active_record.rb
 - spec/support/animal.rb
 - spec/support/configuration.rb
+- spec/support/dog.rb
 - spec/support/drink.rb
 - spec/support/food.rb
 - spec/support/link.rb
@@ -145,7 +146,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.1.11
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Use integers rather than strings for the _type field
@@ -155,6 +156,7 @@ test_files:
 - spec/support/active_record.rb
 - spec/support/animal.rb
 - spec/support/configuration.rb
+- spec/support/dog.rb
 - spec/support/drink.rb
 - spec/support/food.rb
 - spec/support/link.rb