Polish

6 years ago · 82ea3b51b3
1 changed files with 10 additions and 11 deletions
--- 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,10 +53,9 @@ class FlywayEndpointTests {
				@@ -53,10 +53,9 @@ class FlywayEndpointTests {
 	}

 	@Test
-	@SuppressWarnings("deprecation")
 	void whenFlywayHasBeenBaselinedFlywayReportIsProduced() {
-		this.contextRunner.withBean(FlywayMigrationStrategy.class, () -> (flyway) -> {
-			flyway.setBaselineVersionAsString("2");
+		this.contextRunner.withPropertyValues("spring.flyway.baseline-version=2")
+				.withBean(FlywayMigrationStrategy.class, () -> (flyway) -> {
 					flyway.baseline();
 					flyway.migrate();
 				}).run((context) -> {