GRAD2-3176 - adds tests

bcgov · Jan 31, 2025 · 60e3170 · 60e3170
1 parent ee26b2b
commit 60e3170
Show file tree

Hide file tree

Showing 4 changed files with 64 additions and 2 deletions.
diff --git a/api/src/main/java/ca/bc/gov/educ/api/graduation/model/dto/institute/DistrictAddress.java b/api/src/main/java/ca/bc/gov/educ/api/graduation/model/dto/institute/DistrictAddress.java
@@ -1,13 +1,15 @@
 package ca.bc.gov.educ.api.graduation.model.dto.institute;
 
 import ca.bc.gov.educ.api.graduation.model.dto.BaseModel;
+import lombok.Builder;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 
 import java.io.Serializable;
 
 @Data
 @EqualsAndHashCode(callSuper = true)
+@Builder
 public class DistrictAddress extends BaseModel implements Serializable {
     /**
      * The constant serialVersionUID.

diff --git a/api/src/main/java/ca/bc/gov/educ/api/graduation/model/dto/institute/DistrictContact.java b/api/src/main/java/ca/bc/gov/educ/api/graduation/model/dto/institute/DistrictContact.java
@@ -2,6 +2,7 @@
 
 import ca.bc.gov.educ.api.graduation.model.dto.BaseModel;
 import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import lombok.Builder;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 
@@ -10,6 +11,7 @@
 @Data
 @EqualsAndHashCode(callSuper = true)
 @JsonIgnoreProperties(ignoreUnknown = true)
+@Builder
 public class DistrictContact extends BaseModel implements Serializable {
 
     /**

diff --git a/api/src/test/java/ca/bc/gov/educ/api/graduation/service/BaseServiceTest.java b/api/src/test/java/ca/bc/gov/educ/api/graduation/service/BaseServiceTest.java
@@ -1,8 +1,12 @@
 package ca.bc.gov.educ.api.graduation.service;
 
+import ca.bc.gov.educ.api.graduation.constants.AddressTypeCodes;
+import ca.bc.gov.educ.api.graduation.constants.DistrictContactTypeCodes;
 import ca.bc.gov.educ.api.graduation.model.dto.institute.District;
 import ca.bc.gov.educ.api.graduation.model.dto.GradCertificateType;
 import ca.bc.gov.educ.api.graduation.model.dto.ReportGradStudentData;
+import ca.bc.gov.educ.api.graduation.model.dto.institute.DistrictAddress;
+import ca.bc.gov.educ.api.graduation.model.dto.institute.DistrictContact;
 import ca.bc.gov.educ.api.graduation.model.dto.institute.School;
 
 import java.util.ArrayList;
@@ -167,6 +171,28 @@ public District createDistrict() {
     return District.builder()
         .districtId(UUID.randomUUID().toString())
         .displayName("Generic District Name")
+        .districtNumber("005")
+        .build();
+  }
+
+  public DistrictContact createDistrictContact(UUID districtId) {
+    return DistrictContact.builder()
+        .districtId(districtId.toString())
+        .districtContactId(UUID.randomUUID().toString())
+        .districtContactTypeCode(DistrictContactTypeCodes.SUPER.getCode())
+        .firstName("Super")
+        .lastName("Man")
+        .build();
+  }
+
+  public DistrictAddress createDistrictAddress(UUID districtId) {
+    return DistrictAddress.builder()
+        .districtId(districtId.toString())
+        .districtAddressId(UUID.randomUUID().toString())
+        .addressTypeCode(AddressTypeCodes.MAILING.getCode())
+        .addressLine1("1234 Generic Street")
+        .city("Generic City")
+        .postal("V1V1V1")
         .build();
   }
 }
diff --git a/api/src/test/java/ca/bc/gov/educ/api/graduation/service/DistrictReportServiceTest.java b/api/src/test/java/ca/bc/gov/educ/api/graduation/service/DistrictReportServiceTest.java
@@ -1,5 +1,6 @@
 package ca.bc.gov.educ.api.graduation.service;
 
+import ca.bc.gov.educ.api.graduation.constants.ReportTypeCodes;
 import ca.bc.gov.educ.api.graduation.model.dto.institute.District;
 import ca.bc.gov.educ.api.graduation.model.dto.ReportGradStudentData;
 import ca.bc.gov.educ.api.graduation.model.dto.institute.School;
@@ -42,11 +43,10 @@ class DistrictReportServiceTest extends BaseServiceTest {
 
   @BeforeEach
   void setUp() {
-  MockitoAnnotations.openMocks(this);
+    MockitoAnnotations.openMocks(this);
   }
 
 
-
   @Test
   void createAndStoreDistrictYearEndReports() {
     List<ReportGradStudentData> studentData = createStudentSchoolYearEndData();
@@ -65,4 +65,36 @@ void createAndStoreDistrictYearEndReports() {
 
     assertEquals(1, result);
   }
+
+  @Test
+  void createAndStoreDistrictNonGradYearEndReport() {
+    List<ReportGradStudentData> studentData = createStudentSchoolYearEndData();
+    District district = createDistrict();
+    when(reportService.getStudentsForSchoolNonGradYearEndReport()).thenReturn(studentData);
+    studentData.forEach(student -> {
+          UUID schoolId = student.getSchoolAtGradId() == null ? UUID.fromString(student.getSchoolOfRecordId()) : UUID.fromString(student.getSchoolAtGradId());
+          School school = createSchool(UUID.fromString(district.getDistrictId()), schoolId, student.getMincode());
+          when(schoolService.getSchoolById(schoolId)).thenReturn(school);
+        }
+    );
+    when(districtService.getDistrictDetails(any(UUID.class))).thenReturn(district);
+    when(restService.post(any(), any(), any())).thenReturn(new byte[0]);
+
+    int result = districtReportService.createAndStoreDistrictNonGradYearEndReport();
+
+    assertEquals(1, result);
+  }
+
+  @Test
+  void createAndStoreDistrictLabelsReportsFromDistricts() {
+    District district = createDistrict();
+    district.setContacts(List.of(createDistrictContact(UUID.fromString(district.getDistrictId()))));
+    district.setAddresses(List.of(createDistrictAddress(UUID.fromString(district.getDistrictId()))));
+    when(restService.post(any(), any(), any())).thenReturn(new byte[0]);
+
+    int result = districtReportService.createAndStoreDistrictLabelsReportsFromDistricts(ReportTypeCodes.ADDRESS_LABEL_YE.getCode(), List.of(district), null);
+
+    assertEquals(1, result);
+
+  }
 }