package com.atlassian.vcache.internal.redis;
   
   import org.junit.rules.TestWatcher;
   import org.junit.runner.Description;
   import redis.clients.jedis.Jedis;
   import redis.clients.jedis.JedisPool;
   import redis.clients.jedis.JedisPoolConfig;
   
   import java.net.URI;
  import java.util.function.Supplier;
  
  public class JedisTester extends TestWatcher {
  
      JedisPool jedisPool;
      URI redisUri;
  
      public JedisTester() {
          this(true);
      }
  
      public JedisTester(boolean online) {
          redisUri = online
                  ? URI.create(System.getProperty("redis.svr"))
                  : URI.create("tcp://localhost:4");
      }
  
      @Override
      protected void starting(Description description) {
          final JedisPoolConfig poolConfig = new JedisPoolConfig();
          poolConfig.setBlockWhenExhausted(false);
          poolConfig.setMaxTotal(1); // only allow one connection to check for connection leaks
          jedisPool = new ThreadAffinityJedisPool(poolConfig, redisUri);
      }
  
      @Override
      protected void finished(Description description) {
          jedisPool.destroy();
      }
  
      Supplier<Jedis> jedisSupplier() {
          return jedisPool::getResource;
      }
  }