Method mockUID

test/core/BaseTsdbTest.java:295–383

Mocks out the UID calls to match keys and values @param type The type of UID to deal with @param key The String name of the UID @param uid The byte array UID to pair up with

(final UniqueIdType type, final String key,
      final byte[] uid)

   *          The byte array UID to pair up with
   */
  protected void mockUID(final UniqueIdType type, final String key,
      final byte[] uid) {
    switch (type) {
    case METRIC:
      when(metrics.getId(key)).thenReturn(uid);
      when(metrics.getIdAsync(key))
        .thenAnswer(new Answer<Deferred<byte[]>>() {
          @Override
          public Deferred<byte[]> answer(InvocationOnMock invocation)
              throws Throwable {
            return Deferred.fromResult(uid);
          }
        });
      when(metrics.getOrCreateId(key)).thenReturn(uid);
      when(metrics.getOrCreateIdAsync(key))
          .thenAnswer(new Answer<Deferred<byte[]>>() {
        @Override
        public Deferred<byte[]> answer(InvocationOnMock invocation)
            throws Throwable {
          return Deferred.fromResult(uid);
        }
      });
      when(metrics.getName(uid)).thenReturn(key);
      when(metrics.getNameAsync(uid)).thenAnswer(new Answer<Deferred<String>>() {
        @Override
        public Deferred<String> answer(InvocationOnMock invocation)
            throws Throwable {
          return Deferred.fromResult(key);
        }
      });
      break;
    case TAGK:
      when(tag_names.getId(key)).thenReturn(uid);
      when(tag_names.getIdAsync(key))
        .thenAnswer(new Answer<Deferred<byte[]>>() {
          @Override
          public Deferred<byte[]> answer(InvocationOnMock invocation)
              throws Throwable {
            return Deferred.fromResult(uid);
          }
        });
      when(tag_names.getOrCreateId(key)).thenReturn(uid);
      when(tag_names.getOrCreateIdAsync(key))
          .thenAnswer(new Answer<Deferred<byte[]>>() {
        @Override
        public Deferred<byte[]> answer(InvocationOnMock invocation)
            throws Throwable {
          return Deferred.fromResult(uid);
        }
      });
      when(tag_names.getName(uid)).thenReturn(key);
      when(tag_names.getNameAsync(uid)).thenAnswer(new Answer<Deferred<String>>() {
        @Override
        public Deferred<String> answer(InvocationOnMock invocation)
            throws Throwable {
          return Deferred.fromResult(key);
        }
      });