Commit 745ff883 authored by Stephane Nicoll's avatar Stephane Nicoll

Merge pull request #15055 from Aerlant

* pr/15055:
  Polish "Add validation query to DataSourceHealthIndicator details"
  Add validation query to DataSourceHealthIndicator details
parents 9686ae4f 9e84abc4
......@@ -115,6 +115,7 @@ public class DataSourceHealthIndicator extends AbstractHealthIndicator
new SingleColumnRowMapper());
Object result = DataAccessUtils.requiredSingleResult(results);
builder.withDetail("result", result);
builder.withDetail("validationQuery", validationQuery);
}
}
......
......@@ -26,6 +26,7 @@ import org.junit.Test;
import org.springframework.boot.actuate.health.Health;
import org.springframework.boot.actuate.health.Status;
import org.springframework.boot.jdbc.DatabaseDriver;
import org.springframework.boot.jdbc.EmbeddedDatabaseConnection;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.datasource.SingleConnectionDataSource;
......@@ -68,6 +69,8 @@ public class DataSourceHealthIndicatorTests {
Health health = this.indicator.health();
assertThat(health.getDetails().get("database")).isNotNull();
assertThat(health.getDetails().get("result")).isNotNull();
assertThat(health.getDetails().get("validationQuery"))
.isEqualTo(DatabaseDriver.HSQLDB.getValidationQuery());
}
@Test
......@@ -75,12 +78,14 @@ public class DataSourceHealthIndicatorTests {
this.indicator.setDataSource(this.dataSource);
new JdbcTemplate(this.dataSource)
.execute("CREATE TABLE FOO (id INTEGER IDENTITY PRIMARY KEY)");
this.indicator.setQuery("SELECT COUNT(*) from FOO");
String customValidationQuery = "SELECT COUNT(*) from FOO";
this.indicator.setQuery(customValidationQuery);
Health health = this.indicator.health();
System.err.println(health);
assertThat(health.getDetails().get("database")).isNotNull();
assertThat(health.getStatus()).isEqualTo(Status.UP);
assertThat(health.getDetails().get("result")).isNotNull();
assertThat(health.getDetails().get("validationQuery"))
.isEqualTo(customValidationQuery);
}
@Test
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment