diff --git a/src/main/java/uk/gov/hmcts/reform/civil/model/citizenui/TranslatedDocument.java b/src/main/java/uk/gov/hmcts/reform/civil/model/citizenui/TranslatedDocument.java index d61b8a027..845b11ab0 100644 --- a/src/main/java/uk/gov/hmcts/reform/civil/model/citizenui/TranslatedDocument.java +++ b/src/main/java/uk/gov/hmcts/reform/civil/model/citizenui/TranslatedDocument.java @@ -13,6 +13,7 @@ import static uk.gov.hmcts.reform.civil.model.documents.DocumentType.DISMISSAL_ORDER; import static uk.gov.hmcts.reform.civil.model.documents.DocumentType.GENERAL_APPLICATION_DRAFT; import static uk.gov.hmcts.reform.civil.model.documents.DocumentType.GENERAL_ORDER; +import static uk.gov.hmcts.reform.civil.model.documents.DocumentType.HEARING_NOTICE; import static uk.gov.hmcts.reform.civil.model.documents.DocumentType.HEARING_ORDER; import static uk.gov.hmcts.reform.civil.model.documents.DocumentType.JUDGES_DIRECTIONS_APPLICANT_TRANSLATED; import static uk.gov.hmcts.reform.civil.model.documents.DocumentType.JUDGES_DIRECTIONS_RESPONDENT_TRANSLATED; @@ -56,6 +57,8 @@ public DocumentType getCorrespondingDocumentType(TranslatedDocumentType document return JUDGES_DIRECTIONS_RESPONDENT_TRANSLATED; case HEARING_ORDER: return HEARING_ORDER; + case HEARING_NOTICE: + return HEARING_NOTICE; case WITHOUT_NOTICE_TO_WITH_NOTICE_DOCUMENT: return SEND_APP_TO_OTHER_PARTY; case DISMISSAL_ORDER: diff --git a/src/main/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentService.java b/src/main/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentService.java index 5c5cda8b9..ac8f4ce54 100644 --- a/src/main/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentService.java +++ b/src/main/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentService.java @@ -84,6 +84,8 @@ private List> getExistingDocumentsByType(CaseData caseData return ofNullable(caseData.getGeneralOrderDocument()).orElse(new ArrayList<>()); case HEARING_ORDER: return ofNullable(caseData.getHearingOrderDocument()).orElse(new ArrayList<>()); + case HEARING_NOTICE: + return ofNullable(caseData.getHearingNoticeDocument()).orElse(new ArrayList<>()); case DISMISSAL_ORDER: return ofNullable(caseData.getDismissalOrderDocument()).orElse(new ArrayList<>()); case WRITTEN_REPRESENTATION_CONCURRENT: @@ -117,6 +119,9 @@ private void updateCaseDataBuilderByType(CaseData caseData, CaseData.CaseDataBui case HEARING_ORDER: caseDataBuilder.hearingOrderDocument(documents); break; + case HEARING_NOTICE: + caseDataBuilder.hearingNoticeDocument(documents); + break; case WRITTEN_REPRESENTATION_CONCURRENT: caseDataBuilder.writtenRepConcurrentDocument(documents); break; diff --git a/src/test/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentServiceTest.java b/src/test/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentServiceTest.java index e53b2c545..77cc3eacf 100644 --- a/src/test/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentServiceTest.java +++ b/src/test/java/uk/gov/hmcts/reform/civil/service/UploadTranslatedDocumentServiceTest.java @@ -101,6 +101,12 @@ void shouldHandleMultipleDocumentTypes() { .file(mock(Document.class)) .build()).build()); + translatedDocuments.add(Element.builder().value( + TranslatedDocument.builder() + .documentType(TranslatedDocumentType.HEARING_NOTICE) + .file(mock(Document.class)) + .build()).build()); + translatedDocuments.add(Element.builder().value( TranslatedDocument.builder() .documentType(TranslatedDocumentType.JUDGES_DIRECTIONS_ORDER) @@ -142,12 +148,13 @@ void shouldHandleMultipleDocumentTypes() { // Then assertThat(result.getGeneralOrderDocument()).isNotNull(); assertThat(result.getHearingOrderDocument()).isNotNull(); + assertThat(result.getHearingNoticeDocument()).isNotNull(); assertThat(result.getDirectionOrderDocument()).isNotNull(); assertThat(result.getWrittenRepSequentialDocument()).isNotNull(); assertThat(result.getWrittenRepConcurrentDocument()).isNotNull(); assertThat(result.getDismissalOrderDocument()).isNotNull(); assertThat(result.getGaDraftDocument()).isNotNull(); assertThat(result.getGeneralOrderDocument().get(0).getValue().getCreatedBy()).isEqualTo(translator); - verify(assignCategoryId, times(8)).assignCategoryIdToCollection(anyList(), any(), any()); + verify(assignCategoryId, times(9)).assignCategoryIdToCollection(anyList(), any(), any()); } }