Polishing

This commit is contained in:
Sam Brannen
2023-09-05 14:52:22 +02:00
parent 552e7fb1d5
commit 0a7166234d
5 changed files with 15 additions and 21 deletions

View File

@@ -154,34 +154,25 @@ class SimpleJdbcInsertIntegrationTests {
protected EmbeddedDatabase embeddedDatabase;
protected abstract String getSchemaScript();
protected abstract String getUsersTableName();
protected EmbeddedDatabase createEmbeddedDatabase() {
return new EmbeddedDatabaseBuilder(new ClassRelativeResourceLoader(DatabasePopulator.class))
@BeforeEach
void createDatabase() {
this.embeddedDatabase = new EmbeddedDatabaseBuilder(new ClassRelativeResourceLoader(DatabasePopulator.class))
.setType(EmbeddedDatabaseType.H2)
.addScript(getSchemaScript())
.addScript("users-data.sql")
.build();
}
@BeforeEach
void checkDatabaseSetup() {
this.embeddedDatabase = createEmbeddedDatabase();
assertNumUsers(1);
}
@AfterEach
void shutdown() {
void shutdownDatabase() {
this.embeddedDatabase.shutdown();
}
protected void assertNumUsers(long count) {
JdbcClient jdbcClient = JdbcClient.create(this.embeddedDatabase);
long numUsers = jdbcClient.sql("select count(*) from " + getUsersTableName()).query().singleValue();
long numUsers = jdbcClient.sql("select count(*) from " + getUsersTableName()).query(Long.class).single();
assertThat(numUsers).isEqualTo(count);
}
@@ -190,6 +181,10 @@ class SimpleJdbcInsertIntegrationTests {
assertNumUsers(2);
}
protected abstract String getSchemaScript();
protected abstract String getUsersTableName();
}
}