X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/e5604ce98e267cb0cec7dd27078a9e86c919eeb1..0dc86d6cd47faa42ff6b538bd0342a1486e313b2:/test/factories/issues.rb

diff --git a/test/factories/issues.rb b/test/factories/issues.rb
index e209bd25d..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"
+    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