From 82ea3b51b3a58a0ec68bd6ca44b96368d33c115d Mon Sep 17 00:00:00 2001 From: Andy Wilkinson Date: Fri, 30 Aug 2019 08:26:12 +0100 Subject: [PATCH] Polish --- .../actuate/flyway/FlywayEndpointTests.java | 21 +++++++++---------- 1 file changed, 10 insertions(+), 11 deletions(-) diff --git a/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/flyway/FlywayEndpointTests.java b/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/flyway/FlywayEndpointTests.java index 3e92e000621..7a943253a96 100644 --- a/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/flyway/FlywayEndpointTests.java +++ b/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/flyway/FlywayEndpointTests.java @@ -53,18 +53,17 @@ class FlywayEndpointTests { } @Test - @SuppressWarnings("deprecation") void whenFlywayHasBeenBaselinedFlywayReportIsProduced() { - this.contextRunner.withBean(FlywayMigrationStrategy.class, () -> (flyway) -> { - flyway.setBaselineVersionAsString("2"); - flyway.baseline(); - flyway.migrate(); - }).run((context) -> { - Map flywayBeans = context.getBean(FlywayEndpoint.class).flywayBeans() - .getContexts().get(context.getId()).getFlywayBeans(); - assertThat(flywayBeans).hasSize(1); - assertThat(flywayBeans.values().iterator().next().getMigrations()).hasSize(3); - }); + this.contextRunner.withPropertyValues("spring.flyway.baseline-version=2") + .withBean(FlywayMigrationStrategy.class, () -> (flyway) -> { + flyway.baseline(); + flyway.migrate(); + }).run((context) -> { + Map flywayBeans = context.getBean(FlywayEndpoint.class).flywayBeans() + .getContexts().get(context.getId()).getFlywayBeans(); + assertThat(flywayBeans).hasSize(1); + assertThat(flywayBeans.values().iterator().next().getMigrations()).hasSize(3); + }); } }