apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/health/RedisHealthCheckTest.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.health;
  15
  16 import org.junit.Before;
  17 import org.junit.Test;
  18 import org.junit.runner.RunWith;
  19 import org.onap.msb.apiroute.wrapper.dao.RedisAccessWrapper;
  20 import org.onap.msb.apiroute.wrapper.util.JedisUtil;
  21 import org.powermock.api.mockito.PowerMockito;
  22 import org.powermock.core.classloader.annotations.PowerMockIgnore;
  23 import org.powermock.core.classloader.annotations.PrepareForTest;
  24 import org.powermock.modules.junit4.PowerMockRunner;
  25 import org.slf4j.Logger;
  26 import org.slf4j.LoggerFactory;
  27
  28
  29 import com.codahale.metrics.health.HealthCheck.Result;
  30 import com.fiftyonred.mock_jedis.MockJedisPool;
  31
  32 import redis.clients.jedis.JedisPool;
  33 import redis.clients.jedis.JedisPoolConfig;
  34
  35 @RunWith(PowerMockRunner.class)
  36 @PrepareForTest({JedisUtil.class, RedisAccessWrapper.class})
  37 @PowerMockIgnore({"javax.management.*", "jdk.internal.reflect.*"})
  38 public class RedisHealthCheckTest {
  39     private static final Logger LOGGER = LoggerFactory.getLogger(RedisHealthCheckTest.class);
  40
  41     @Before
  42     public void setUpBeforeTest() throws Exception {
  43
  44     }
  45
  46     @SuppressWarnings("static-access")
  47     @Test
  48     public void testchecksuccess() {
  49
  50         try {
  51             final JedisPool mockJedisPool = new MockJedisPool(new JedisPoolConfig(), "localhost");
  52             PowerMockito.mockStatic(JedisUtil.class);
  53             JedisUtil jedisUtil = PowerMockito.mock(JedisUtil.class);
  54             PowerMockito.when(jedisUtil.borrowJedisInstance()).thenReturn(mockJedisPool.getResource());
  55
  56             RedisHealthCheck check = new RedisHealthCheck();
  57             Result rst = check.execute();
  58
  59             if (!rst.isHealthy()) {
  60                 LOGGER.warn("testchecksuccess health check failed:" + rst.getMessage());
  61             } else {
  62                 LOGGER.debug(" testchecksuccess health");
  63             }
  64
  65         } catch (Exception e) {
  66             // TODO Auto-generated catch block
  67             e.printStackTrace();
  68         }
  69     }
  70
  71     @Test
  72     public void testcheckfailed() {
  73         RedisHealthCheck check = new RedisHealthCheck();
  74         Result rst = check.execute();
  75
  76         if (!rst.isHealthy()) {
  77             LOGGER.warn("testcheckfailed health check failed:" + rst.getMessage());
  78         } else {
  79             LOGGER.debug("testcheckfailed health");
  80         }
  81
  82     }
  83 }