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

   1 /**
   2  * Copyright 2020 - 2021 Fujitsu, ZTE Limited.
   3  * <p>
   4  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
   5  * in compliance with the License. You may obtain a copy of the License at
   6  * <p>
   7  * http://www.apache.org/licenses/LICENSE-2.0
   8  * <p>
   9  * Unless required by applicable law or agreed to in writing, software distributed under the License
  10  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  11  * or implied. See the License for the specific language governing permissions and limitations under
  12  * the License.
  13  */
  14
  15 package org.onap.holmes.common.aai;
  16
  17 import com.google.gson.JsonObject;
  18 import com.google.gson.JsonParser;
  19 import org.easymock.EasyMock;
  20 import org.junit.*;
  21 import org.junit.rules.ExpectedException;
  22 import org.junit.runner.RunWith;
  23 import org.onap.holmes.common.aai.config.AaiConfig;
  24 import org.onap.holmes.common.config.MicroServiceConfig;
  25 import org.onap.holmes.common.utils.JerseyClient;
  26 import org.powermock.api.easymock.PowerMock;
  27 import org.powermock.core.classloader.annotations.PowerMockIgnore;
  28 import org.powermock.core.classloader.annotations.PrepareForTest;
  29 import org.powermock.modules.junit4.PowerMockRunner;
  30 import org.powermock.reflect.Whitebox;
  31
  32 import javax.ws.rs.client.Entity;
  33 import javax.ws.rs.core.MultivaluedHashMap;
  34 import javax.ws.rs.core.MultivaluedMap;
  35 import java.io.*;
  36 import java.util.HashMap;
  37 import java.util.Map;
  38
  39 import static org.easymock.EasyMock.anyObject;
  40 import static org.junit.Assert.assertEquals;
  41 import static org.onap.holmes.common.config.MicroServiceConfig.MSB_ADDR;
  42
  43 @RunWith(PowerMockRunner.class)
  44 @PowerMockIgnore("javax.net.ssl.*")
  45 @PrepareForTest({AaiQueryMdons.class, MicroServiceConfig.class, JerseyClient.class})
  46 public class AaiQueryMdonsTest {
  47
  48     @Rule
  49     public ExpectedException thrown = ExpectedException.none();
  50
  51     private static AaiQueryMdons aaiMdonsQuery = AaiQueryMdons.newInstance();
  52     private static MultivaluedMap<String, Object> headers = new MultivaluedHashMap<>();
  53
  54     private static JsonObject data;
  55
  56     private static final String AAI_ADDR = "https://aai.onap:8443/aai/v19/";
  57
  58     @BeforeClass
  59     public static void beforeClass() {
  60         System.setProperty(MSB_ADDR, "127.0.0.1:80");
  61         System.setProperty("ENABLE_ENCRYPT", "true");
  62
  63         File file = new File(AaiQueryMdonsTest.class.getClassLoader().getResource("./aai-mdons.json").getFile());
  64         BufferedReader reader = null;
  65         try {
  66             reader = new BufferedReader(new FileReader(file));
  67             StringBuilder sb = new StringBuilder();
  68             reader.lines().forEach(l -> sb.append(l));
  69             data = JsonParser.parseString(sb.toString()).getAsJsonObject();
  70         } catch (FileNotFoundException e) {
  71             // Do nothing
  72         } catch (IOException e) {
  73             // Do nothing
  74         } finally {
  75             if (reader != null) {
  76                 try {
  77                     reader.close();
  78                 } catch (IOException e) {
  79                     // Do nothing
  80                 }
  81             }
  82         }
  83
  84         headers.add("X-TransactionId", AaiConfig.X_TRANSACTION_ID);
  85         headers.add("X-FromAppId", AaiConfig.X_FROMAPP_ID);
  86         headers.add("Authorization", AaiConfig.getAuthenticationCredentials());
  87         headers.add("Accept", "application/json");
  88         headers.add("Content-Type", "application/json");
  89
  90     }
  91
  92     @Before
  93     public void before() {
  94         Whitebox.setInternalState(aaiMdonsQuery, "headers", headers);
  95
  96     }
  97
  98     @After
  99     public void after() {
 100         PowerMock.resetAll();
 101     }
 102
 103     @Test
 104     public void testProcessPnf() throws Exception {
 105         String pnfUrl = AAI_ADDR + "network/pnfs/pnf/test1?depth=all";
 106         String domainService = AAI_ADDR
 107                 + "business/customers/customer/Orange/service-subscriptions/service-subscription/MDONS_OTN/service-instances/service-instance/789";
 108         PowerMock.resetAll();
 109
 110         aaiMdonsQuery = PowerMock.createPartialMock(AaiQueryMdons.class, "getResponse");
 111         PowerMock.expectPrivate(aaiMdonsQuery, "getResponse", pnfUrl).andReturn(data.get("pnf-depth-all").toString());
 112         PowerMock.expectPrivate(aaiMdonsQuery, "getResponse", domainService)
 113                 .andReturn(data.get("get-domain-service").toString());
 114
 115         PowerMock.replayAll();
 116         Map<String, String> accessMap = aaiMdonsQuery.processPnf("test1");
 117         PowerMock.verifyAll();
 118         Map<String, String> verifyMap = new HashMap<>();
 119         verifyMap.put("123", "access-service");
 120         assertEquals(accessMap, verifyMap);
 121     }
 122
 123     @Test
 124     public void testGetPnfName() throws Exception {
 125         String pnfUrl = AAI_ADDR + "network/pnfs?pnf-id=test1-id";
 126
 127         Whitebox.setInternalState(aaiMdonsQuery, "headers", headers);
 128
 129         aaiMdonsQuery = PowerMock.createPartialMock(AaiQueryMdons.class, "getResponse");
 130         PowerMock.expectPrivate(aaiMdonsQuery, "getResponse", pnfUrl).andReturn(data.get("get-pnf-by-id").toString());
 131         PowerMock.replayAll();
 132         String pnfName = Whitebox.invokeMethod(aaiMdonsQuery, "getPnfNameFromPnfId", "test1-id");
 133
 134         PowerMock.verifyAll();
 135         assertEquals(pnfName, "test1");
 136
 137     }
 138
 139     @Test
 140     public void testUpdatelinksAccess() throws Exception {
 141         Map<String, String> accessMap = new HashMap<>();
 142         accessMap.put("123", "access-service");
 143         String accessService = AAI_ADDR
 144                 + "business/customers/customer/Orange/service-subscriptions/service-subscription/MDONS_OTN/service-instances/service-instance/123";
 145         String linkUrl = AAI_ADDR + "network/logical-links/logical-link/link1";
 146
 147         String response =
 148                 "{\"link-name\":\"link1\",\"in-maint\":false,\"link-type\":\"inter-domain\",\"available-capacity\":\"ODU2\",\"resource-version\":\"1584338211407\",\"operational-status\":\"down\"}";
 149
 150         aaiMdonsQuery = PowerMock.createPartialMock(AaiQueryMdons.class, "getResponse");
 151
 152         JerseyClient mockedClient = PowerMock.createMock(JerseyClient.class);
 153         PowerMock.expectNew(JerseyClient.class).andReturn(mockedClient);
 154         EasyMock.expect(mockedClient.headers(anyObject(Map.class))).andReturn(mockedClient);
 155         EasyMock.expect(mockedClient.put(anyObject(String.class), anyObject(Entity.class))).andReturn("");
 156
 157         PowerMock.expectPrivate(aaiMdonsQuery, "getResponse", accessService)
 158                 .andReturn(data.get("get-access-service").toString());
 159         PowerMock.expectPrivate(aaiMdonsQuery, "getResponse", linkUrl).andReturn(data.get("get-inter-link").toString());
 160
 161         PowerMock.replayAll();
 162         Whitebox.invokeMethod(aaiMdonsQuery, "updateLinksForAccessService", accessMap);
 163
 164         PowerMock.verifyAll();
 165
 166     }
 167
 168 }