Add unit tests

CMSgov · Nov 19, 2024 · 6073972 · 6073972
1 parent ae49bd9
commit 6073972
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/worker/src/main/java/gov/cms/ab2d/worker/processor/coverage/CoverageDriverImpl.java b/worker/src/main/java/gov/cms/ab2d/worker/processor/coverage/CoverageDriverImpl.java
@@ -555,15 +555,15 @@ public int numberOfBeneficiariesToProcess(Job job, ContractDTO contract) {
     @Trace(metricName = "EnrollmentLoadFromDB", dispatcher = true)
     @Override
     public CoveragePagingResult pageCoverage(Job job, ContractDTO contract) {
+        ZonedDateTime now = getEndDateTime();
 
         if (contract == null) {
             throw new CoverageDriverException("cannot retrieve metadata for job missing contract");
         }
 
+        ZonedDateTime startDateTime = getStartDateTime(contract);
         Optional<String> additionalDetails = Optional.empty();
         try {
-            ZonedDateTime now = getEndDateTime();
-            ZonedDateTime startDateTime = getStartDateTime(contract);
             // Check that all coverage periods necessary are present before beginning to page
             while (startDateTime.isBefore(now)) {
                 additionalDetails = Optional.of(String.format("contract='%s' month='%s', year='%s'",

diff --git a/worker/src/test/java/gov/cms/ab2d/worker/processor/coverage/CoverageDriverUnitTest.java b/worker/src/test/java/gov/cms/ab2d/worker/processor/coverage/CoverageDriverUnitTest.java
@@ -234,7 +234,7 @@ void pageRequestWhenSinceDateAfterNow() {
 
     @DisplayName("Paging coverage fails when coverage periods are missing")
     @Test
-    void failPagingWhenCoveragePeriodMissing() {
+    void failPagingWhenCoveragePeriodMissing(CapturedOutput output) {
 
         when(coverageService.getCoveragePeriod(any(), anyInt(), anyInt())).thenThrow(new EntityNotFoundException());
 
@@ -243,6 +243,8 @@ void failPagingWhenCoveragePeriodMissing() {
 
         CoverageDriverException startDateInFuture = assertThrows(CoverageDriverException.class, () -> driver.pageCoverage(job, contract));
         assertEquals(EntityNotFoundException.class, startDateInFuture.getCause().getClass());
+        assertTrue(output.getOut().contains("coverage period missing or year,month query incorrect, driver should have resolved earlier - contract='null' month='1', year='2020'"));
+
     }
 
     @DisplayName("Paging coverage periods")