diff --git a/spring-webmvc/src/test/java/org/springframework/web/servlet/mvc/method/annotation/DeferredResultReturnValueHandlerTests.java b/spring-webmvc/src/test/java/org/springframework/web/servlet/mvc/method/annotation/DeferredResultReturnValueHandlerTests.java index 872498c6cf..397da63985 100644 --- a/spring-webmvc/src/test/java/org/springframework/web/servlet/mvc/method/annotation/DeferredResultReturnValueHandlerTests.java +++ b/spring-webmvc/src/test/java/org/springframework/web/servlet/mvc/method/annotation/DeferredResultReturnValueHandlerTests.java @@ -96,8 +96,8 @@ public class DeferredResultReturnValueHandlerTests { @Test public void completableFuture() throws Exception { - SettableListenableFuture future = new SettableListenableFuture<>(); - testHandle(future, CompletableFuture.class, () -> future.set("foo"), "foo"); + CompletableFuture future = new CompletableFuture<>(); + testHandle(future, CompletableFuture.class, () -> future.complete("foo"), "foo"); } @Test @@ -115,9 +115,9 @@ public class DeferredResultReturnValueHandlerTests { @Test public void completableFutureWithError() throws Exception { - SettableListenableFuture future = new SettableListenableFuture<>(); + CompletableFuture future = new CompletableFuture<>(); IllegalStateException ex = new IllegalStateException(); - testHandle(future, CompletableFuture.class, () -> future.setException(ex), ex); + testHandle(future, CompletableFuture.class, () -> future.completeExceptionally(ex), ex); }