Browse Source

Polish mock verification and formatting

See gh-10773
pull/10511/merge
Andy Wilkinson 8 years ago
parent
commit
194e03c913
  1. 21
      spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/jdbc/DataSourceInitializerTests.java

21
spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/jdbc/DataSourceInitializerTests.java

@ -45,8 +45,8 @@ public class DataSourceInitializerTests { @@ -45,8 +45,8 @@ public class DataSourceInitializerTests {
@Test
public void initializeEmbeddedByDefault() {
try (HikariDataSource dataSource = createDataSource()) {
DataSourceInitializer initializer = new DataSourceInitializer(
dataSource, new DataSourceProperties());
DataSourceInitializer initializer = new DataSourceInitializer(dataSource,
new DataSourceProperties());
JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
assertThat(initializer.createSchema()).isTrue();
assertNumberOfRows(jdbcTemplate, 0);
@ -60,8 +60,8 @@ public class DataSourceInitializerTests { @@ -60,8 +60,8 @@ public class DataSourceInitializerTests {
try (HikariDataSource dataSource = createDataSource()) {
DataSourceProperties properties = new DataSourceProperties();
properties.setInitializationMode(DataSourceInitializationMode.ALWAYS);
DataSourceInitializer initializer = new DataSourceInitializer(
dataSource, properties);
DataSourceInitializer initializer = new DataSourceInitializer(dataSource,
properties);
JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
assertThat(initializer.createSchema()).isTrue();
assertNumberOfRows(jdbcTemplate, 0);
@ -80,8 +80,8 @@ public class DataSourceInitializerTests { @@ -80,8 +80,8 @@ public class DataSourceInitializerTests {
try (HikariDataSource dataSource = createDataSource()) {
DataSourceProperties properties = new DataSourceProperties();
properties.setInitializationMode(DataSourceInitializationMode.NEVER);
DataSourceInitializer initializer = new DataSourceInitializer(
dataSource, properties);
DataSourceInitializer initializer = new DataSourceInitializer(dataSource,
properties);
assertThat(initializer.createSchema()).isFalse();
}
}
@ -95,15 +95,14 @@ public class DataSourceInitializerTests { @@ -95,15 +95,14 @@ public class DataSourceInitializerTests {
DataSource dataSource = mock(DataSource.class);
given(dataSource.getConnection()).willReturn(connection);
DataSourceInitializer initializer = new DataSourceInitializer(
dataSource, new DataSourceProperties());
DataSourceInitializer initializer = new DataSourceInitializer(dataSource,
new DataSourceProperties());
assertThat(initializer.createSchema()).isFalse();
verify(dataSource.getConnection());
verify(dataSource).getConnection();
}
private HikariDataSource createDataSource() {
return DataSourceBuilder.create()
.type(HikariDataSource.class)
return DataSourceBuilder.create().type(HikariDataSource.class)
.url("jdbc:h2:mem:" + UUID.randomUUID().toString()).build();
}

Loading…
Cancel
Save