@Test
public void testAbortWithCohorts() throws Exception {

    doReturn(true).when(mockShardDataTree).startAbort(cohort);
    doReturn(Optional.of(CompletableFuture.completedFuture(null))).when(mockUserCohorts).abort();
    final Future<?> abortFuture = abort(cohort);
    abortFuture.get();
    verify(mockShardDataTree).startAbort(cohort);
}