Reports spec for media attachments, clean up method (#2660)

* Add coverage for Report#media_attachments

* Direct query on media attachment
master
Matt Jankowski 7 years ago committed by Eugen Rochko
parent b83bc0ae64
commit 5259319cf5
  1. 6
      app/models/report.rb
  2. 11
      spec/models/report_spec.rb

@ -13,10 +13,6 @@ class Report < ApplicationRecord
end
def media_attachments
media_attachments = []
statuses.each do |s|
media_attachments.concat s.media_attachments
end
media_attachments
MediaAttachment.where(status_id: status_ids)
end
end

@ -10,4 +10,15 @@ describe Report do
expect(report.statuses).to eq [status]
end
end
describe 'media_attachments' do
it 'returns media attachments from statuses' do
status = Fabricate(:status)
media_attachment = Fabricate(:media_attachment, status: status)
_other_media_attachment = Fabricate(:media_attachment)
report = Fabricate(:report, status_ids: [status.id])
expect(report.media_attachments).to eq [media_attachment]
end
end
end

Loading…
Cancel
Save