X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/8920a56133d24d328b774ef166a07949746e008a..f064a18a16de7d5001107116c7916796e0b15a26:/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