@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); } }