holmes-actions/src/test/java/org/onap/holmes/common/aai/AaiQuery4Ccvpn2Test.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.onap.holmes.common.aai
   4  * ================================================================================
   5  * Copyright (C) 2018-2022 Huawei, ZTE. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.holmes.common.aai;
  22
  23 import com.google.gson.JsonObject;
  24 import com.google.gson.JsonParser;
  25 import jakarta.ws.rs.core.Response;
  26 import org.easymock.EasyMock;
  27 import org.junit.After;
  28 import org.junit.Before;
  29 import org.junit.BeforeClass;
  30 import org.junit.Test;
  31 import org.junit.runner.RunWith;
  32 import org.onap.holmes.common.aai.config.AaiConfig;
  33 import org.onap.holmes.common.utils.JerseyClient;
  34 import org.powermock.api.easymock.PowerMock;
  35 import org.powermock.core.classloader.annotations.PrepareForTest;
  36 import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
  37 import org.powermock.modules.junit4.PowerMockRunner;
  38 import org.powermock.reflect.Whitebox;
  39
  40 import java.io.*;
  41 import java.util.HashMap;
  42 import java.util.Map;
  43
  44 import static org.hamcrest.MatcherAssert.assertThat;
  45 import static org.hamcrest.core.Is.is;
  46 import static org.hamcrest.core.IsNull.nullValue;
  47 import static org.onap.holmes.common.config.MicroServiceConfig.MSB_ADDR;
  48
  49
  50 @RunWith(PowerMockRunner.class)
  51 @PrepareForTest({JerseyClient.class})
  52 @SuppressStaticInitializationFor("org.onap.holmes.common.utils.JerseyClient")
  53 public class AaiQuery4Ccvpn2Test {
  54     private static JsonObject data;
  55
  56     private static AaiQuery4Ccvpn2 aai = AaiQuery4Ccvpn2.newInstance();
  57
  58     private static Map<String, Object> headers = new HashMap<>();
  59     private static JerseyClient client;
  60     private static Response response;
  61
  62     @BeforeClass
  63     static public void beforeClass() {
  64         System.setProperty(MSB_ADDR, "127.0.0.1:80");
  65
  66         File file = new File(AaiQuery4Ccvpn2Test.class.getClassLoader().getResource("./ccvpn2.data.json").getFile());
  67         BufferedReader reader = null;
  68         try {
  69             reader = new BufferedReader(new FileReader(file));
  70             StringBuilder sb = new StringBuilder();
  71             reader.lines().forEach(l -> sb.append(l));
  72             data = JsonParser.parseString(sb.toString()).getAsJsonObject();
  73         } catch (FileNotFoundException e) {
  74             // Do nothing
  75         } catch (IOException e) {
  76             // Do nothing
  77         } finally {
  78             if (reader != null) {
  79                 try {
  80                     reader.close();
  81                 } catch (IOException e) {
  82                     // Do nothing
  83                 }
  84             }
  85         }
  86
  87         headers.put("X-TransactionId", AaiConfig.X_TRANSACTION_ID);
  88         headers.put("X-FromAppId", AaiConfig.X_FROMAPP_ID);
  89         headers.put("Authorization", AaiConfig.getAuthenticationCredentials());
  90         headers.put("Accept", "application/json");
  91         headers.put("Content-Type", "application/json");
  92         Whitebox.setInternalState(aai, "headers", headers);
  93     }
  94
  95     @Before
  96     public void before() throws Exception {
  97         client = PowerMock.createMock(JerseyClient.class);
  98         response = PowerMock.createMock(Response.class);
  99         PowerMock.expectNew(JerseyClient.class).andReturn(client).anyTimes();
 100     }
 101
 102     @After
 103     public void after() {
 104         PowerMock.resetAll();
 105     }
 106
 107     @Test
 108     public void test_getServiceInstances_exception() {
 109         mockGetMethod();
 110         EasyMock.expect(client.get(EasyMock.anyString())).andReturn(data.get("site-resources").toString());
 111
 112         mockGetMethod();
 113         EasyMock.expect(client.get(EasyMock.anyString())).andReturn(data.get("499hkg9933NNN").toString());
 114
 115         mockGetMethod();
 116         EasyMock.expect(client.get(EasyMock.anyString())).andReturn(data.get("499hkg9933NNN").toString());
 117
 118         PowerMock.replayAll();
 119
 120         aai.getSiteServiceInstance("HkHubONSDEMOBJHKCustomer");
 121
 122         PowerMock.verifyAll();
 123     }
 124
 125     @Test
 126     public void test_getServiceInstancesNull_exception() {
 127         mockGetMethod();
 128         EasyMock.expect(client.get(EasyMock.anyString())).andReturn(data.get("site-resources1").toString());
 129
 130         PowerMock.replayAll();
 131
 132         assertThat(aai.getSiteServiceInstance("HkHubONSDEMOSZHKCustomer"), is(nullValue()));
 133
 134         PowerMock.verifyAll();
 135     }
 136
 137     private void mockGetMethod() {
 138         EasyMock.expect(client.headers(headers)).andReturn(client);
 139         EasyMock.expect(client.path(EasyMock.anyString())).andReturn(client);
 140     }
 141 }