update RPA for intermediate track

hmcts · Feb 21, 2025 · a8aa677 · a8aa677
1 parent 3efea8e
commit a8aa677
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 15 deletions.
diff --git a/src/main/java/uk/gov/hmcts/reform/civil/service/robotics/mapper/RoboticsDataMapper.java b/src/main/java/uk/gov/hmcts/reform/civil/service/robotics/mapper/RoboticsDataMapper.java
@@ -104,16 +104,13 @@ private CaseHeader buildCaseHeader(CaseData caseData, String authToken) {
     }
 
     private String buildAllocatedTrack(AllocatedTrack allocatedTrack) {
-        switch (allocatedTrack) {
-            case FAST_CLAIM:
-                return "FAST TRACK";
-            case MULTI_CLAIM:
-                return "MULTI TRACK";
-            case SMALL_CLAIM:
-                return "SMALL CLAIM TRACK";
-            default:
-                return "";
-        }
+        return switch (allocatedTrack) {
+            case FAST_CLAIM -> "FAST TRACK";
+            case MULTI_CLAIM -> "MULTI TRACK";
+            case SMALL_CLAIM -> "SMALL CLAIM TRACK";
+            case INTERMEDIATE_CLAIM -> "INTERMEDIATE TRACK";
+            default -> "";
+        };
     }
 
     private List<Solicitor> buildSolicitors(CaseData caseData) {

diff --git a/src/test/java/uk/gov/hmcts/reform/civil/service/robotics/mapper/RoboticsDataMapperTest.java b/src/test/java/uk/gov/hmcts/reform/civil/service/robotics/mapper/RoboticsDataMapperTest.java
@@ -15,11 +15,13 @@
 import uk.gov.hmcts.reform.ccd.model.PreviousOrganisationCollectionItem;
 import uk.gov.hmcts.reform.civil.assertion.CustomAssertions;
 import uk.gov.hmcts.reform.civil.config.PrdAdminUserConfiguration;
+import uk.gov.hmcts.reform.civil.enums.AllocatedTrack;
 import uk.gov.hmcts.reform.civil.enums.CaseState;
 import uk.gov.hmcts.reform.civil.helpers.CaseDetailsConverter;
 import uk.gov.hmcts.reform.civil.model.Address;
 import uk.gov.hmcts.reform.civil.model.CaseData;
 import uk.gov.hmcts.reform.civil.model.SolicitorOrganisationDetails;
+import uk.gov.hmcts.reform.civil.model.robotics.CaseHeader;
 import uk.gov.hmcts.reform.civil.model.robotics.NoticeOfChange;
 import uk.gov.hmcts.reform.civil.model.robotics.RoboticsCaseData;
 import uk.gov.hmcts.reform.civil.prd.client.OrganisationApi;
@@ -384,21 +386,46 @@ void shouldMapToRoboticsCaseDataWhen2ndDefendantIsNotRepresented() {
 
     @Test
     void shouldMapToRoboticsCaseDataWhenPreferredCourtCodeFetchedFromRefData() {
-        CaseData caseData = CaseDataBuilder.builder().atStatePaymentSuccessful().build();
-        when(locationRefDataUtil.getPreferredCourtData(any(), any(), eq(true))).thenReturn("121");
+        CaseData caseData = CaseDataBuilder.builder().atStatePaymentSuccessful().build().toBuilder()
+            .allocatedTrack(AllocatedTrack.INTERMEDIATE_CLAIM)
+            .build();
+        RoboticsCaseData testHeader = RoboticsCaseData.builder()
+            .header(CaseHeader.builder()
+                        .caseNumber(caseData.getLegacyCaseReference())
+                        .owningCourtCode("807")
+                        .owningCourtName("CCMCC")
+                        .caseType("PERSONAL INJURY")
+                        .preferredCourtCode("121")
+                        .caseAllocatedTo("INTERMEDIATE TRACK")
+                        .build())
+            .build();
 
+        when(locationRefDataUtil.getPreferredCourtData(any(), any(), eq(true))).thenReturn("121");
         RoboticsCaseData roboticsCaseData = mapper.toRoboticsCaseData(caseData, BEARER_TOKEN);
+
         CustomAssertions.assertThat(roboticsCaseData).isEqualTo(caseData);
-        assertThat(roboticsCaseData.getHeader().getPreferredCourtCode()).isEqualTo("121");
+        assertThat(roboticsCaseData.getHeader()).isEqualTo(testHeader.getHeader());
     }
 
     @Test
-    void shouldReturnEmptyStringWhenPreferredCourtCodeisUnavailableFromLocationRefData() {
+    void shouldReturnHeaderWhenPreferredCourtCodeisUnavailableFromLocationRefData() {
         CaseData caseData = CaseDataBuilder.builder().atStatePaymentSuccessful().build();
+        RoboticsCaseData testHeader = RoboticsCaseData.builder()
+            .header(CaseHeader.builder()
+                        .caseNumber(caseData.getLegacyCaseReference())
+                        .owningCourtCode("807")
+                        .owningCourtName("CCMCC")
+                        .caseType("PERSONAL INJURY")
+                        .preferredCourtCode("")
+                        .caseAllocatedTo("FAST TRACK")
+                        .build())
+            .build();
+
         when(locationRefDataUtil.getPreferredCourtData(any(), any(), eq(true))).thenReturn("");
         RoboticsCaseData roboticsCaseData = mapper.toRoboticsCaseData(caseData, BEARER_TOKEN);
+
         CustomAssertions.assertThat(roboticsCaseData).isEqualTo(caseData);
-        assertThat(roboticsCaseData.getHeader().getPreferredCourtCode()).isEqualTo("");
+        assertThat(roboticsCaseData.getHeader()).isEqualTo(testHeader.getHeader());
     }
 
     @Test