X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/8920a56133d24d328b774ef166a07949746e008a..49da87fb6d2a9fe8c033a41cd935ee34a8767293:/test/factories/messages.rb

diff --git a/test/factories/messages.rb b/test/factories/messages.rb
index 75e65cdbf..57a434fa1 100644
--- a/test/factories/messages.rb
+++ b/test/factories/messages.rb
@@ -1,21 +1,18 @@
-FactoryGirl.define do
+FactoryBot.define do
   factory :message do
     sequence(:title) { |n| "Message #{n}" }
     sequence(:body) { |n| "Body text for message #{n}" }
-    sent_on Time.now
+    sent_on { Time.now }
 
-    # FIXME: needs user factory
-    from_user_id 1
-
-    # FIXME: needs user factory
-    to_user_id 2
+    association :sender, :factory => :user
+    association :recipient, :factory => :user
 
     trait :unread do
-      message_read false
+      message_read { false }
     end
 
     trait :read do
-      message_read true
+      message_read { true }
     end
   end
 end