package com.atlassian.cache.hazelcast;
   
   import com.atlassian.cache.CacheSettings;
   import com.atlassian.cache.CacheSettingsBuilder;
   import com.atlassian.cache.CacheSettingsDefaultsProvider;
   import com.atlassian.cache.memory.MemoryCacheManager;
   
   import com.hazelcast.core.HazelcastInstance;
   
  import static org.mockito.Matchers.anyString;
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.when;
  
  public class HazelcastTestSupport
  {
      public static HazelcastCacheManager  createDistributedFactory(HazelcastInstance node)
      {
          return createDistributedFactory(node, getDefaultsProvider());
      }
  
      public static HazelcastCacheManager createDistributedFactory(HazelcastInstance node, CacheSettings defaultSettings)
      {
          return createDistributedFactory(node, getDefaultsProvider(defaultSettings));
      }
  
      public static HazelcastCacheManager createDistributedFactory(HazelcastInstance node, CacheSettingsDefaultsProvider defaultsProvider)
      {
          HazelcastCacheManager cacheManager = new HazelcastCacheManager(node, new MemoryCacheManager(), defaultsProvider);
          cacheManager.init();
          return cacheManager;
      }
  
      public static CacheSettingsDefaultsProvider getDefaultsProvider()
      {
          return getDefaultsProvider(new CacheSettingsBuilder().build());
      }
  
      public static CacheSettingsDefaultsProvider getDefaultsProvider(CacheSettings defaultSettings)
      {
          CacheSettingsDefaultsProvider defaultsProvider = mock(CacheSettingsDefaultsProvider.class);
          when(defaultsProvider.getDefaults(anyString())).thenReturn(defaultSettings);
  
          return defaultsProvider;
      }
  }