Merge pull request #9 from JavaWebStack/TimothyGillespie/expandFactoryTest

Expand factory tests

Author: JanHolger

src/test/java/org/javawebstack/orm/test/SQLDriverFactoryTest.java

@@ -2,22 +2,38 @@
 
 import static org.junit.jupiter.api.Assertions.*;
 
+import org.javawebstack.orm.wrapper.MySQL;
 import org.javawebstack.orm.wrapper.SQLDriverFactory;
 import org.javawebstack.orm.wrapper.SQLDriverNotFoundException;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
 import java.util.HashMap;
+import java.util.Map;
 
 class SQLDriverFactoryTest {
-    private SQLDriverFactory factory = new SQLDriverFactory(new HashMap<String, String>() {{
-        put("file", "sb.sqlite");
-        put("host", "localhost");
-        put("port", "3306");
-        put("name", "app");
-        put("user", "root");
-        put("password", "");
-    }});
 
+    private SQLDriverFactory factory;
+    private final Map<String, String > configMap = new HashMap<String, String >(){
+        {
+            put("file", "sb.sqlite");
+            put("host", "localhost");
+            put("port", "3306");
+            put("name", "app");
+            put("user", "root");
+            put("password", "");
+        }
+    };
+
+
+    @BeforeEach
+    public void beforeEach() {
+        factory = new SQLDriverFactory(configMap);
+    }
+
+    /*
+     * Normal Cases
+     */
     @Test
     public void testSQLite() throws SQLDriverNotFoundException {
         assertNotNull(factory.getDriver("sqlite"));
@@ -27,4 +43,41 @@ public void testSQLite() throws SQLDriverNotFoundException {
     public void testMySQL() throws SQLDriverNotFoundException {
         assertNotNull(factory.getDriver("mysql"));
     }
+
+    @Test
+    public void testRegisrtration() throws SQLDriverNotFoundException {
+        factory.registerDriver("mariadb", () -> new MySQL(
+                configMap.get("host"),
+                Integer.parseInt(configMap.get("port")),
+                configMap.get("name"),
+                configMap.get("user"),
+                configMap.get("password")
+        ));
+
+        assertNotNull(factory.getDriver("mariadb"));
+    }
+
+    /*
+     * Edge Cases
+     */
+
+    @Test
+    public void testDriverNameIsCaseSensitive() {
+        assertThrows(
+                SQLDriverNotFoundException.class,
+                () -> factory.getDriver("MySQL")
+        );
+    }
+
+    /*
+     * Error Cases
+     */
+
+    @Test
+    public void testCannotFindUninitializedDriver() {
+        assertThrows(
+                SQLDriverNotFoundException.class,
+                () -> factory.getDriver("nodriver")
+        );
+    }
 }