X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/783b5e3729228908d7404ae7404af1023501a906..86fae8980d97ba7fdd9345eb5e372fd4f8f20b7f:/test/factories/issues.rb

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