private void assertPrepareCleanup(CountingCollectStage stage) { assertEquals(1, stage.getPrepareCount()); assertEquals(1, stage.getCleanupCount()); }