X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/a280f72beead569784c6bf353131d10706dfab00..899fc66a62b036adcfd59f5da80c7fc6f30cd3a8:/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