Amend message.attach to includ first

ministryofjustice · Jul 12, 2024 · 27b844d · 27b844d
1 parent 63c8f68
commit 27b844d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/app/interfaces/api/entities/document.rb b/app/interfaces/api/entities/document.rb
@@ -9,19 +9,19 @@ class Document < BaseEntity
       private
 
       def attachment
-        object.is_a?(ActiveStorage::Attachment) ? object : object.attachment
+        object.is_a?(ActiveStorage::Attachment) ? object : object.attachment.first
       end
 
       def url
-        attachment.blob.url(disposition: 'attachment') if attachment.attached?
+        attachment.first.blob.url(disposition: 'attachment') if attachment.first.attached?
       end
 
       def file_name
-        attachment.filename if attachment.attached?
+        attachment.first.filename if attachment.first.attached?
       end
 
       def size
-        attachment.byte_size if attachment.attached?
+        attachment.first.byte_size if attachment.first.attached?
       end
     end
   end

diff --git a/app/presenters/message_presenter.rb b/app/presenters/message_presenter.rb
@@ -8,7 +8,7 @@ def sender_is_a?(klass)
   def body
     h.tag.div do
       h.concat(simple_format(message.body))
-      attachment_field if message.attachment.present?
+      attachment_field if message.attachment.first.present?
     end
   end