Dfr 3515 seal orders add court stamp

src/main/java/uk/gov/hmcts/reform/finrem/caseorchestration/handler/DirectionUploadOrderAboutToSubmitHandler.java

@@ -6,6 +6,7 @@
 import uk.gov.hmcts.reform.finrem.caseorchestration.ccd.callback.CallbackType;
 import uk.gov.hmcts.reform.finrem.caseorchestration.controllers.GenericAboutToStartOrSubmitCallbackResponse;
 import uk.gov.hmcts.reform.finrem.caseorchestration.error.CourtDetailsParseException;
+import uk.gov.hmcts.reform.finrem.caseorchestration.helper.DocumentHelper;
 import uk.gov.hmcts.reform.finrem.caseorchestration.mapper.FinremCaseDetailsMapper;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.EventType;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.CaseDocument;
@@ -19,7 +20,10 @@
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.draftorders.review.DraftOrderDocReviewCollection;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.draftorders.review.PsaDocReviewCollection;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.wrapper.DraftOrdersWrapper;
+import uk.gov.hmcts.reform.finrem.caseorchestration.model.document.DocumentCategory;
 import uk.gov.hmcts.reform.finrem.caseorchestration.service.AdditionalHearingDocumentService;
+import uk.gov.hmcts.reform.finrem.caseorchestration.service.GenericDocumentService;
+import uk.gov.hmcts.reform.finrem.caseorchestration.service.StampType;
 import uk.gov.hmcts.reform.finrem.caseorchestration.service.draftorders.HasApprovableCollectionReader;
 
 import java.util.ArrayList;
@@ -41,12 +45,19 @@ public class DirectionUploadOrderAboutToSubmitHandler extends FinremCallbackHand
 
     private final HasApprovableCollectionReader hasApprovableCollectionReader;
 
+    private final DocumentHelper documentHelper;
+
+    private final GenericDocumentService genericDocumentService;
+
     public DirectionUploadOrderAboutToSubmitHandler(FinremCaseDetailsMapper finremCaseDetailsMapper,
                                                     AdditionalHearingDocumentService additionalHearingDocumentService,
-                                                    HasApprovableCollectionReader hasApprovableCollectionReader) {
+                                                    HasApprovableCollectionReader hasApprovableCollectionReader,
+                                                    DocumentHelper documentHelper, GenericDocumentService genericDocumentService) {
         super(finremCaseDetailsMapper);
         this.additionalHearingDocumentService = additionalHearingDocumentService;
         this.hasApprovableCollectionReader = hasApprovableCollectionReader;
+        this.genericDocumentService = genericDocumentService;
+        this.documentHelper = documentHelper;
     }
 
     @Override
@@ -75,9 +86,9 @@ public GenericAboutToStartOrSubmitCallbackResponse<FinremCaseData> handle(Finrem
                 .data(caseData).errors(List.of("There was an unexpected error")).build();
         }
 
-        handleDraftOrderDocuments(caseData);
-        handlePsaDocuments(caseData);
-        handleAgreedDraftOrdersCollection(caseData);
+        handleDraftOrderDocuments(caseDetails, userAuthorisation);
+        handlePsaDocuments(caseDetails, userAuthorisation);
+        handleAgreedDraftOrdersCollection(caseDetails, userAuthorisation);
         clearTemporaryFields(caseData);
 
         return GenericAboutToStartOrSubmitCallbackResponse.<FinremCaseData>builder().data(caseData).errors(errors).build();
@@ -95,47 +106,76 @@ private void handleNewDocumentInUnprocessedApprovedDocuments(FinremCaseData case
         });
     }
 
