Add belongs_to optional true to setup_association

xxx44yyy · Feb 20, 2018 · fde9d04 · fde9d04
1 parent 9da66fe
commit fde9d04
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/lib/impressionist/setup_association.rb b/lib/impressionist/setup_association.rb
@@ -12,7 +12,11 @@ def include_attr_acc?
     end
 
     def define_belongs_to
-      receiver.belongs_to(:impressionable, :polymorphic => true)
+      if ::Rails::VERSION::MAJOR.to_i >= 5
+        receiver.belongs_to(:impressionable, :polymorphic => true, :optional => true)	         
+      else
+        receiver.belongs_to(:impressionable, :polymorphic => true)
+      end
     end
 
     def set

diff --git a/tests/spec/minitest_helper.rb b/tests/spec/minitest_helper.rb
@@ -1,4 +1,7 @@
 $:.unshift(File.dirname __FILE__)
 
+require 'rails/all'
+require 'rails/test_help'
 require "minitest/autorun"
 require "minitest/pride"
+require 'minitest/rails'