X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/3123822bba5865e81e8c10be16452a0a8b5362f9..50c5f444ed0d20f4195320421b68dd3568519cc2:/test/factories/notes.rb

diff --git a/test/factories/notes.rb b/test/factories/notes.rb
index 52c23e7d6..ab7d5df56 100644
--- a/test/factories/notes.rb
+++ b/test/factories/notes.rb
@@ -1,7 +1,30 @@
-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
+      transient do
+        closed_by { nil }
+      end
+
+      status { "closed" }
+      closed_at { Time.now.utc }
+
+      after(:create) do |note, context|
+        create(:note_comment, :author => context.closed_by, :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