apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/wrapper/IuiRouteServiceWrapperTest.java

   1 /*******************************************************************************
   2  * Copyright 2016-2017 ZTE, Inc. and others.
   3  *
   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  *
   7  * http://www.apache.org/licenses/LICENSE-2.0
   8  *
   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 package org.onap.msb.apiroute.wrapper;
  15
  16 import java.lang.reflect.InvocationHandler;
  17 import java.lang.reflect.Method;
  18 import java.util.ArrayList;
  19 import java.util.Collections;
  20 import java.util.Comparator;
  21 import java.util.List;
  22
  23 import org.junit.Assert;
  24 import org.junit.Before;
  25 import org.junit.BeforeClass;
  26 import org.junit.Test;
  27 import org.junit.runner.RunWith;
  28 import org.onap.msb.apiroute.api.IuiRouteInfo;
  29 import org.onap.msb.apiroute.api.RouteServer;
  30 import org.onap.msb.apiroute.api.exception.ExtendedNotFoundException;
  31 import org.onap.msb.apiroute.wrapper.dao.RedisAccessWrapper;
  32 import org.onap.msb.apiroute.wrapper.util.ConfigUtil;
  33 import org.onap.msb.apiroute.wrapper.util.JedisUtil;
  34 import org.onap.msb.apiroute.wrapper.util.RouteUtil;
  35 import org.powermock.api.mockito.PowerMockito;
  36 import org.powermock.core.classloader.annotations.PowerMockIgnore;
  37 import org.powermock.core.classloader.annotations.PrepareForTest;
  38 import org.powermock.modules.junit4.PowerMockRunner;
  39
  40 import com.fiftyonred.mock_jedis.MockJedisPool;
  41
  42 import redis.clients.jedis.JedisPool;
  43 import redis.clients.jedis.JedisPoolConfig;
  44
  45
  46 @RunWith(PowerMockRunner.class)
  47 @PrepareForTest({JedisUtil.class, RouteUtil.class, RedisAccessWrapper.class})
  48 @PowerMockIgnore({"javax.management.*", "jdk.internal.reflect.*"})
  49 public class IuiRouteServiceWrapperTest {
  50     private static IuiRouteServiceWrapper iuiRouteServiceWrapper;
  51     private static Comparator<IuiRouteInfo> iuiRouteComparator = null;
  52
  53     @BeforeClass
  54     public static void setUpBeforeClass() throws Exception {
  55         iuiRouteServiceWrapper = IuiRouteServiceWrapper.getInstance();
  56         iuiRouteComparator = new Comparator<IuiRouteInfo>() {
  57             @Override
  58             public int compare(IuiRouteInfo o1, IuiRouteInfo o2) {
  59                 if (!o1.getServiceName().equals(o2.getServiceName()))
  60                     return (o1.getServiceName()).compareTo(o2.getServiceName());
  61                 return 0;
  62             }
  63         };
  64
  65         PowerMockito.mockStatic(System.class);
  66         PowerMockito.when(System.getenv("ROUTE_WAY")).thenReturn(null);
  67         PowerMockito.when(System.getenv("ROUTE_WAY")).thenReturn("ip|domain");
  68         ConfigUtil.getInstance().initRouteWay();
  69     }
  70
  71     @Before
  72     public void setUpBeforeTest() throws Exception {
  73         final JedisPool mockJedisPool = new MockJedisPool(new JedisPoolConfig(), "localhost");
  74         PowerMockito.mockStatic(JedisUtil.class);
  75         JedisUtil jedisUtil = PowerMockito.mock(JedisUtil.class);
  76         PowerMockito.when(jedisUtil.borrowJedisInstance()).thenReturn(mockJedisPool.getResource());
  77
  78         PowerMockito.replace(PowerMockito.method(RedisAccessWrapper.class, "filterKeys")).with(new InvocationHandler() {
  79             @Override
  80             public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
  81                 return mockJedisPool.getResource().keys((String) args[0]);
  82             }
  83         });
  84     }
  85
  86
  87     @Test
  88     public void test_getIuiRouteInstance_not_exist() {
  89         try {
  90             iuiRouteServiceWrapper.getIuiRouteInstance("testForJunit", "", "", "ip");
  91             Assert.fail("should not process to here.");
  92         } catch (Exception e) {
  93             Assert.assertTrue(e instanceof ExtendedNotFoundException);
  94
  95         }
  96
  97     }
  98
  99     @Test
 100     public void test_getIuiRouteInstance() {
 101
 102         IuiRouteInfo iuirouteInfo = buildIuiRouteInfo();
 103         try {
 104             iuiRouteServiceWrapper.saveIuiRouteInstance4Rest(iuirouteInfo, "ip");
 105             IuiRouteInfo dbIuiRouteInfo = iuiRouteServiceWrapper.getIuiRouteInstance("testiui", "", "", "ip");
 106             Assert.assertEquals(iuirouteInfo, dbIuiRouteInfo);
 107         } catch (Exception e) {
 108             Assert.fail("throw exception means error occured!" + e.getMessage());
 109         }
 110
 111     }
 112
 113     @Test
 114     public void test_getAllIuiRouteInstances() {
 115         IuiRouteInfo iuirouteInfo = buildIuiRouteInfo();
 116         IuiRouteInfo iuirouteInfo2 = buildIuiRouteInfo2();
 117         List<IuiRouteInfo> expected = new ArrayList<>();
 118         expected.add(iuirouteInfo);
 119         expected.add(iuirouteInfo2);
 120         Collections.sort(expected, iuiRouteComparator);
 121
 122         try {
 123             iuiRouteServiceWrapper.saveIuiRouteInstance4Rest(iuirouteInfo, "ip");
 124             iuiRouteServiceWrapper.saveIuiRouteInstance4Rest(iuirouteInfo2, "ip");
 125
 126
 127
 128             PowerMockito.mockStatic(RouteUtil.class);
 129             PowerMockito.when(RouteUtil.getMutiRedisKey(RouteUtil.IUIROUTE, "ip")).thenReturn("msb:routing:iui:*");
 130             List<IuiRouteInfo> iuiRouterList = iuiRouteServiceWrapper.getAllIuiRouteInstances("ip");
 131             Collections.sort(iuiRouterList, iuiRouteComparator);
 132
 133             Assert.assertEquals(expected, iuiRouterList);
 134
 135         } catch (Exception e) {
 136             Assert.fail("throw exception means error occured!" + e.getMessage());
 137         }
 138
 139     }
 140
 141     @Test
 142     public void test_updateIuiRouteStatus() {
 143         IuiRouteInfo iuirouteInfo = buildIuiRouteInfo();
 144         try {
 145             iuiRouteServiceWrapper.saveIuiRouteInstance4Rest(iuirouteInfo, "ip");
 146             IuiRouteInfo dbIuirouteInfo = iuiRouteServiceWrapper.getIuiRouteInstance("testiui", "", "", "ip");
 147             Assert.assertEquals("1", dbIuirouteInfo.getStatus());
 148             iuiRouteServiceWrapper.updateIuiRouteStatus("testiui", "", "", "0", "ip");
 149             dbIuirouteInfo = iuiRouteServiceWrapper.getIuiRouteInstance("testiui", "", "", "ip");
 150             Assert.assertEquals("0", dbIuirouteInfo.getStatus());
 151         } catch (Exception e) {
 152             Assert.fail("throw exception means error occured!" + e.getMessage());
 153         }
 154
 155     }
 156
 157
 158
 159     @Test
 160     public void test_deleteIuiRoute() {
 161         IuiRouteInfo iuirouteInfo2 = buildIuiRouteInfo2();
 162         try {
 163             iuiRouteServiceWrapper.saveIuiRouteInstance4Rest(iuirouteInfo2, "ip");
 164             IuiRouteInfo dbIuirouteInfo = iuiRouteServiceWrapper.getIuiRouteInstance("testiui2", "", "", "ip");
 165             Assert.assertNotNull(dbIuirouteInfo);
 166
 167         } catch (Exception e) {
 168             Assert.fail("throw exception means error occured!" + e.getMessage());
 169         }
 170         try {
 171             iuiRouteServiceWrapper.deleteIuiRoute("testiui2", "", "", "ip");
 172             iuiRouteServiceWrapper.getIuiRouteInstance("testiui2", "", "", "ip");
 173         } catch (Exception e) {
 174             Assert.assertTrue(e instanceof ExtendedNotFoundException);
 175         }
 176     }
 177
 178
 179     private IuiRouteInfo buildIuiRouteInfo() {
 180         IuiRouteInfo iuirouteInfo = new IuiRouteInfo();
 181         iuirouteInfo.setServiceName("testiui");
 182         iuirouteInfo.setStatus("1");
 183         iuirouteInfo.setUrl("/iui/testiui");
 184         iuirouteInfo.setUseOwnUpstream("0");
 185         iuirouteInfo.setVisualRange("0");
 186         iuirouteInfo.setEnable_ssl(false);
 187         RouteServer[] servers = new RouteServer[] {new RouteServer("10.74.148.88", "8080")};
 188         iuirouteInfo.setServers(servers);
 189         return iuirouteInfo;
 190     }
 191
 192     private IuiRouteInfo buildIuiRouteInfo2() {
 193         IuiRouteInfo iuirouteInfo = new IuiRouteInfo();
 194         iuirouteInfo.setServiceName("testiui2");
 195         iuirouteInfo.setStatus("1");
 196         iuirouteInfo.setUrl("/iui/testiui");
 197         iuirouteInfo.setUseOwnUpstream("0");
 198         iuirouteInfo.setVisualRange("1");
 199         iuirouteInfo.setEnable_ssl(true);
 200         RouteServer[] servers = new RouteServer[] {new RouteServer("10.74.148.89", "8080")};
 201         iuirouteInfo.setServers(servers);
 202         return iuirouteInfo;
 203     }
 204
 205
 206 }