Merge pull request #681 from bcgov/develop/alex-GRAD2-2591

GRAD2-2591
bcgov · Sep 18, 2024 · a063c59 · a063c59
2 parents 934a78b + c082043
commit a063c59
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/api/src/main/java/ca/bc/gov/educ/api/gradstudent/service/GraduationStatusService.java b/api/src/main/java/ca/bc/gov/educ/api/gradstudent/service/GraduationStatusService.java
@@ -1118,6 +1118,15 @@ public Pair<GraduationStudentRecord, GradStatusEvent> undoCompletionStudent(UUID
                     if (gradStatusEvent != null) {
                         gradStatusEventRepository.save(gradStatusEvent);
                     }
+                    List<StudentOptionalProgramEntity> studentOptionalProgramEntities = gradStudentOptionalProgramRepository.findByStudentID(studentID);
+                    for(StudentOptionalProgramEntity studentOptionalProgramEntity: studentOptionalProgramEntities) {
+                        studentOptionalProgramEntity.setOptionalProgramCompletionDate(null);
+                        studentOptionalProgramEntity.setStudentOptionalProgramData(null);
+                        studentOptionalProgramEntity.setUpdateDate(LocalDateTime.now());
+                        studentOptionalProgramEntity.setUpdateUser(ThreadLocalStateUtil.getCurrentUser());
+                        gradStudentOptionalProgramRepository.save(studentOptionalProgramEntity);
+                        historyService.createStudentOptionalProgramHistory(studentOptionalProgramEntity,USER_UNDO_CMPL);
+                    }
                     return Pair.of(graduationStatusTransformer.transformToDTOWithModifiedProgramCompletionDate(gradEntity), gradStatusEvent);
                 } else {
                     validation.addErrorAndStop(String.format("Student ID [%s] does not exists", studentID));

diff --git a/api/src/test/java/ca/bc/gov/educ/api/gradstudent/service/GraduationStatusServiceTest.java b/api/src/test/java/ca/bc/gov/educ/api/gradstudent/service/GraduationStatusServiceTest.java
@@ -2030,6 +2030,14 @@ public void testUgradStudent() throws JsonProcessingException {
         when(graduationStatusRepository.findById(studentID)).thenReturn(Optional.of(graduationStatusEntity));
         when(graduationStatusRepository.save(responseGraduationStatus)).thenReturn(responseGraduationStatus);
 
+        StudentOptionalProgramEntity studentOptionalProgramEntity = new StudentOptionalProgramEntity();
+        studentOptionalProgramEntity.setId(UUID.randomUUID());
+        studentOptionalProgramEntity.setStudentID(studentID);
+        studentOptionalProgramEntity.setOptionalProgramCompletionDate(new Date(System.currentTimeMillis()));
+
+        when(gradStudentOptionalProgramRepository.findByStudentID(studentID)).thenReturn(List.of(studentOptionalProgramEntity));
+        doNothing().when(historyService).createStudentOptionalProgramHistory(any(), any());
+
         var response = graduationStatusService.undoCompletionStudent(studentID, ungradReasonCode,ungradReasonDesc, "accessToken");
         assertThat(response).isNotNull();