package com.atlassian.vcache.internal.redis;
   
   import com.atlassian.marshalling.jdk.JavaSerializationMarshalling;
   import com.atlassian.vcache.DirectExternalCache;
   import com.atlassian.vcache.ExternalCacheSettings;
   import com.atlassian.vcache.StableReadExternalCache;
   import com.atlassian.vcache.internal.RequestContext;
   import com.atlassian.vcache.internal.RequestMetrics;
   import com.atlassian.vcache.internal.core.DefaultRequestContext;
  import com.atlassian.vcache.internal.core.Sha1ExternalCacheKeyGenerator;
  import com.atlassian.vcache.internal.core.metrics.DefaultMetricsCollector;
  import com.atlassian.vcache.internal.core.metrics.MetricsCollector;
  import com.atlassian.vcache.internal.test.AbstractStableReadExternalCacheIT;
  import org.junit.BeforeClass;
  import redis.clients.jedis.JedisPool;
  import redis.clients.jedis.JedisPoolConfig;
  
  import java.io.IOException;
  import java.net.URI;
  import java.net.URISyntaxException;
  import java.time.Duration;
  
  public class RedisStableReadExternalCacheIT extends AbstractStableReadExternalCacheIT {
      private static JedisPool jedisPool;
      private RequestContext requestContext = new DefaultRequestContext("tenant-id");
      private final MetricsCollector metricsCollector = new DefaultMetricsCollector(() -> requestContext);
  
      @BeforeClass
      public static void init_client() throws IOException, URISyntaxException {
          final JedisPoolConfig poolConfig = new JedisPoolConfig();
          poolConfig.setMaxTotal(1); // only allow one connection to check for connection leaks
          jedisPool = new JedisPool(poolConfig, new URI(System.getProperty("redis.svr")));
      }
  
      @Override
      protected StableReadExternalCache<String> createCache(String name, ExternalCacheSettings settings, Duration lockTimeout) {
          return metricsCollector.wrap(
                  new RedisStableReadExternalCache<>(
                          jedisPool::getResource,
                          () -> requestContext,
                          new Sha1ExternalCacheKeyGenerator("redis-test"),
                          name,
                          JavaSerializationMarshalling.pair(String.class),
                          settings,
                          metricsCollector,
                          lockTimeout));
      }
  
      @Override
      protected DirectExternalCache<String> obtainDirectCache(String name, ExternalCacheSettings settings) {
          return new RedisDirectExternalCache<>(
                  jedisPool::getResource,
                  () -> requestContext,
                  new Sha1ExternalCacheKeyGenerator("redis-test"),
                  name,
                  JavaSerializationMarshalling.pair(String.class),
                  settings,
                  Duration.ofSeconds(1));
      }
  
      @Override
      protected RequestMetrics requestMetrics() {
          return metricsCollector.obtainRequestMetrics(requestContext);
      }
  }