rulemgt/src/test/java/org/onap/holmes/rulemgt/dcae/DcaeConfigurationPollingTest.java

   1 /**
   2  * Copyright 2017 ZTE Corporation.
   3  * <p>
   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
   7  * <p>
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  * <p>
  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.
  15  */
  16 package org.onap.holmes.rulemgt.dcae;
  17
  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;
  41
  42 import java.util.ArrayList;
  43 import java.util.HashMap;
  44 import java.util.List;
  45 import java.util.Map;
  46
  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.*;
  52
  53 @PrepareForTest({HttpsUtils.class, DcaeConfigurationQuery.class, DcaeConfigurationPolling.class})
  54 @SuppressStaticInitializationFor("org.onap.holmes.common.utils.HttpsUtils")
  55 @RunWith(PowerMockRunner.class)
  56 public class DcaeConfigurationPollingTest {
  57
  58     @org.junit.Rule
  59     public ExpectedException thrown = ExpectedException.none();
  60
  61     @Test
  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");
  70
  71         RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
  72         List<RuleResult4API> ruleResult4APIList = new ArrayList<RuleResult4API>(){
  73             {
  74                 add(new RuleResult4API());
  75             }
  76         };
  77         ruleQueryListResponse.setCorrelationRules(ruleResult4APIList);
  78         ruleQueryListResponse.setTotalCount(ruleResult4APIList.size());
  79         expect(dcaeConfigurationPolling.getAllCorrelationRules()).andReturn(ruleQueryListResponse);
  80
  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);
  86         clientMock.close();
  87         expectLastCall();
  88
  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);
  93         clientMock.close();
  94         expectLastCall();
  95
  96         StatusLine sl = createMock(StatusLine.class);
  97         expect(httpResponseMock.getStatusLine()).andReturn(sl);
  98         expect(sl.getStatusCode()).andReturn(200);
  99
 100         replayAll();
 101
 102         dcaeConfigurationPolling.run();
 103
 104         verifyAll();
 105     }
 106
 107     @Test
 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");
 116
 117         RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
 118         List<RuleResult4API> ruleResult4APIList = new ArrayList<RuleResult4API>(){
 119             {
 120                 add(new RuleResult4API());
 121             }
 122         };
 123         ruleQueryListResponse.setCorrelationRules(ruleResult4APIList);
 124         ruleQueryListResponse.setTotalCount(ruleResult4APIList.size());
 125         expect(dcaeConfigurationPolling.getAllCorrelationRules()).andReturn(ruleQueryListResponse);
 126
 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);
 132         clientMock.close();
 133         expectLastCall();
 134
 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);
 139         clientMock.close();
 140         expectLastCall();
 141
 142         StatusLine sl = createMock(StatusLine.class);
 143         expect(httpResponseMock.getStatusLine()).andReturn(sl);
 144         expect(sl.getStatusCode()).andReturn(200);
 145
 146         replayAll();
 147
 148         dcaeConfigurationPolling.run();
 149         dcaeConfigurationPolling.run();
 150
 151         verifyAll();
 152     }
 153
 154
 155
 156     @Test
 157     public void getAllCorrelationRules() throws Exception {
 158         Map<String, Object> responseObj = new HashMap(){
 159             {
 160                 put("correlationRules", new ArrayList<String>());
 161                 put("totalCount", 0);
 162             }
 163         };
 164
 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));
 171         clientMock.close();
 172         expectLastCall();
 173
 174         replayAll();
 175         DcaeConfigurationPolling daceConfigurationPolling = new DcaeConfigurationPolling("holmes-rule-mgmt");
 176         RuleQueryListResponse response = daceConfigurationPolling.getAllCorrelationRules();
 177         assertThat(response.getTotalCount(), is(0));
 178         verifyAll();
 179     }
 180
 181     @Before
 182     public void setUp() {
 183         mockStatic(HttpsUtils.class);
 184     }
 185
 186     @After
 187     public void tearDown() {
 188         resetAll();
 189     }
 190 }