Merge pull request #4300 from tomhughes/relation-members-primary-key

[rails.git] / test / factories / issues.rb

diff --git a/test/factories/issues.rb b/test/factories/issues.rb
index c575c3398e3fe16108329098316eb2430cb822d4..cc6b4c3671576cd241f07732939068cf730107eb 100644 (file)
--- a/test/factories/issues.rb
+++ b/test/factories/issues.rb
@@ -1,10 +1,21 @@
 FactoryBot.define do
   factory :issue do
     # Default to reporting users
-    association :reportable, :factory => :user
-    association :reported_user, :factory => :user
+    reportable :factory => :user
+    reported_user :factory => :user
 
     # Default to assigning to an administrator
     assigned_role { "administrator" }
+
+    # Optionally create some reports for this issue
+    factory :issue_with_reports do
+      transient do
+        reports_count { 1 }
+      end
+
+      after(:create) do |issue, evaluator|
+        create_list(:report, evaluator.reports_count, :issue => issue)
+      end
+    end
   end
 end