@Override protected void onSourceConsumed(InputStream fetchedSource) { if (consumptionMark.compareAndSet(false, true)) { source.onConsumed(fetchedSource); } }
@Override public void onConsumed(File file) { try { if (!tempStorage.delete()) { LOGGER.warn("Could not delete temporary file {}", tempStorage); } } finally { inputStreamSource.onConsumed(inputStream); } }
@Test(timeout = DEFAULT_CONVERSION_TIMEOUT) public void testInputStreamSourceToInputStreamConsumerExecute() throws Exception { File source = validFile(true); InputStream inputStream = spy(new FileInputStream(source)); IInputStreamSource inputStreamSource = mock(IInputStreamSource.class); when(inputStreamSource.getInputStream()).thenReturn(inputStream); OutputStream outputStream = mock(OutputStream.class); IInputStreamConsumer inputStreamConsumer = mock(IInputStreamConsumer.class); doAnswer(new CloseStreamAnswer()).when(inputStreamConsumer).onComplete(any(InputStream.class)); assertTrue(getConverter().convert(inputStreamSource).as(validInputType()).to(inputStreamConsumer).as(validTargetType()).execute()); assertTrue(source.exists()); verify(inputStreamSource, times(1)).getInputStream(); verify(inputStreamSource, times(1)).onConsumed(any(InputStream.class)); verify(inputStream, never()).close(); inputStream.close(); verify(inputStreamConsumer, times(1)).onComplete(any(InputStream.class)); verifyNoMoreInteractions(inputStreamConsumer); verify(outputStream, never()).close(); }
@Test(timeout = DEFAULT_CONVERSION_TIMEOUT) public void testInputStreamSourceToInputStreamConsumerFuture() throws Exception { File source = validFile(true); InputStream inputStream = spy(new FileInputStream(source)); IInputStreamSource inputStreamSource = mock(IInputStreamSource.class); when(inputStreamSource.getInputStream()).thenReturn(inputStream); OutputStream outputStream = mock(OutputStream.class); IInputStreamConsumer inputStreamConsumer = mock(IInputStreamConsumer.class); doAnswer(new CloseStreamAnswer()).when(inputStreamConsumer).onComplete(any(InputStream.class)); assertTrue(getConverter().convert(inputStreamSource).as(validInputType()).to(inputStreamConsumer).as(validTargetType()).schedule().get()); assertTrue(source.exists()); verify(inputStreamSource, times(1)).getInputStream(); verify(inputStreamSource, times(1)).onConsumed(any(InputStream.class)); verify(inputStream, never()).close(); inputStream.close(); verify(inputStreamConsumer, times(1)).onComplete(any(InputStream.class)); verifyNoMoreInteractions(inputStreamConsumer); verify(outputStream, never()).close(); }