Parcourir la source

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

* Add coverage for Report#media_attachments

* Direct query on media attachment
master
Matt Jankowski il y a 7 ans
committed by Eugen Rochko
Parent
révision
5259319cf5
2 fichiers modifiés avec 12 ajouts et 5 suppressions
  1. +1
    -5
      app/models/report.rb
  2. +11
    -0
      spec/models/report_spec.rb

+ 1
- 5
app/models/report.rb Voir le fichier

@@ -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

+ 11
- 0
spec/models/report_spec.rb Voir le fichier

@@ -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

Chargement…
Annuler
Enregistrer