package com.atlassian.cache.ehcache;
   
   import com.atlassian.cache.CacheLoader;
   import com.atlassian.cache.ManagedCache;
   import com.atlassian.cache.impl.WeakSupplier;
   import java.util.function.Supplier;
   import org.junit.Before;
   import org.junit.Test;
   
  import static org.junit.Assert.assertNotNull;
  
  public class EhCacheManagerTest {
  
      private EhCacheManagerHelper ehCacheManager;
  
      @Before
      public void setup() {
          ehCacheManager = new EhCacheManagerHelper();
      }
  
      @Test
      public void testShouldNotReturnNullCache() {
          final String cacheName = "test";
          ehCacheManager.getCache(cacheName, new CacheLoader<String, String>() {
  
              @Override
              public String load(String key) {
                  return "value";
              }
          });
  
          WeakSupplier<ManagedCache> supplier = (WeakSupplier<ManagedCache>)ehCacheManager.getSupplier(cacheName);
          supplier.clear();
  
          assertNotNull("returned cache item should not be null", ehCacheManager.getCache(cacheName));
      }
  
      /**
       * Exposes the weak supplier of a cache so that we can clear the referent.
       */
      private static class EhCacheManagerHelper extends EhCacheManager {
  
          public Supplier<ManagedCache> getSupplier(String cacheName) {
              return this.caches.get(cacheName);
          }
      }
  
  }