@Test(expected = IllegalStateException.class) public void testAlreadyCompletedTransition() { // cannot complete a partition that has already been completed ConsumablePartition partition = new DefaultConsumablePartition(generateUniqueKey()); partition.take(); partition.complete(); partition.complete(); }
@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(); }