Method getMock

test/storage/MockDataPoints.java:78–128

Retrieves the DataPoints object and sets up the mock calls @return A DataPoints object for iteration

()

   * @return A DataPoints object for iteration
   */
  public DataPoints getMock() {
    when(dps.metricName()).thenReturn(metric);
    when(dps.metricNameAsync()).thenReturn(Deferred.fromResult(metric));
    when(dps.getTags()).thenReturn(tags);
    when(dps.getTagsAsync()).thenReturn(Deferred.fromResult(tags));
    when(dps.getAggregatedTags()).thenReturn(agg_tags);
    when(dps.getAggregatedTagsAsync()).thenReturn(Deferred.fromResult(agg_tags));
    when(dps.getTSUIDs()).thenReturn(tsuids);
    when(dps.getAnnotations()).thenReturn(annotations);
    when(dps.size()).thenReturn(limit);
    when(dps.aggregatedSize()).thenReturn(limit * 2);
    when(dps.iterator()).thenReturn(it);
    when(dps.getQueryIndex()).thenReturn(0);
    
    // iterator mocking
    when(it.hasNext()).thenAnswer(new Answer<Boolean>() {
      @Override
      public Boolean answer(final InvocationOnMock args) throws Throwable {
        if (value > limit) {
          return false;
        }
        return true;
      }
    });
    when(it.next()).thenAnswer(new Answer<DataPoint>() {
      @Override
      public DataPoint answer(final InvocationOnMock args) throws Throwable {
        value++;
        timestamp += interval;
        return dp;
      }
    });
    doThrow(new RuntimeException("OpenTSDB doesn't support remove"))
      .when(it).remove();

    // data point mocking
    when(dp.timestamp()).thenAnswer(new Answer<Long>() {
      @Override
      public Long answer(final InvocationOnMock args) throws Throwable {
        return timestamp;
      }
    });
    when(dp.isInteger()).thenReturn(true);
    when(dp.longValue()).thenAnswer(new Answer<Long>() {
      @Override
      public Long answer(final InvocationOnMock args) throws Throwable {
        return value;
      }
    });
    return dps;
  }
  
  /**
   * Returns the DataPoint object so that the test can override the mocks.