X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/2e1ccafdce30a06fab7a72e5c856ce8a6c23bfdd..6272b041c6dd9c39cc7a23d6163651e74509da2b:/test/factories/notes.rb

diff --git a/test/factories/notes.rb b/test/factories/notes.rb
index 52c23e7d6..63dcd7804 100644
--- a/test/factories/notes.rb
+++ b/test/factories/notes.rb
@@ -1,7 +1,26 @@
-FactoryGirl.define do
+FactoryBot.define do
   factory :note do
-    latitude 1 * GeoRecord::SCALE
-    longitude 1 * GeoRecord::SCALE
-    # tile QuadTile.tile_for_point(1,1)
+    latitude { 1 * GeoRecord::SCALE }
+    longitude { 1 * GeoRecord::SCALE }
+    # tile { QuadTile.tile_for_point(1,1) }
+
+    trait :closed do
+      status { "closed" }
+      closed_at { Time.now.utc }
+
+      after(:create) do |note|
+        create(:note_comment, :body => "Closing comment", :event => "closed", :note => note)
+      end
+    end
+
+    factory :note_with_comments do
+      transient do
+        comments_count { 1 }
+      end
+
+      after(:create) do |note, evaluator|
+        create_list(:note_comment, evaluator.comments_count, :note => note)
+      end
+    end
   end
 end