@Test public void writeTaskExceptionDuringWrite() throws IOException { TimerData timerData = new TimerData(); WriteTask writeTask = storageWriter.new WriteTask(timerData, Collections.emptyMap()); doThrow(IOException.class).when(writingChannelManager).write(Matchers.<ExtendedByteBufferOutputStream> anyObject(), Matchers.<Path> anyObject(), Matchers.<WriteReadCompletionRunnable> anyObject()); writeTask.run(); verify(storageIndexingTreeHandler, times(1)).writeFailed(writeTask); verify(extendedByteBufferOutputStream, times(1)).close(); verify(serializerQueue, times(1)).add(serializer); }
@Test public void writeTaskThrowableDuringWrite() throws IOException { TimerData timerData = new TimerData(); WriteTask writeTask = storageWriter.new WriteTask(timerData, Collections.emptyMap()); doThrow(Throwable.class).when(writingChannelManager).write(Matchers.<ExtendedByteBufferOutputStream> anyObject(), Matchers.<Path> anyObject(), Matchers.<WriteReadCompletionRunnable> anyObject()); writeTask.run(); verify(storageIndexingTreeHandler, times(1)).writeFailed(writeTask); verify(extendedByteBufferOutputStream, times(1)).close(); verify(serializerQueue, times(1)).add(serializer); }
try { writingChannelManager.write(extendedByteBufferOutputStream, channelPath, completionRunnable); } catch (IOException e) {
Mockito.when(outputStream.getAllByteBuffers()).thenReturn(buffers); Mockito.when(outputStream.getTotalWriteSize()).thenReturn((long) bytes.length); long position = writingChannelManager.write(outputStream, file, new WriteReadCompletionRunnable(buffers.size()) { @Override public void run() {
long position = writingChannelManager.write(byteBuffer, file, new WriteReadCompletionRunnable() { @Override public void run() {
long position = writingChannelManager.write(byteBuffer, file, new WriteReadCompletionRunnable() { @Override public void run() {