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
  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.*;
  51
  52 @PrepareForTest({HttpsUtils.class, DcaeConfigurationQuery.class})
  53 @SuppressStaticInitializationFor("org.onap.holmes.common.utils.HttpsUtils")
  54 @RunWith(PowerMockRunner.class)
  55 public class DcaeConfigurationPollingTest {
  56
  57     @org.junit.Rule
  58     public ExpectedException thrown = ExpectedException.none();
  59
  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 = new DcaeConfigurationPolling("localhost");
  68
  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
  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));
  86         clientMock.close();
  87         expectLastCall();
  88
  89         expect(HttpsUtils.getHttpClient(30000)).andReturn(clientMock);
  90         expect(HttpsUtils.delete(anyObject(HttpDelete.class), anyObject(HashMap.class), anyObject(CloseableHttpClient.class)))
  91                 .andReturn(httpResponseMock);
  92         clientMock.close();
  93         expectLastCall();
  94
  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);
  99         clientMock.close();
 100         expectLastCall();
 101
 102         StatusLine sl = createMock(StatusLine.class);
 103         expect(httpResponseMock.getStatusLine()).andReturn(sl);
 104         expect(sl.getStatusCode()).andReturn(200);
 105
 106         replayAll();
 107
 108         dcaeConfigurationPolling.run();
 109
 110         verifyAll();
 111     }
 112
 113     @Test
 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");
 122
 123         RuleQueryListResponse ruleQueryListResponse = new RuleQueryListResponse();
 124         List<RuleResult4API> ruleResult4APIList = new ArrayList<RuleResult4API>(){
 125             {
 126                 add(new RuleResult4API());
 127             }
 128         };
 129         ruleQueryListResponse.setCorrelationRules(ruleResult4APIList);
 130         ruleQueryListResponse.setTotalCount(ruleResult4APIList.size());
 131         expect(dcaeConfigurationPolling.getAllCorrelationRules()).andReturn(ruleQueryListResponse);
 132
 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);
 138         clientMock.close();
 139         expectLastCall();
 140
 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);
 145         clientMock.close();
 146         expectLastCall();
 147
 148         StatusLine sl = createMock(StatusLine.class);
 149         expect(httpResponseMock.getStatusLine()).andReturn(sl);
 150         expect(sl.getStatusCode()).andReturn(200);
 151
 152         replayAll();
 153
 154         dcaeConfigurationPolling.run();
 155         dcaeConfigurationPolling.run();
 156
 157         verifyAll();
 158     }
 159
 160
 161
 162     @Test
 163     public void getAllCorrelationRules() throws Exception {
 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("{\"correlationRules\": [], \"totalCount\": 0}");
 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 }