Fix ShutdownSaveTest

This commit is contained in:
Aurora Lahtela 2024-02-17 09:11:37 +02:00
parent 2daf3943b7
commit bede36957b
1 changed files with 4 additions and 2 deletions

View File

@ -119,7 +119,8 @@ class ShutdownSaveTest {
@Test
void sessionsAreNotSavedOnReload() {
shutdownStatus = false;
underTest.performSave();
Optional<Future<?>> future = underTest.performSave();
assertTrue(future.isEmpty());
database.init();
assertTrue(database.query(SessionQueries.fetchAllSessions()).isEmpty());
@ -127,10 +128,11 @@ class ShutdownSaveTest {
}
@Test
void sessionsAreSavedOnServerShutdown() {
void sessionsAreSavedOnServerShutdown() throws Exception {
shutdownStatus = true;
Optional<Future<?>> save = underTest.performSave();
assertTrue(save.isPresent());
save.get().get(); // Wait for save to be done, test fails without.
database.init();
assertFalse(database.query(SessionQueries.fetchAllSessions()).isEmpty());