Improve Test coverage of DFC

[nonrtric/plt/ranpm.git] / datafilecollector / src / test / java / org / oran / datafile / controllers / StatusControllerTest.java

diff --git a/datafilecollector/src/test/java/org/oran/datafile/controllers/StatusControllerTest.java b/datafilecollector/src/test/java/org/oran/datafile/controllers/StatusControllerTest.java
index 1826096..39d4dc1 100644 (file)
--- a/datafilecollector/src/test/java/org/oran/datafile/controllers/StatusControllerTest.java
+++ b/datafilecollector/src/test/java/org/oran/datafile/controllers/StatusControllerTest.java
@@ -21,6 +21,7 @@
 
 package org.oran.datafile.controllers;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockito.Mockito.doReturn;
 
@@ -32,11 +33,12 @@ import org.mockito.junit.jupiter.MockitoExtension;
 import org.oran.datafile.model.Counters;
 import org.oran.datafile.tasks.CollectAndReportFiles;
 import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatusCode;
 import org.springframework.http.ResponseEntity;
 import reactor.core.publisher.Mono;
 
 @ExtendWith(MockitoExtension.class)
-public class StatusControllerTest {
+class StatusControllerTest {
     @Mock
     CollectAndReportFiles scheduledTasksMock;
 
@@ -48,7 +50,7 @@ public class StatusControllerTest {
     }
 
     @Test
-    public void heartbeat_success() {
+    void heartbeat_success() {
         HttpHeaders httpHeaders = new HttpHeaders();
 
         Mono<ResponseEntity<String>> result = controllerUnderTest.heartbeat(httpHeaders);
@@ -58,7 +60,7 @@ public class StatusControllerTest {
     }
 
     @Test
-    public void status() {
+    void status() {
         Counters counters = new Counters();
         doReturn(counters).when(scheduledTasksMock).getCounters();
 
@@ -67,6 +69,8 @@ public class StatusControllerTest {
         Mono<ResponseEntity<String>> result = controllerUnderTest.status(httpHeaders);
 
         String body = result.block().getBody();
+        HttpStatusCode httpStatusCode = result.block().getStatusCode();
+        assertEquals(200, httpStatusCode.value());
         System.out.println(body);
     }