diff --git a/Plan/common/src/test/java/com/djrapitops/plan/gathering/ShutdownSaveTest.java b/Plan/common/src/test/java/com/djrapitops/plan/gathering/ShutdownSaveTest.java index 2a5ce6448..6f594f88a 100644 --- a/Plan/common/src/test/java/com/djrapitops/plan/gathering/ShutdownSaveTest.java +++ b/Plan/common/src/test/java/com/djrapitops/plan/gathering/ShutdownSaveTest.java @@ -119,7 +119,8 @@ class ShutdownSaveTest { @Test void sessionsAreNotSavedOnReload() { shutdownStatus = false; - underTest.performSave(); + Optional> 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> 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());