Add test for FanOutOnWriteService

1 files changed, 9 insertions, 1 deletions

diff --git a/app/services/fan_out_on_write_service.rb b/app/services/fan_out_on_write_service.rb
index a36f80150..674665592 100644
--- a/app/services/fan_out_on_write_service.rb
+++ b/app/services/fan_out_on_write_service.rb
@@ -15,10 +15,13 @@ class FanOutOnWriteService < BaseService
   private
 
   def deliver_to_self(status)
+    Rails.logger.debug "Delivering status #{status.id} to author"
     FeedManager.instance.push(:home, status.account, status)
   end
 
   def deliver_to_followers(status)
+    Rails.logger.debug "Delivering status #{status.id} to followers"
+
     status.account.followers.find_each do |follower|
       next if !follower.local? || FeedManager.instance.filter?(:home, status, follower)
       FeedManager.instance.push(:home, follower, status)
@@ -26,7 +29,9 @@ class FanOutOnWriteService < BaseService
   end
 
   def deliver_to_mentioned(status)
-    status.mentions.find_each do |mention|
+    Rails.logger.debug "Delivering status #{status.id} to mentioned accounts"
+
+    status.mentions.includes(:account).each do |mention|
       mentioned_account = mention.account
       next if !mentioned_account.local? || mentioned_account.id == status.account_id || FeedManager.instance.filter?(:mentions, status, mentioned_account)
       FeedManager.instance.push(:mentions, mentioned_account, status)
@@ -34,12 +39,15 @@ class FanOutOnWriteService < BaseService
   end
 
   def deliver_to_hashtags(status)
+    Rails.logger.debug "Delivering status #{status.id} to hashtags"
+
     status.tags.find_each do |tag|
       FeedManager.instance.broadcast("hashtag:#{tag.name}", id: status.id)
     end
   end
 
   def deliver_to_public(status)
+    Rails.logger.debug "Delivering status #{status.id} to public timeline"
     FeedManager.instance.broadcast(:public, id: status.id)
   end
 end