package com.atlassian.vcache.internal.core.metrics;
   
   import com.atlassian.vcache.JvmCache;
   import com.atlassian.vcache.LocalCacheOperations;
   import org.junit.Before;
   import org.junit.Test;
   import org.junit.runner.RunWith;
   import org.mockito.Mock;
   import org.mockito.runners.MockitoJUnitRunner;
  
  import java.util.HashSet;
  import java.util.Set;
  
  import static com.atlassian.vcache.internal.MetricLabel.TIMED_GET_KEYS_CALL;
  import static org.hamcrest.Matchers.equalTo;
  import static org.junit.Assert.assertThat;
  import static org.mockito.Matchers.anyLong;
  import static org.mockito.Matchers.eq;
  import static org.mockito.Mockito.verify;
  import static org.mockito.Mockito.verifyNoMoreInteractions;
  import static org.mockito.Mockito.when;
  
  @RunWith(MockitoJUnitRunner.class)
  public class TimedJvmCacheTest extends TimedLocalCacheOperationsTest {
      @Mock
      private JvmCache<String, String> delegate;
  
      private TimedJvmCache<String, String> timed;
  
      @Before
      public void init() {
          when(delegate.getName()).thenReturn(CACHE_NAME);
          timed = new TimedJvmCache<>(delegate, metricsRecorder);
      }
  
  
      @Override
      protected CacheType getCacheType() {
          return CacheType.JVM;
      }
  
      @Override
      protected LocalCacheOperations<String, String> getDelegateOps() {
          return delegate;
      }
  
      @Override
      protected TimedLocalCacheOperations<String, String> getTimed() {
          return timed;
      }
  
      @Test
      public void getKeys() throws Exception {
          final Set<String> result = new HashSet<>();
          when(delegate.getKeys()).thenReturn(result);
  
          final Set<String> passedResult = timed.getKeys();
  
          assertThat(passedResult, equalTo(result));
  
          verify(metricsRecorder).record(eq(CACHE_NAME), eq(getCacheType()), eq(TIMED_GET_KEYS_CALL), anyLong());
          verifyNoMoreInteractions(metricsRecorder);
      }
  }