@Test public void testBatchResultsStream() throws Exception { SalesforceConnector connector = new SalesforceConnector(); BatchInfo batchInfo = setupBulkConnection(connector); BulkConnection bulkConnection = connector.getBulkConnection(); InputStream stream = Mockito.mock(InputStream.class); when(bulkConnection.getBatchResultStream(batchInfo.getJobId(), batchInfo.getId())) .thenReturn(stream); InputStream actualIs = connector.batchResultStream(batchInfo); assertEquals(stream, actualIs); }