Fix to delete StatusCapabilityToken when deleting status batch

app/services/batched_remove_status_service.rb

@@ -23,6 +23,10 @@ def call(statuses, **options)
       status.send(:unlink_from_conversations)
     end
 
+    status_ids_with_capability_tokens = statuses.local.joins(:capability_tokens).where.not(capability_tokens: { id: nil }).pluck(:id)
+    status_ids_with_capability_tokens += Status.where(reblog_of_id: statuses).local.joins(:capability_tokens).where.not(capability_tokens: { id: nil }).pluck(:id)
+    status_ids_with_capability_tokens.each_slice(50) { |slice| StatusCapabilityToken.where(status_id: slice).delete_all }
+
     # We do not batch all deletes into one to avoid having a long-running
     # transaction lock the database, but we use the delete method instead
     # of destroy to avoid all callbacks. We rely on foreign keys to

spec/services/batched_remove_status_service_spec.rb

@@ -23,7 +23,9 @@
     status1
     status2
 
-    subject.call([status1, status2])
+    statuses = Status.where(id: [status1.id, status2.id])
+
+    subject.call(statuses)
   end
 
   it 'removes statuses' do