2 * Copyright 2017 ZTE Corporation.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 package org.onap.holmes.rulemgt.dcae;
18 import com.alibaba.fastjson.JSONObject;
19 import org.apache.http.HttpResponse;
20 import org.apache.http.StatusLine;
21 import org.apache.http.client.methods.HttpDelete;
22 import org.apache.http.client.methods.HttpGet;
23 import org.apache.http.client.methods.HttpPut;
24 import org.apache.http.entity.StringEntity;
25 import org.apache.http.impl.client.CloseableHttpClient;
26 import org.junit.After;
27 import org.junit.Before;
28 import org.junit.Test;
29 import org.junit.rules.ExpectedException;
30 import org.junit.runner.RunWith;
31 import org.onap.holmes.common.dcae.DcaeConfigurationQuery;
32 import org.onap.holmes.common.dcae.entity.DcaeConfigurations;
33 import org.onap.holmes.common.dcae.entity.Rule;
34 import org.onap.holmes.common.utils.HttpsUtils;
35 import org.onap.holmes.rulemgt.bean.response.RuleQueryListResponse;
36 import org.onap.holmes.rulemgt.bean.response.RuleResult4API;
37 import org.powermock.core.classloader.annotations.PrepareForTest;
38 import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
39 import org.powermock.modules.junit4.PowerMockRunner;
40 import org.powermock.reflect.Whitebox;
42 import java.util.ArrayList;
43 import java.util.HashMap;
44 import java.util.List;
46 import static org.easymock.EasyMock.anyObject;
47 import static org.easymock.EasyMock.expect;
48 import static org.hamcrest.CoreMatchers.is;
49 import static org.junit.Assert.assertThat;
50 import static org.powermock.api.easymock.PowerMock.*;
52 @PrepareForTest({HttpsUtils.class, DcaeConfigurationQuery.class})
53 @SuppressStaticInitializationFor("org.onap.holmes.common.utils.HttpsUtils")
54 @RunWith(PowerMockRunner.class)
55 public class DcaeConfigurationPollingTest {
58 public ExpectedException thrown = ExpectedException.none();
62 public void run() throws Exception {
63 DcaeConfigurations dcaeConfigurations = new DcaeConfigurations();
64 dcaeConfigurations.addDefaultRule(new Rule("test", "clName", "contents", 1));
65 mockStatic(DcaeConfigurationQuery.class);
66 expect(DcaeConfigurationQuery.getDcaeConfigurations(anyObject(String.class))).andReturn(dcaeConfigurations);
67 DcaeConfigurationPolling dcaeConfigurationPolling = new DcaeConfigurationPolling("localhost");
69 Whitebox.setInternalState(dcaeConfigurationPolling, "url", "http://127.0.0.1");
71 RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
72 List<RuleResult4API> ruleResult4APIList = new ArrayList<RuleResult4API>(){
74 add(new RuleResult4API());
77 ruleQueryListResponse.setCorrelationRules(ruleResult4APIList);
78 ruleQueryListResponse.setTotalCount(ruleResult4APIList.size());
80 CloseableHttpClient clientMock = createMock(CloseableHttpClient.class);
81 HttpResponse httpResponseMock = createMock(HttpResponse.class);
82 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
83 expect(HttpsUtils.get(anyObject(HttpGet.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
84 .andReturn(httpResponseMock);
85 expect(HttpsUtils.extractResponseEntity(httpResponseMock)).andReturn(JSONObject.toJSONString(ruleQueryListResponse));
89 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
90 expect(HttpsUtils.delete(anyObject(HttpDelete.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
91 .andReturn(httpResponseMock);
95 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
96 expect(HttpsUtils.put(anyObject(HttpPut.class), anyObject(HashMap.class), anyObject(HashMap.class),
97 anyObject(StringEntity.class), anyObject(CloseableHttpClient.class)))
98 .andReturn(httpResponseMock);
102 StatusLine sl = createMock(StatusLine.class);
103 expect(httpResponseMock.getStatusLine()).andReturn(sl);
104 expect(sl.getStatusCode()).andReturn(200);
108 dcaeConfigurationPolling.run();
114 public void run_identical_contents() throws Exception {
115 DcaeConfigurations dcaeConfigurations = new DcaeConfigurations();
116 dcaeConfigurations.addDefaultRule(new Rule("test", "clName", "contents", 1));
117 mockStatic(DcaeConfigurationQuery.class);
118 expect(DcaeConfigurationQuery.getDcaeConfigurations(anyObject(String.class))).andReturn(dcaeConfigurations).times(2);
119 DcaeConfigurationPolling dcaeConfigurationPolling = createPartialMock(DcaeConfigurationPolling.class,
120 "getAllCorrelationRules");
121 Whitebox.setInternalState(dcaeConfigurationPolling, "url", "http://127.0.0.1");
123 RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
124 List<RuleResult4API> ruleResult4APIList = new ArrayList<RuleResult4API>(){
126 add(new RuleResult4API());
129 ruleQueryListResponse.setCorrelationRules(ruleResult4APIList);
130 ruleQueryListResponse.setTotalCount(ruleResult4APIList.size());
131 expect(dcaeConfigurationPolling.getAllCorrelationRules()).andReturn(ruleQueryListResponse);
133 CloseableHttpClient clientMock = createMock(CloseableHttpClient.class);
134 HttpResponse httpResponseMock = createMock(HttpResponse.class);
135 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
136 expect(HttpsUtils.delete(anyObject(HttpDelete.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
137 .andReturn(httpResponseMock);
141 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
142 expect(HttpsUtils.put(anyObject(HttpPut.class), anyObject(HashMap.class), anyObject(HashMap.class),
143 anyObject(StringEntity.class), anyObject(CloseableHttpClient.class)))
144 .andReturn(httpResponseMock);
148 StatusLine sl = createMock(StatusLine.class);
149 expect(httpResponseMock.getStatusLine()).andReturn(sl);
150 expect(sl.getStatusCode()).andReturn(200);
154 dcaeConfigurationPolling.run();
155 dcaeConfigurationPolling.run();
163 public void getAllCorrelationRules() throws Exception {
165 CloseableHttpClient clientMock = createMock(CloseableHttpClient.class);
166 HttpResponse httpResponseMock = createMock(HttpResponse.class);
167 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
168 expect(HttpsUtils.get(anyObject(HttpGet.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
169 .andReturn(httpResponseMock);
170 expect(HttpsUtils.extractResponseEntity(httpResponseMock)).andReturn("{\"correlationRules\": [], \"totalCount\": 0}");
175 DcaeConfigurationPolling daceConfigurationPolling = new DcaeConfigurationPolling("holmes-rule-mgmt");
176 RuleQueryListResponse response = daceConfigurationPolling.getAllCorrelationRules();
177 assertThat(response.getTotalCount(), is(0));
182 public void setUp() {
183 mockStatic(HttpsUtils.class);
187 public void tearDown() {