summaryrefslogtreecommitdiffstats
path: root/main/src/test/java
diff options
context:
space:
mode:
Diffstat (limited to 'main/src/test/java')
-rw-r--r--main/src/test/java/org/onap/policy/pap/main/rest/TestPolicyComponentsHealthCheckProvider.java36
1 files changed, 35 insertions, 1 deletions
diff --git a/main/src/test/java/org/onap/policy/pap/main/rest/TestPolicyComponentsHealthCheckProvider.java b/main/src/test/java/org/onap/policy/pap/main/rest/TestPolicyComponentsHealthCheckProvider.java
index 3f4ecd32..f27a8d32 100644
--- a/main/src/test/java/org/onap/policy/pap/main/rest/TestPolicyComponentsHealthCheckProvider.java
+++ b/main/src/test/java/org/onap/policy/pap/main/rest/TestPolicyComponentsHealthCheckProvider.java
@@ -2,7 +2,7 @@
* ============LICENSE_START=======================================================
* Copyright (C) 2020 Nordix Foundation.
* Modifications Copyright (C) 2020-2021 AT&T Corp.
- * Modifications Copyright (C) 2020 Bell Canada. All rights reserved.
+ * Modifications Copyright (C) 2020-2021 Bell Canada. All rights reserved.
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -30,6 +30,7 @@ import static org.mockito.Mockito.when;
import java.io.File;
import java.net.HttpURLConnection;
+import java.util.Arrays;
import java.util.List;
import java.util.Map;
import javax.ws.rs.core.Response;
@@ -89,11 +90,17 @@ public class TestPolicyComponentsHealthCheckProvider {
private HttpClient client2;
@Mock
+ private HttpClient client3;
+
+ @Mock
private Response response1;
@Mock
private Response response2;
+ @Mock
+ private Response response3;
+
private List<PdpGroup> groups;
private PapParameterGroup savedPapParameterGroup;
@@ -133,10 +140,17 @@ public class TestPolicyComponentsHealthCheckProvider {
when(response2.readEntity(HealthCheckReport.class)).thenReturn(createReport(HttpURLConnection.HTTP_OK, true));
when(client2.get()).thenReturn(response2);
+ when(client3.getName()).thenReturn("dmaap");
+ when(client3.getBaseUrl()).thenReturn("message-router");
+ when(response3.getStatus()).thenReturn(HttpURLConnection.HTTP_OK);
+ when(response3.readEntity(DmaapGetTopicResponse.class)).thenReturn(createDmaapResponse());
+ when(client3.get()).thenReturn(response3);
+
PapParameterGroup papParameterGroup = ParameterService.get(PAP_GROUP_PARAMS_NAME);
List<RestClientParameters> params = papParameterGroup.getHealthCheckRestClientParameters();
when(clientFactory.build(params.get(0))).thenReturn(client1);
when(clientFactory.build(params.get(1))).thenReturn(client2);
+ when(clientFactory.build(params.get(2))).thenReturn(client3);
PolicyComponentsHealthCheckProvider.initializeClientHealthCheckExecutorService(papParameterGroup,
clientFactory);
@@ -180,6 +194,20 @@ public class TestPolicyComponentsHealthCheckProvider {
assertFalse((Boolean) result2.get(HEALTHY));
HealthCheckReport report2 = (HealthCheckReport) result.get(CLIENT_1);
assertFalse(report2.isHealthy());
+
+ when(response3.getStatus()).thenReturn(HttpURLConnection.HTTP_INTERNAL_ERROR);
+ when(response3.readEntity(DmaapGetTopicResponse.class)).thenReturn(null);
+ Map<String, Object> result3 = callFetchPolicyComponentsHealthStatus();
+ assertFalse((Boolean) result3.get(HEALTHY));
+ HealthCheckReport report3 = (HealthCheckReport) result3.get("dmaap");
+ assertFalse(report3.isHealthy());
+
+ when(response3.getStatus()).thenReturn(HttpURLConnection.HTTP_OK);
+ when(response3.readEntity(DmaapGetTopicResponse.class)).thenReturn(new DmaapGetTopicResponse());
+ Map<String, Object> result4 = callFetchPolicyComponentsHealthStatus();
+ assertFalse((Boolean) result4.get(HEALTHY));
+ HealthCheckReport report4 = (HealthCheckReport) result4.get("dmaap");
+ assertFalse(report4.isHealthy());
}
@SuppressWarnings("unchecked")
@@ -234,4 +262,10 @@ public class TestPolicyComponentsHealthCheckProvider {
throw new RuntimeException(e);
}
}
+
+ private DmaapGetTopicResponse createDmaapResponse() {
+ DmaapGetTopicResponse response = new DmaapGetTopicResponse();
+ response.setTopics(Arrays.asList(PapConstants.TOPIC_POLICY_PDP_PAP));
+ return response;
+ }
}