From 6356852e2f97e0c58be105e5561328efe05013ab Mon Sep 17 00:00:00 2001 From: Stephane Nicoll Date: Mon, 29 Jul 2019 11:18:07 +0200 Subject: [PATCH] Polish "Log health check failure with Reactive health indicators" See gh-17635 --- .../AbstractReactiveHealthIndicatorTests.java | 92 ---------------- .../health/ReactiveHealthIndicatorTests.java | 103 ++++++++++++++++++ 2 files changed, 103 insertions(+), 92 deletions(-) delete mode 100644 spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/AbstractReactiveHealthIndicatorTests.java create mode 100644 spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/ReactiveHealthIndicatorTests.java diff --git a/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/AbstractReactiveHealthIndicatorTests.java b/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/AbstractReactiveHealthIndicatorTests.java deleted file mode 100644 index bae95261bfe..00000000000 --- a/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/AbstractReactiveHealthIndicatorTests.java +++ /dev/null @@ -1,92 +0,0 @@ -/* - * Copyright 2012-2019 the original author or authors. - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * https://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -package org.springframework.boot.actuate.health; - -import org.junit.jupiter.api.Test; -import org.junit.jupiter.api.extension.ExtendWith; -import reactor.core.publisher.Mono; - -import org.springframework.boot.actuate.health.Health.Builder; -import org.springframework.boot.test.system.CapturedOutput; -import org.springframework.boot.test.system.OutputCaptureExtension; - -import static org.assertj.core.api.Assertions.assertThat; - -/** - * Tests for {@link AbstractReactiveHealthIndicator}. - * - * @author Dmytro Nosan - */ -@ExtendWith(OutputCaptureExtension.class) -class AbstractReactiveHealthIndicatorTests { - - @Test - void doHealth() { - Health health = new SimpleReactiveHealthIndicator().health().block(); - assertThat(health).isEqualTo(Health.up().build()); - } - - @Test - void doHealthCustomErrorMessage(CapturedOutput output) { - Health health = new ErrorMessageReactiveHealthIndicator().health().block(); - assertThat(health).isEqualTo(Health.down(new UnsupportedOperationException()).build()); - assertThat(output).contains("Unsupported Operation"); - } - - @Test - void doHealthCustomErrorMessageFunction(CapturedOutput output) { - Health health = new CustomErrorFunctionReactiveHealthIndicator().health().block(); - assertThat(health).isEqualTo(Health.down(new RuntimeException()).build()); - assertThat(output).contains("Runtime Exception"); - } - - private static final class SimpleReactiveHealthIndicator extends AbstractReactiveHealthIndicator { - - @Override - protected Mono doHealthCheck(Builder builder) { - return Mono.just(builder.up().build()); - } - - } - - private static final class ErrorMessageReactiveHealthIndicator extends AbstractReactiveHealthIndicator { - - ErrorMessageReactiveHealthIndicator() { - super("Unsupported Operation"); - } - - @Override - protected Mono doHealthCheck(Builder builder) { - return Mono.error(new UnsupportedOperationException()); - } - - } - - private static final class CustomErrorFunctionReactiveHealthIndicator extends AbstractReactiveHealthIndicator { - - CustomErrorFunctionReactiveHealthIndicator() { - super((ex) -> "Runtime Exception"); - } - - @Override - protected Mono doHealthCheck(Builder builder) { - throw new RuntimeException(); - } - - } - -} diff --git a/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/ReactiveHealthIndicatorTests.java b/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/ReactiveHealthIndicatorTests.java new file mode 100644 index 00000000000..fcd4da3a7ca --- /dev/null +++ b/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/ReactiveHealthIndicatorTests.java @@ -0,0 +1,103 @@ +/* + * Copyright 2012-2019 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.boot.actuate.health; + +import org.junit.Rule; +import org.junit.Test; +import reactor.core.publisher.Mono; +import reactor.test.StepVerifier; + +import org.springframework.boot.actuate.health.Health.Builder; +import org.springframework.boot.test.rule.OutputCapture; + +import static org.assertj.core.api.Assertions.assertThat; + +/** + * Tests for {@link AbstractReactiveHealthIndicator}. + * + * @author Dmytro Nosan + * @author Stephane Nicoll + */ +public class ReactiveHealthIndicatorTests { + + @Rule + public OutputCapture output = new OutputCapture(); + + @Test + public void healthUp() { + StepVerifier.create(new SimpleReactiveHealthIndicator().health()) + .consumeNextWith((health) -> assertThat(health).isEqualTo(Health.up().build())).verifyComplete(); + assertThat(this.output.toString()).doesNotContain("Health check failed for simple"); + } + + @Test + public void healthDownWithCustomErrorMessage() { + StepVerifier.create(new CustomErrorMessageReactiveHealthIndicator().health()).consumeNextWith( + (health) -> assertThat(health).isEqualTo(Health.down(new UnsupportedOperationException()).build())) + .verifyComplete(); + assertThat(this.output.toString()).contains("Health check failed for custom"); + } + + @Test + public void healthDownWithCustomErrorMessageFunction() { + StepVerifier.create(new CustomErrorMessageFunctionReactiveHealthIndicator().health()) + .consumeNextWith((health) -> assertThat(health).isEqualTo(Health.down(new RuntimeException()).build())) + .verifyComplete(); + assertThat(this.output.toString()).contains("Health check failed with RuntimeException"); + } + + private static final class SimpleReactiveHealthIndicator extends AbstractReactiveHealthIndicator { + + SimpleReactiveHealthIndicator() { + super("Health check failed for simple"); + } + + @Override + protected Mono doHealthCheck(Builder builder) { + return Mono.just(builder.up().build()); + } + + } + + private static final class CustomErrorMessageReactiveHealthIndicator extends AbstractReactiveHealthIndicator { + + CustomErrorMessageReactiveHealthIndicator() { + super("Health check failed for custom"); + } + + @Override + protected Mono doHealthCheck(Builder builder) { + return Mono.error(new UnsupportedOperationException()); + } + + } + + private static final class CustomErrorMessageFunctionReactiveHealthIndicator + extends AbstractReactiveHealthIndicator { + + CustomErrorMessageFunctionReactiveHealthIndicator() { + super((ex) -> "Health check failed with " + ex.getClass().getSimpleName()); + } + + @Override + protected Mono doHealthCheck(Builder builder) { + throw new RuntimeException(); + } + + } + +}