@Override public void untake(ConsumerWorkingSet workingSet, List<? extends PartitionKey> partitionKeys) { doExpiry(workingSet); for (PartitionKey key : partitionKeys) { ConsumablePartition consumablePartition = workingSet.lookup(key); // don't need to assertInProgress because untake() already does that consumablePartition.untake(); } }
@Test public void testSimpleTransitions() { // tests simple success case ConsumablePartition partition = new DefaultConsumablePartition(generateUniqueKey()); Assert.assertEquals(0, partition.getNumFailures()); partition.take(); partition.retry(); Assert.assertEquals(1, partition.getNumFailures()); partition.take(); // test that untake doesn't increment failure count partition.untake(); Assert.assertEquals(1, partition.getNumFailures()); partition.take(); partition.complete(); }