Add :closed trait to note factory

[rails.git] / test / factories / notes.rb

diff --git a/test/factories/notes.rb b/test/factories/notes.rb
index 52c23e7d69cd9ead70abf291e7522101f5710d18..63dcd78048c4d44e61ef9776de980f4512e4c814 100644 (file)
--- 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