@Test(dataProvider = "simulatedCounterIterations")
public void testCounterWithSimulatedRuns(
int windowLengthInSlots, int[] incrementsPerIteration, long[] expCountsPerIteration) {
// given
SlidingWindowCounter<Object> counter = new SlidingWindowCounter<Object>(windowLengthInSlots);
int numIterations = incrementsPerIteration.length;
for (int i = 0; i < numIterations; i++) {
int numIncrements = incrementsPerIteration[i];
long expCounts = expCountsPerIteration[i];
// Objects are absent if they were zero both this iteration
// and the last -- if only this one, we need to report zero.
boolean expAbsent = ((expCounts == 0) && ((i == 0) || (expCountsPerIteration[i - 1] == 0)));
// given (for this iteration)
for (int j = 0; j < numIncrements; j++) {
counter.incrementCount(ANY_OBJECT);
}
// when (for this iteration)
Map<Object, Long> counts = counter.getCountsThenAdvanceWindow();
// then (for this iteration)
if (expAbsent) {
assertThat(counts).doesNotContainKey(ANY_OBJECT);
} else {
assertThat(counts.get(ANY_OBJECT)).isEqualTo(expCounts);
}
}
}
private void countObjAndAck(Tuple tuple) {
Object obj = tuple.getValue(0);
counter.incrementCount(obj);
collector.ack(tuple);
}
