CCMSPUI-530 Added title to DocumentMapper when mapping title

Signed-off-by: Jamie Briggs <jamie.briggs@digital.justice.gov.uk>
ministryofjustice · Feb 6, 2025 · 328a213 · 328a213
1 parent b92a21a
commit 328a213
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/data-service/src/main/java/uk/gov/laa/ccms/data/mapper/DocumentMapper.java b/data-service/src/main/java/uk/gov/laa/ccms/data/mapper/DocumentMapper.java
@@ -51,7 +51,6 @@ public interface DocumentMapper {
    *   <li>fileData</li>
    *   <li>statusDescription</li>
    *   <li>fileExtension</li>
-   *   <li>title</li>
    *   <li>status</li>
    *   <li>documentType</li>
    * </ul>
@@ -64,9 +63,9 @@ public interface DocumentMapper {
   @Mapping(target = "fileData",  constant = "")
   @Mapping(target = "statusDescription",  constant = "")
   @Mapping(target = "fileExtension",  constant = "")
-  @Mapping(target = "title", constant = "")
   @Mapping(target = "status", constant = "")
   @Mapping(target = "documentType", constant = "")
+  @Mapping(target = "title", source = "attachmentTitle")
   @Mapping(target = "text", source = "attachmentDescription")
   @Mapping(target = "documentId", source = "attachmentId")
   Document mapToNotification(NotificationAttachment notificationAttachment);

diff --git a/data-service/src/test/java/uk/gov/laa/ccms/data/mapper/DocumentMapperImplTest.java b/data-service/src/test/java/uk/gov/laa/ccms/data/mapper/DocumentMapperImplTest.java
@@ -50,6 +50,7 @@ void shouldMapNotificationAttachmentToDocument(){
     Document result = mapper.mapToNotification(attachment);
     // Then
     assertEquals("1", result.getDocumentId());
+    assertEquals("Title", result.getTitle());
     assertEquals("Description", result.getText());
   }