BAEL-7187 implemented (#15208)
* BAEL-7139 implemented * BAEL-7187 implemented
This commit is contained in:
parent
f8e9ed2925
commit
8532499dcc
@ -0,0 +1,126 @@
|
|||||||
|
package com.baeldung.exceptions_completablefuture;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.concurrent.CompletableFuture;
|
||||||
|
import java.util.concurrent.CountDownLatch;
|
||||||
|
import java.util.concurrent.ExecutionException;
|
||||||
|
import java.util.stream.Stream;
|
||||||
|
|
||||||
|
import org.assertj.core.api.Assertions;
|
||||||
|
import org.junit.jupiter.params.ParameterizedTest;
|
||||||
|
import org.junit.jupiter.params.provider.Arguments;
|
||||||
|
import org.junit.jupiter.params.provider.MethodSource;
|
||||||
|
|
||||||
|
public class CompletableFutureExceptionsHandlingUnitTest {
|
||||||
|
|
||||||
|
@ParameterizedTest
|
||||||
|
@MethodSource("parametersSource_handle")
|
||||||
|
void whenCompletableFutureIsScheduled_thenHandleStageIsAlwaysInvoked(int radius, long expected)
|
||||||
|
throws ExecutionException, InterruptedException {
|
||||||
|
long actual = CompletableFuture
|
||||||
|
.supplyAsync(() -> {
|
||||||
|
if (radius <= 0) {
|
||||||
|
throw new IllegalArgumentException("Supplied with non-positive radius '%d'");
|
||||||
|
}
|
||||||
|
return Math.round(Math.pow(radius, 2) * Math.PI);
|
||||||
|
})
|
||||||
|
.handle((result, ex) -> {
|
||||||
|
if (ex == null) {
|
||||||
|
return result;
|
||||||
|
} else {
|
||||||
|
return -1L;
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.get();
|
||||||
|
|
||||||
|
Assertions.assertThat(actual).isEqualTo(expected);
|
||||||
|
}
|
||||||
|
|
||||||
|
@ParameterizedTest
|
||||||
|
@MethodSource("parametersSource_exceptionally")
|
||||||
|
void whenCompletableFutureIsScheduled_thenExceptionallyExecutedOnlyOnFailure(int a, int b, int c, long expected)
|
||||||
|
throws ExecutionException, InterruptedException {
|
||||||
|
long actual = CompletableFuture
|
||||||
|
.supplyAsync(() -> {
|
||||||
|
if (a <= 0 || b <= 0 || c <= 0) {
|
||||||
|
throw new IllegalArgumentException(String.format("Supplied with incorrect edge length [%s]", List.of(a, b, c)));
|
||||||
|
}
|
||||||
|
return a * b * c;
|
||||||
|
})
|
||||||
|
.exceptionally((ex) -> -1)
|
||||||
|
.get();
|
||||||
|
|
||||||
|
Assertions.assertThat(actual).isEqualTo(expected);
|
||||||
|
}
|
||||||
|
|
||||||
|
@ParameterizedTest
|
||||||
|
@MethodSource("parametersSource_exceptionally")
|
||||||
|
void givenCompletableFutureIsScheduled_whenHandleIsAlreadyPresent_thenExceptionallyIsNotExecuted(int a, int b, int c, long expected)
|
||||||
|
throws ExecutionException, InterruptedException {
|
||||||
|
long actual = CompletableFuture
|
||||||
|
.supplyAsync(() -> {
|
||||||
|
if (a <= 0 || b <= 0 || c <= 0) {
|
||||||
|
throw new IllegalArgumentException(String.format("Supplied with incorrect edge length [%s]", List.of(a, b, c)));
|
||||||
|
}
|
||||||
|
return a * b * c;
|
||||||
|
})
|
||||||
|
.handle((result, throwable) -> {
|
||||||
|
if (throwable != null) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
})
|
||||||
|
.exceptionally((ex) -> {
|
||||||
|
System.exit(1);
|
||||||
|
return 0;
|
||||||
|
})
|
||||||
|
.get();
|
||||||
|
|
||||||
|
Assertions.assertThat(actual).isEqualTo(expected);
|
||||||
|
}
|
||||||
|
|
||||||
|
@ParameterizedTest
|
||||||
|
@MethodSource("parametersSource_whenComplete")
|
||||||
|
void whenCompletableFutureIsScheduled_thenWhenCompletedExecutedAlways(Double a, long expected, Class<Exception> ifAny) {
|
||||||
|
try {
|
||||||
|
CountDownLatch countDownLatch = new CountDownLatch(1);
|
||||||
|
|
||||||
|
long actual = CompletableFuture
|
||||||
|
.supplyAsync(() -> {
|
||||||
|
if (a.isNaN()) {
|
||||||
|
throw new IllegalArgumentException("Supplied value is NaN");
|
||||||
|
}
|
||||||
|
return Math.round(Math.pow(a, 2));
|
||||||
|
})
|
||||||
|
.whenComplete((result, exception) -> countDownLatch.countDown())
|
||||||
|
.get();
|
||||||
|
|
||||||
|
Assertions.assertThat(countDownLatch.await(20L, java.util.concurrent.TimeUnit.SECONDS));
|
||||||
|
Assertions.assertThat(actual).isEqualTo(expected);
|
||||||
|
} catch (Exception e) {
|
||||||
|
Assertions.assertThat(e.getClass()).isSameAs(ExecutionException.class);
|
||||||
|
Assertions.assertThat(e.getCause().getClass()).isSameAs(ifAny);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static Stream<Arguments> parametersSource_handle() {
|
||||||
|
return Stream.of(
|
||||||
|
Arguments.of(1, 3),
|
||||||
|
Arguments.of(-1, -1)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
static Stream<Arguments> parametersSource_exceptionally() {
|
||||||
|
return Stream.of(
|
||||||
|
Arguments.of(1, 5, 5, 25),
|
||||||
|
Arguments.of(-1, 10, 15, -1)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
static Stream<Arguments> parametersSource_whenComplete() {
|
||||||
|
return Stream.of(
|
||||||
|
Arguments.of(2d, 4, null),
|
||||||
|
Arguments.of(Double.NaN, 1, IllegalArgumentException.class)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user