GRAD2-2591

Optional Program Completion Date is not being reset when a User UNGRADS a student
bcgov · Sep 12, 2024 · 487fc5c · 487fc5c
1 parent c5a5f30
commit 487fc5c
Showing 1 changed file with 7 additions and 0 deletions.
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,13 @@ 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));
+
         var response = graduationStatusService.undoCompletionStudent(studentID, ungradReasonCode,ungradReasonDesc, "accessToken");
         assertThat(response).isNotNull();