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;
47 import static org.easymock.EasyMock.anyObject;
48 import static org.easymock.EasyMock.expect;
49 import static org.hamcrest.CoreMatchers.is;
50 import static org.junit.Assert.assertThat;
51 import static org.powermock.api.easymock.PowerMock.*;
53 @PrepareForTest({HttpsUtils.class, DcaeConfigurationQuery.class, DcaeConfigurationPolling.class})
54 @SuppressStaticInitializationFor("org.onap.holmes.common.utils.HttpsUtils")
55 @RunWith(PowerMockRunner.class)
56 public class DcaeConfigurationPollingTest {
59 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 = createPartialMock(DcaeConfigurationPolling.class,
68 "getAllCorrelationRules");
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());
79 expect(dcaeConfigurationPolling.getAllCorrelationRules()).andReturn(ruleQueryListResponse);
81 CloseableHttpClient clientMock = createMock(CloseableHttpClient.class);
82 HttpResponse httpResponseMock = createMock(HttpResponse.class);
83 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
84 expect(HttpsUtils.delete(anyObject(HttpDelete.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
85 .andReturn(httpResponseMock);
89 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
90 expect(HttpsUtils.put(anyObject(HttpPut.class), anyObject(HashMap.class), anyObject(HashMap.class),
91 anyObject(StringEntity.class), anyObject(CloseableHttpClient.class)))
92 .andReturn(httpResponseMock);
96 StatusLine sl = createMock(StatusLine.class);
97 expect(httpResponseMock.getStatusLine()).andReturn(sl);
98 expect(sl.getStatusCode()).andReturn(200);
102 dcaeConfigurationPolling.run();
108 public void run_identical_contents() throws Exception {
109 DcaeConfigurations dcaeConfigurations = new DcaeConfigurations();
110 dcaeConfigurations.addDefaultRule(new Rule("test", "clName", "contents", 1));
111 mockStatic(DcaeConfigurationQuery.class);
112 expect(DcaeConfigurationQuery.getDcaeConfigurations(anyObject(String.class))).andReturn(dcaeConfigurations).times(2);
113 DcaeConfigurationPolling dcaeConfigurationPolling = createPartialMock(DcaeConfigurationPolling.class,
114 "getAllCorrelationRules");
115 Whitebox.setInternalState(dcaeConfigurationPolling, "url", "http://127.0.0.1");
117 RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
118 List<RuleResult4API> ruleResult4APIList = new ArrayList<RuleResult4API>(){
120 add(new RuleResult4API());
123 ruleQueryListResponse.setCorrelationRules(ruleResult4APIList);
124 ruleQueryListResponse.setTotalCount(ruleResult4APIList.size());
125 expect(dcaeConfigurationPolling.getAllCorrelationRules()).andReturn(ruleQueryListResponse);
127 CloseableHttpClient clientMock = createMock(CloseableHttpClient.class);
128 HttpResponse httpResponseMock = createMock(HttpResponse.class);
129 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
130 expect(HttpsUtils.delete(anyObject(HttpDelete.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
131 .andReturn(httpResponseMock);
135 expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
136 expect(HttpsUtils.put(anyObject(HttpPut.class), anyObject(HashMap.class), anyObject(HashMap.class),
137 anyObject(StringEntity.class), anyObject(CloseableHttpClient.class)))
138 .andReturn(httpResponseMock);
142 StatusLine sl = createMock(StatusLine.class);
143 expect(httpResponseMock.getStatusLine()).andReturn(sl);
144 expect(sl.getStatusCode()).andReturn(200);
148 dcaeConfigurationPolling.run();
149 dcaeConfigurationPolling.run();
157 public void getAllCorrelationRules() throws Exception {
158 Map<String, Object> responseObj = new HashMap(){
160 put("correlationRules", new ArrayList<String>());
161 put("totalCount", 0);
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(JSONObject.toJSONString(responseObj));
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() {