Fix tests and code be selecting the first element of the attachments

ministryofjustice · Jul 11, 2024 · a855677 · a855677
1 parent 64dc49a
commit a855677
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/app/presenters/message_presenter.rb b/app/presenters/message_presenter.rb
@@ -39,11 +39,11 @@ def download_file_link
   end
 
   def attachment_file_name
-    message.attachment.filename.to_s
+    message.attachment.first.filename.to_s
   end
 
   def attachment_file_size
-    h.number_to_human_size(message.attachment.byte_size)
+    h.number_to_human_size(message.attachment.first.byte_size)
   end
 
   def hide_author?

diff --git a/spec/presenters/message_presenter_spec.rb b/spec/presenters/message_presenter_spec.rb
@@ -16,10 +16,10 @@
     context 'with an attachment' do
       let(:file) { File.expand_path('features/examples/shorter_lorem.docx', Rails.root) }
       let(:file_size) { number_to_human_size(File.size(file)) }
-      let(:attachment) { Rack::Test::UploadedFile.new(file) }
+      let(:attachment) { [Rack::Test::UploadedFile.new(file)] }
 
       before do
-        allow(message.attachment).to receive(:url).and_return('http://example.com')
+        allow(message.attachment.first).to receive(:url).and_return('http://example.com')
       end
 
       it 'includes a download link to the attachment' do