@Before public void defaultStubs() throws Exception { when(client.invoke(any())) .thenReturn( CompletableFuture.completedFuture( InvokeResponse.builder() .payload(ByteBuffer.wrap(mapper.writeValueAsBytes(new TestOutput("value")))) .build())); }
@Override @SneakyThrows public CompletableFuture<InvokeResponse> invoke(final InvokeRequest invokeRequest) { ByteBufferBackedInputStream inputStream = new ByteBufferBackedInputStream(invokeRequest.payload()); ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); handler.handleRequest(inputStream, outputStream, null); return CompletableFuture.completedFuture( InvokeResponse.builder().payload(ByteBuffer.wrap(outputStream.toByteArray())).build()); } }