package com.atlassian.vcache.internal.memcached;
   
   import com.atlassian.vcache.ChangeRate;
   import com.atlassian.vcache.VCacheFactory;
   import com.atlassian.vcache.internal.RequestContext;
   import com.atlassian.vcache.internal.VCacheLifecycleManager;
   import com.atlassian.vcache.internal.VCacheManagement;
   import com.atlassian.vcache.internal.core.DefaultVCacheCreationHandler;
   import com.atlassian.vcache.internal.core.ThreadLocalRequestContextSupplier;
  import com.atlassian.vcache.internal.test.AbstractVCacheServiceIT;
  import com.atlassian.vcache.internal.test.EmptyVCacheSettingsDefaultsProvider;
  import net.spy.memcached.AddrUtil;
  import net.spy.memcached.BinaryConnectionFactory;
  import net.spy.memcached.MemcachedClient;
  import org.junit.After;
  import org.junit.Before;
  import org.junit.BeforeClass;
  
  import java.io.IOException;
  
  public class MemcachedVCacheServiceIT extends AbstractVCacheServiceIT {
      private static MemcachedClient client;
  
      private final ThreadLocalRequestContextSupplier requestContextSupplier =
              ThreadLocalRequestContextSupplier.strictSupplier();
  
      private MemcachedVCacheService service;
  
      @Before
      public void initService() {
          final MemcachedVCacheServiceSettings settings = new MemcachedVCacheServiceSettingsBuilder()
                  .productIdentifier("confira")
                  .clientSupplier(() -> client)
                  .threadLocalContextSupplier(requestContextSupplier)
                  .workContextContextSupplier(requestContextSupplier)
                  .defaultsProvider(new EmptyVCacheSettingsDefaultsProvider())
                  .creationHandler(new DefaultVCacheCreationHandler(
                          MAX_ENTRIES,
                          MAX_TTL,
                          MAX_ENTRIES,
                          ChangeRate.HIGH_CHANGE,
                          ChangeRate.HIGH_CHANGE))
                  .begunTransactionalActivityHandler(begunTxnActivityHandler)
                  .build();
          service = new MemcachedVCacheService(settings);
      }
  
      @Override
      protected VCacheFactory vCacheFactory() {
          return service;
      }
  
      @Override
      protected VCacheManagement vCacheManagement() {
          return service;
      }
  
      @Override
      protected VCacheLifecycleManager vCacheLifecycleManager() {
          return service;
      }
  
      @Override
      protected RequestContext currentRequestContext() {
          return requestContextSupplier.get();
      }
  
      @Override
      protected void forceNewRequestContext() {
          clearThread();
          initThread();
      }
  
      @BeforeClass
      public static void initClient() throws IOException {
          client = new MemcachedClient(
                  new BinaryConnectionFactory(),
                  AddrUtil.getAddresses(System.getProperty("memcached.svr")));
      }
  
      @Before
      public void initThread() {
          requestContextSupplier.initThread("tenant-123");
      }
  
      @After
      public void clearThread() {
          requestContextSupplier.clearThread();
      }
  }