/** Test the getDatasetIdByName method. */
@Test
public void testGetDatasetIdByName() throws Exception {
final HistoryUtils historyUtils = new HistoryUtils();
final HistoriesClient historiesClientMock = Mockito.mock(HistoriesClient.class);
final String historyId = "history-id";
final String datasetId = "dataset-id";
final String datasetName = "dataset-name";
final HistoryContents historyContents = new HistoryContents();
historyContents.setId(datasetId);
historyContents.setName(datasetName);
Mockito.when(historiesClientMock.showHistoryContents(historyId))
.thenReturn(Arrays.asList(historyContents));
assertEquals(
datasetId, historyUtils.getDatasetIdByName(datasetName, historiesClientMock, historyId));
}
/** Test the downloadDataset method. */
@Test
public void testDownloadDataset() throws Exception {
final HistoryUtils historyUtilsSpy = Mockito.spy(new HistoryUtils());
final HistoriesClient historiesClient = Mockito.mock(HistoriesClient.class);
final String historyId = "123456";
final String datasetId = "dataset-id";
final Dataset dataset = new Dataset();
final GalaxyInstance galaxyInstanceMock = Mockito.mock(GalaxyInstance.class);
final WebResource webResourceMock = Mockito.mock(WebResource.class);
final String filePath = Paths.get("tmp", "testDownloadDataset.txt").toString();
Mockito.when(historiesClient.showDataset(Mockito.eq(historyId), Mockito.eq(datasetId)))
.thenReturn(dataset);
Mockito.when(galaxyInstanceMock.getGalaxyUrl()).thenReturn("http://");
Mockito.when(galaxyInstanceMock.getWebResource()).thenReturn(webResourceMock);
Mockito.when(webResourceMock.path(Mockito.eq("histories"))).thenReturn(webResourceMock);
Mockito.when(webResourceMock.path(Mockito.eq(historyId))).thenReturn(webResourceMock);
Mockito.when(webResourceMock.path(Mockito.eq("contents"))).thenReturn(webResourceMock);
Mockito.when(webResourceMock.path(Mockito.eq(datasetId))).thenReturn(webResourceMock);
Mockito.when(webResourceMock.path(Mockito.eq("display"))).thenReturn(webResourceMock);
Mockito.when(webResourceMock.get(Mockito.eq(File.class))).thenReturn(new File(filePath));
final Answer getFileAnswer =
new Answer() {
@Override
public Object answer(final InvocationOnMock invocationOnMock) throws Throwable {
FileUtils.createFile(filePath, "testDownloadDataset");
return new File(filePath);
}
};
Mockito.when(webResourceMock.get(Mockito.eq(File.class))).thenAnswer(getFileAnswer);
assertTrue(
historyUtilsSpy.downloadDataset(
galaxyInstanceMock, historiesClient, historyId, datasetId, filePath));
assertTrue(new File(filePath).delete());
}
