GRAD2-2949 - Adding test case.

bcgov · Nov 18, 2024 · 719fd37 · 719fd37
1 parent 5e36999
commit 719fd37
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/src/test/java/ca/bc/gov/educ/api/trax/controller/DistrictControllerTest.java b/api/src/test/java/ca/bc/gov/educ/api/trax/controller/DistrictControllerTest.java
@@ -13,6 +13,7 @@
 import org.mockito.Mockito;
 import org.mockito.junit.MockitoJUnitRunner;
 import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 
 import java.util.ArrayList;
@@ -90,7 +91,7 @@ public void whenGetDistrictDetailsByDistNo_ReturnNULL() {
 
         districtControllerV2.getDistrictDetailsByDistNo(distNo);
         Mockito.verify(districtServiceV2, never()).getDistrictByDistNoFromRedisCache(distNo);
-        Assertions.assertEquals(null, districtControllerV2.getDistrictDetailsByDistNo(distNo));
+        Assertions.assertEquals(new ResponseEntity<>(HttpStatus.NOT_FOUND), districtControllerV2.getDistrictDetailsByDistNo(distNo));
     }
 
     @Test