From e2e86cadc3ff098dc8974a875210544dcdbcd903 Mon Sep 17 00:00:00 2001
From: Risto Lahtela <24460436+Rsl1122@users.noreply.github.com>
Date: Thu, 12 Mar 2020 18:04:10 +0200
Subject: [PATCH] Extension RemoveEverythingTransaction test

---
 .../{ => queries}/ExtensionsDatabaseTest.java | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)
 rename Plan/common/src/test/java/com/djrapitops/plan/storage/database/{ => queries}/ExtensionsDatabaseTest.java (96%)

diff --git a/Plan/common/src/test/java/com/djrapitops/plan/storage/database/ExtensionsDatabaseTest.java b/Plan/common/src/test/java/com/djrapitops/plan/storage/database/queries/ExtensionsDatabaseTest.java
similarity index 96%
rename from Plan/common/src/test/java/com/djrapitops/plan/storage/database/ExtensionsDatabaseTest.java
rename to Plan/common/src/test/java/com/djrapitops/plan/storage/database/queries/ExtensionsDatabaseTest.java
index 1910733d5..40fe64db8 100644
--- a/Plan/common/src/test/java/com/djrapitops/plan/storage/database/ExtensionsDatabaseTest.java
+++ b/Plan/common/src/test/java/com/djrapitops/plan/storage/database/queries/ExtensionsDatabaseTest.java
@@ -14,7 +14,7 @@
  *  You should have received a copy of the GNU Lesser General Public License
  *  along with Plan. If not, see <https://www.gnu.org/licenses/>.
  */
-package com.djrapitops.plan.storage.database;
+package com.djrapitops.plan.storage.database.queries;
 
 import com.djrapitops.plan.data.element.TableContainer;
 import com.djrapitops.plan.extension.CallEvents;
@@ -32,7 +32,8 @@ import com.djrapitops.plan.extension.implementation.storage.transactions.results
 import com.djrapitops.plan.extension.implementation.storage.transactions.results.RemoveUnsatisfiedConditionalServerResultsTransaction;
 import com.djrapitops.plan.extension.table.Table;
 import com.djrapitops.plan.gathering.domain.Session;
-import com.djrapitops.plan.storage.database.queries.DataStoreQueries;
+import com.djrapitops.plan.storage.database.DatabaseTestPreparer;
+import com.djrapitops.plan.storage.database.transactions.commands.RemoveEverythingTransaction;
 import com.djrapitops.plan.storage.database.transactions.events.WorldNameStoreTransaction;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -63,6 +64,20 @@ public interface ExtensionsDatabaseTest extends DatabaseTestPreparer {
         extensionService.unregister(new TableExtension());
     }
 
+    @Test
+    default void removeEverythingRemovesPlayerExtensionData() {
+        extensionPlayerValuesAreStored();
+        db().executeTransaction(new RemoveEverythingTransaction());
+        assertTrue(db().query(new ExtensionPlayerDataQuery(playerUUID)).isEmpty());
+    }
+
+    @Test
+    default void removeEverythingRemovesServerExtensionData() {
+        extensionServerValuesAreStored();
+        db().executeTransaction(new RemoveEverythingTransaction());
+        assertTrue(db().query(new ExtensionServerDataQuery(serverUUID())).isEmpty());
+    }
+
     @Test
     default void extensionPlayerValuesAreStored() {
         ExtensionSvc extensionService = (ExtensionSvc) system().getExtensionService();