@Test
public void testDestroyForSuccessfulJob() {
// setup
DestroyerContext context = new DestroyerContext(null, true, null, user);
when(KiteDatasetExecutor.listTemporaryDatasetUris(toJobConfig.toJobConfig.uri))
.thenReturn(expectedUris);
// exercise
destroyer.destroy(context, linkConfig, toJobConfig);
// verify
for (String uri : expectedUris) {
verify(executorMock, times(1)).mergeDataset(uri);
}
}
@Test
public void testDestroyForFailedJob() {
// setup
DestroyerContext context = new DestroyerContext(null, false, null, user);
when(KiteDatasetExecutor.listTemporaryDatasetUris(toJobConfig.toJobConfig.uri))
.thenReturn(expectedUris);
for (String uri : expectedUris) {
when(Datasets.delete(uri)).thenReturn(true);
}
// exercise
destroyer.destroy(context, linkConfig, toJobConfig);
// verify
for (String uri : expectedUris) {
verifyStatic(times(1));
Datasets.delete(uri);
}
}
