DatasetAffinityFunctionWrapper implements AffinityFunction { @Override public void reset() { delegate.reset(); } DatasetAffinityFunctionWrapper(AffinityFunction delegate); @Override void reset(); @Override int partitions(); @Override int partition(Object key); @Override List<List<ClusterNode>> assignPartitions(AffinityFunctionContext affCtx); @Override void removeNode(UUID nodeId);  }
@Test public void testReset() { wrapper.reset(); verify(affinityFunction, times(1)).reset(); }