-    private void handleDraftOrderDocuments(FinremCaseData caseData) {
+    private void handleDraftOrderDocuments(FinremCaseDetails caseDetails, String authorisation) {
+        FinremCaseData caseData = caseDetails.getData();
+
         List<DraftOrderDocReviewCollection> collector = new ArrayList<>();
         hasApprovableCollectionReader.filterAndCollectDraftOrderDocs(caseData.getDraftOrdersWrapper().getDraftOrdersReviewCollection(),
             collector, APPROVED_BY_JUDGE::equals);
 
         getApprovedDocumentsToProcess(caseData)
             .forEach(unprocessedApprovedOrder ->
                 collector.stream().filter(psa -> doesDocumentMatch(psa, unprocessedApprovedOrder)).forEach(toBeUpdated -> {
+
+                    StampType stampType = documentHelper.getStampType(caseData);
+                    CaseDocument stampedDoc = genericDocumentService.stampDocument(
+                        unprocessedApprovedOrder.getValue().getUploadDraftDocument(), authorisation, stampType,
+                        String.valueOf(caseDetails.getId()));
+                    stampedDoc.setCategoryId(
+                        DocumentCategory.APPROVED_ORDERS.getDocumentCategoryId());
+
                     toBeUpdated.getValue().setOrderStatus(PROCESSED);
-                    toBeUpdated.getValue().setDraftOrderDocument(unprocessedApprovedOrder.getValue().getUploadDraftDocument());
+                    toBeUpdated.getValue().setDraftOrderDocument(stampedDoc);
                 }));
     }
 
-    private void handlePsaDocuments(FinremCaseData caseData) {
+    private void handlePsaDocuments(FinremCaseDetails caseDetails, String authorisation) {
+        FinremCaseData caseData = caseDetails.getData();
         List<PsaDocReviewCollection> psaCollector = new ArrayList<>();
+
         hasApprovableCollectionReader.filterAndCollectPsaDocs(caseData.getDraftOrdersWrapper().getDraftOrdersReviewCollection(),
             psaCollector, APPROVED_BY_JUDGE::equals);
 
         getApprovedDocumentsToProcess(caseData)
             .forEach(unprocessedApprovedOrder ->
                 psaCollector.stream().filter(psa -> doesDocumentMatch(psa, unprocessedApprovedOrder)).forEach(toBeUpdated -> {
+
+                    StampType stampType = documentHelper.getStampType(caseData);
+                    CaseDocument stampedDoc = genericDocumentService.stampDocument(
+                        unprocessedApprovedOrder.getValue().getUploadDraftDocument(), authorisation, stampType,
+                        String.valueOf(caseDetails.getId()));
+                    stampedDoc.setCategoryId(
+                        DocumentCategory.APPROVED_ORDERS.getDocumentCategoryId());
+
                     toBeUpdated.getValue().setOrderStatus(PROCESSED);
-                    toBeUpdated.getValue().setPsaDocument(unprocessedApprovedOrder.getValue().getUploadDraftDocument());
+                    toBeUpdated.getValue().setPsaDocument(stampedDoc);
                 }));
     }
 
-    private void handleAgreedDraftOrdersCollection(FinremCaseData caseData) {
+    private void handleAgreedDraftOrdersCollection(FinremCaseDetails caseDetails, String authorisation) {
+        FinremCaseData caseData = caseDetails.getData();
         List<AgreedDraftOrderCollection> agreedOrderCollector = new ArrayList<>();
+
         hasApprovableCollectionReader.collectAgreedDraftOrders(caseData.getDraftOrdersWrapper().getAgreedDraftOrderCollection(),
             agreedOrderCollector, APPROVED_BY_JUDGE::equals);
 
         getApprovedDocumentsToProcess(caseData)
             .forEach(unprocessedApprovedOrder ->
                 agreedOrderCollector.stream().filter(agreedDraftOrder -> doesDocumentMatch(agreedDraftOrder, unprocessedApprovedOrder))
                     .forEach(toBeUpdated -> {
+
+                        StampType stampType = documentHelper.getStampType(caseData);
+                        CaseDocument stampedDoc = genericDocumentService.stampDocument(
+                            unprocessedApprovedOrder.getValue().getUploadDraftDocument(), authorisation, stampType,
+                            String.valueOf(caseDetails.getId()));
+                        stampedDoc.setCategoryId(unprocessedApprovedOrder.getValue().getUploadDraftDocument().getCategoryId());
+
                         toBeUpdated.getValue().setOrderStatus(PROCESSED);
-                        // replace the document by the new uploaded approved document
+                        // replace the document by the new stamped document
                         if (toBeUpdated.getValue().getPensionSharingAnnex() != null) {
-                            toBeUpdated.getValue().setPensionSharingAnnex(unprocessedApprovedOrder.getValue().getUploadDraftDocument());
+                            toBeUpdated.getValue().setPensionSharingAnnex(stampedDoc);
                         } else if (toBeUpdated.getValue().getDraftOrder() != null) {
-                            toBeUpdated.getValue().setDraftOrder(unprocessedApprovedOrder.getValue().getUploadDraftDocument());
+                            toBeUpdated.getValue().setDraftOrder(stampedDoc);
                         }
                     }));
     }

src/test/java/uk/gov/hmcts/reform/finrem/caseorchestration/handler/DirectionUploadOrderAboutToSubmitHandlerTest.java

@@ -12,6 +12,7 @@
 import uk.gov.hmcts.reform.finrem.caseorchestration.FinremCallbackRequestFactory;
 import uk.gov.hmcts.reform.finrem.caseorchestration.controllers.GenericAboutToStartOrSubmitCallbackResponse;
 import uk.gov.hmcts.reform.finrem.caseorchestration.error.CourtDetailsParseException;
+import uk.gov.hmcts.reform.finrem.caseorchestration.helper.DocumentHelper;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.CaseDocument;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.DirectionOrder;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.DirectionOrderCollection;
@@ -28,6 +29,8 @@
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.draftorders.review.PsaDocumentReview;
 import uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.wrapper.DraftOrdersWrapper;
 import uk.gov.hmcts.reform.finrem.caseorchestration.service.AdditionalHearingDocumentService;
+import uk.gov.hmcts.reform.finrem.caseorchestration.service.GenericDocumentService;
+import uk.gov.hmcts.reform.finrem.caseorchestration.service.StampType;
 import uk.gov.hmcts.reform.finrem.caseorchestration.service.draftorders.HasApprovableCollectionReader;
 
 import java.util.ArrayList;
@@ -40,14 +43,17 @@
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.lenient;
+import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.TestConstants.AUTH_TOKEN;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.ccd.callback.CallbackType.ABOUT_TO_SUBMIT;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.model.EventType.DIRECTION_UPLOAD_ORDER;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.CaseType.CONTESTED;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.draftorders.review.OrderStatus.APPROVED_BY_JUDGE;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.draftorders.review.OrderStatus.PROCESSED;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.model.ccd.draftorders.review.OrderStatus.TO_BE_REVIEWED;
+import static uk.gov.hmcts.reform.finrem.caseorchestration.service.StampType.FAMILY_COURT_STAMP;
 import static uk.gov.hmcts.reform.finrem.caseorchestration.test.Assertions.assertCanHandle;
 
 @ExtendWith(MockitoExtension.class)
@@ -70,6 +76,12 @@ class DirectionUploadOrderAboutToSubmitHandlerTest {
     @Mock
     private AdditionalHearingDocumentService additionalHearingDocumentService;
 
+    @Mock
+    private GenericDocumentService genericDocumentService;
+
+    @Mock
+    private DocumentHelper documentHelper;
+
     @Test
     void testCanHandle() {
         assertCanHandle(underTest, ABOUT_TO_SUBMIT, CONTESTED, DIRECTION_UPLOAD_ORDER);
@@ -172,6 +184,14 @@ void shouldMarkDraftOrdersReviewCollectionProcessed() {
         AgreedDraftOrderCollection test4 = null;
         AgreedDraftOrderCollection test5 = null;
         AgreedDraftOrderCollection test6 = null;
+        CaseDocument stampedDocumentA = CaseDocument.builder().documentFilename("stampedA.pdf").build();
+        CaseDocument stampedDocumentB = CaseDocument.builder().documentFilename("stampedB.pdf").build();
+
+        when(documentHelper.getStampType(any(FinremCaseData.class))).thenReturn(FAMILY_COURT_STAMP);
+        when(genericDocumentService.stampDocument(eq(TARGET_DOCUMENT_1), eq(AUTH_TOKEN), eq(FAMILY_COURT_STAMP), any(String.class)))
+            .thenReturn(stampedDocumentA);
+        when(genericDocumentService.stampDocument(eq(TARGET_DOCUMENT_2), eq(AUTH_TOKEN), eq(FAMILY_COURT_STAMP), any(String.class)))
+            .thenReturn(stampedDocumentB);
 
         FinremCallbackRequest finremCallbackRequest = FinremCallbackRequestFactory.from(FinremCaseData.builder()
             .draftOrdersWrapper(DraftOrdersWrapper.builder()
@@ -212,10 +232,20 @@ void shouldMarkDraftOrdersReviewCollectionProcessed() {
 
         underTest.handle(finremCallbackRequest, AUTH_TOKEN);
 
+        verify(genericDocumentService, times(4)).stampDocument(any(CaseDocument.class), eq(AUTH_TOKEN),
+            eq(StampType.FAMILY_COURT_STAMP), any(String.class));
+        //Check DraftOrderDocReviewCollection is updated
+        assertEquals(stampedDocumentA, test1.getValue().getDraftOrderDocument());
         assertEquals(PROCESSED, test1.getValue().getOrderStatus());
+        assertEquals(stampedDocumentB, test2.getValue().getDraftOrderDocument());
         assertEquals(PROCESSED, test2.getValue().getOrderStatus());
+
+        //Check AgreedDraftOrderCollection is updated
+        assertEquals(stampedDocumentA, test3.getValue().getDraftOrder());
         assertEquals(PROCESSED, test3.getValue().getOrderStatus());
+        assertEquals(stampedDocumentB, test4.getValue().getDraftOrder());
         assertEquals(PROCESSED, test4.getValue().getOrderStatus());
+
         assertEquals(APPROVED_BY_JUDGE, test5.getValue().getOrderStatus());
         assertEquals(TO_BE_REVIEWED, test6.getValue().getOrderStatus());
     }
@@ -227,6 +257,14 @@ void shouldMarkPsaCollectionProcessed() {
         AgreedDraftOrderCollection test3 = null;
         AgreedDraftOrderCollection test4 = null;
         AgreedDraftOrderCollection test5 = null;
+        CaseDocument stampedDocumentA = CaseDocument.builder().documentFilename("stampedA.pdf").build();
+        CaseDocument stampedDocumentB = CaseDocument.builder().documentFilename("stampedB.pdf").build();
+
+        when(documentHelper.getStampType(any(FinremCaseData.class))).thenReturn(FAMILY_COURT_STAMP);
+        when(genericDocumentService.stampDocument(eq(TARGET_DOCUMENT_1), eq(AUTH_TOKEN), eq(FAMILY_COURT_STAMP), any(String.class)))
+            .thenReturn(stampedDocumentA);
+        when(genericDocumentService.stampDocument(eq(TARGET_DOCUMENT_2), eq(AUTH_TOKEN), eq(FAMILY_COURT_STAMP), any(String.class)))
+            .thenReturn(stampedDocumentB);
 
         FinremCallbackRequest finremCallbackRequest = FinremCallbackRequestFactory.from(FinremCaseData.builder()
             .draftOrdersWrapper(DraftOrdersWrapper.builder()
@@ -267,10 +305,20 @@ void shouldMarkPsaCollectionProcessed() {
 
         underTest.handle(finremCallbackRequest, AUTH_TOKEN);
 
+        verify(genericDocumentService, times(4)).stampDocument(any(CaseDocument.class), eq(AUTH_TOKEN),
+            eq(StampType.FAMILY_COURT_STAMP), any(String.class));
+        //Check PsaDocReviewCollection is updated
+        assertEquals(stampedDocumentA, test1.getValue().getPsaDocument());
         assertEquals(PROCESSED, test1.getValue().getOrderStatus());
+        assertEquals(stampedDocumentB, test2.getValue().getPsaDocument());
         assertEquals(PROCESSED, test2.getValue().getOrderStatus());
+
+        //Check AgreedDraftOrderCollection is updated
+        assertEquals(stampedDocumentA, test3.getValue().getPensionSharingAnnex());
         assertEquals(PROCESSED, test3.getValue().getOrderStatus());
+        assertEquals(stampedDocumentB, test4.getValue().getPensionSharingAnnex());
         assertEquals(PROCESSED, test4.getValue().getOrderStatus());
+
         assertEquals(APPROVED_BY_JUDGE, test5.getValue().getOrderStatus());
     }
 
@@ -281,6 +329,15 @@ void shouldReplaceApprovedDocumentAndMarkAsProcessed() {
         AgreedDraftOrderCollection test3 = null;
         AgreedDraftOrderCollection test4 = null;
 
+        CaseDocument stampedDocumentA = CaseDocument.builder().documentFilename("stampedA.pdf").build();
+        CaseDocument stampedDocumentB = CaseDocument.builder().documentFilename("stampedB.pdf").build();
+
+        when(documentHelper.getStampType(any(FinremCaseData.class))).thenReturn(FAMILY_COURT_STAMP);
+        when(genericDocumentService.stampDocument(eq(TARGET_DOCUMENT_3), eq(AUTH_TOKEN), eq(FAMILY_COURT_STAMP), any(String.class)))
+            .thenReturn(stampedDocumentA);
+        when(genericDocumentService.stampDocument(eq(TARGET_DOCUMENT_4), eq(AUTH_TOKEN), eq(FAMILY_COURT_STAMP), any(String.class)))
+            .thenReturn(stampedDocumentB);
+
         FinremCallbackRequest finremCallbackRequest = FinremCallbackRequestFactory.from(FinremCaseData.builder()
             .draftOrdersWrapper(DraftOrdersWrapper.builder()
                 .unprocessedApprovedDocuments(List.of(
@@ -316,12 +373,18 @@ void shouldReplaceApprovedDocumentAndMarkAsProcessed() {
 
         underTest.handle(finremCallbackRequest, AUTH_TOKEN);
 
+        verify(genericDocumentService, times(4)).stampDocument(any(CaseDocument.class), eq(AUTH_TOKEN),
+            eq(StampType.FAMILY_COURT_STAMP), any(String.class));
+        //Check PsaDocReviewCollection is updated
         assertEquals(PROCESSED, test1.getValue().getOrderStatus());
-        assertEquals(TARGET_DOCUMENT_3, test1.getValue().getPsaDocument());
+        assertEquals(stampedDocumentA, test1.getValue().getPsaDocument());
         assertEquals(PROCESSED, test2.getValue().getOrderStatus());
-        assertEquals(TARGET_DOCUMENT_4, test2.getValue().getPsaDocument());
+        assertEquals(stampedDocumentB, test2.getValue().getPsaDocument());
+
         assertEquals(PROCESSED, test3.getValue().getOrderStatus());
+        assertEquals(stampedDocumentA, test1.getValue().getPsaDocument());
         assertEquals(PROCESSED, test4.getValue().getOrderStatus());
+        assertEquals(stampedDocumentA, test1.getValue().getPsaDocument());
     }
 
     private DraftOrderDocReviewCollection buildDraftOrderDocReviewCollection(OrderStatus orderStatus) {