Ensure that polymorphic associations does not require a klass to exis…

…ts [#84 status:resolved]
pixeltrix · Jun 5, 2009 · f824ea5 · f824ea5
1 parent cfbf8fb
commit f824ea5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/remarkable_activerecord/lib/remarkable_activerecord/matchers/association_matcher.rb b/remarkable_activerecord/lib/remarkable_activerecord/matchers/association_matcher.rb
@@ -32,7 +32,10 @@ def source_exists?
             reflection.source_reflection rescue false
           end
 
+          # Polymorphic associations does not have a klass.
+          #
           def klass_exists?
+            return true if @options[:polymorphic]
             reflection.klass rescue nil
           end
 

diff --git a/remarkable_activerecord/spec/association_matcher_spec.rb b/remarkable_activerecord/spec/association_matcher_spec.rb
@@ -15,6 +15,7 @@ def define_and_validate(options={})
       @model = define_model :project, columns do
         belongs_to :company, options
         belongs_to :unknown
+        belongs_to :accountable, :polymorphic => true
       end
 
       belong_to :company
@@ -145,6 +146,11 @@ def define_and_validate(options={})
         it { should_not define_and_validate(:association_columns => { :association_count => :integer }).counter_cache(:association_count) }
       end
 
+      describe "with polymorphic option" do
+        before(:each){ define_and_validate(:model_columns => {:accountable_id => :integer, :accountable_type => :string}) }
+        it { should belong_to(:accountable).polymorphic }
+      end
+
       create_optional_boolean_specs(:readonly, self)
       create_optional_boolean_specs(:validate, self)
       create_optional_boolean_specs(:autosave, self) if RAILS_VERSION =~ /^2.3/