From bede36957b8ab03abb64b00e028e49b44e0c9b15 Mon Sep 17 00:00:00 2001
From: Aurora Lahtela <24460436+AuroraLS3@users.noreply.github.com>
Date: Sat, 17 Feb 2024 09:11:37 +0200
Subject: [PATCH] Fix ShutdownSaveTest

---
 .../com/djrapitops/plan/gathering/ShutdownSaveTest.java     | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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<?>> 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());