- @Test
- public void testDaceConfigurationPolling_getDcaeConfigurations_null() throws Exception {
- PowerMock.resetAll();
- thrown.expect(CorrelationException.class);
- PowerMockito.mockStatic(MicroServiceConfig.class);
- when(MicroServiceConfig.getServiceAddrInfoFromCBS("holmes-rule-mgmt"))
- .thenReturn("host");
- PowerMock.createMock(DcaeConfigurationParser.class);
- PowerMock.expectPrivate(DcaeConfigurationParser.class, "parse", "host")
- .andReturn(null).anyTimes();
-
- PowerMock.replayAll();
- DcaeConfigurations dcaeConfigurations = Whitebox
- .invokeMethod(daceConfigurationPolling, "getDcaeConfigurations");
- PowerMock.verifyAll();
-
- assertThat(dcaeConfigurations == null, equalTo(true));
+ Whitebox.setInternalState(dcaeConfigurationPolling, "url", "http://127.0.0.1");
+
+ RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
+ List<RuleResult4API> ruleResult4APIList = new ArrayList<RuleResult4API>(){
+ {
+ add(new RuleResult4API());
+ }
+ };
+ ruleQueryListResponse.setCorrelationRules(ruleResult4APIList);
+ ruleQueryListResponse.setTotalCount(ruleResult4APIList.size());
+
+ CloseableHttpClient clientMock = createMock(CloseableHttpClient.class);
+ HttpResponse httpResponseMock = createMock(HttpResponse.class);
+ expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
+ expect(HttpsUtils.get(anyObject(HttpGet.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
+ .andReturn(httpResponseMock);
+ expect(HttpsUtils.extractResponseEntity(httpResponseMock)).andReturn(JSONObject.toJSONString(ruleQueryListResponse));
+ clientMock.close();
+ expectLastCall();
+
+ expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
+ expect(HttpsUtils.delete(anyObject(HttpDelete.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
+ .andReturn(httpResponseMock);
+ clientMock.close();
+ expectLastCall();
+
+ expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
+ expect(HttpsUtils.put(anyObject(HttpPut.class), anyObject(HashMap.class), anyObject(HashMap.class),
+ anyObject(StringEntity.class), anyObject(CloseableHttpClient.class)))
+ .andReturn(httpResponseMock);
+ clientMock.close();
+ expectLastCall();
+
+ StatusLine sl = createMock(StatusLine.class);
+ expect(httpResponseMock.getStatusLine()).andReturn(sl);
+ expect(sl.getStatusCode()).andReturn(200);
+
+ replayAll();
+
+ dcaeConfigurationPolling.run();
+
+ verifyAll();