X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/083500f056f5b995f1979f88b37793046d9e4d37..735a5cf9e25b81d03db80ca4393aa34ea74d612a:/test/factories/issues.rb?ds=sidebyside

diff --git a/test/factories/issues.rb b/test/factories/issues.rb
index c575c3398..cc6b4c367 100644
--- 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