Update test

GeyserMC · Feb 11, 2024 · 2d85131 · 2d85131
1 parent 0dcbc9f
commit 2d85131
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/core/src/test/java/org/geysermc/databaseutils/SqlTest.java b/core/src/test/java/org/geysermc/databaseutils/SqlTest.java
@@ -29,13 +29,14 @@
 import java.util.concurrent.Executors;
 import org.geysermc.databaseutils.data.BasicRepository;
 import org.geysermc.databaseutils.data.TestEntity;
+import org.geysermc.databaseutils.sql.SqlDialect;
 import org.junit.jupiter.api.Test;
 
 final class SqlTest {
     @Test
     void hello() throws ExecutionException, InterruptedException {
         var utils = DatabaseUtils.builder()
-                .config(new DatabaseConfig("jdbc:h2:./test", "sa", "", "hello", 2))
+                .config(new DatabaseConfig("jdbc:h2:./test", "sa", "", "hello", 2, SqlDialect.H2))
                 .executorService(Executors.newCachedThreadPool())
                 .build();
         utils.start();

diff --git a/database/core/src/main/java/org/geysermc/databaseutils/sql/SqlDialect.java b/database/core/src/main/java/org/geysermc/databaseutils/sql/SqlDialect.java
@@ -45,4 +45,12 @@ public static SqlDialect byName(String dialectName) {
         }
         return null;
     }
+
+    public static SqlDialect requireByName(String dialectName) {
+        var result = byName(dialectName);
+        if (result == null) {
+            throw new IllegalStateException("Could not find dialect for " + dialectName);
+        }
+        return result;
+    }
 }