diff --git a/src/main/java/edu/group5/app/App.java b/src/main/java/edu/group5/app/App.java index 8eae42a..dbd2319 100644 --- a/src/main/java/edu/group5/app/App.java +++ b/src/main/java/edu/group5/app/App.java @@ -7,7 +7,7 @@ */ public class App { public static void main(String[] args) throws InterruptedException { - DbWrapper db = new DbWrapper("test_init.sql"); + DbWrapper db = new DbWrapper(false); try { System.out.println(db.connect()); } catch (Exception e) { diff --git a/src/main/java/edu/group5/app/control/DbWrapper.java b/src/main/java/edu/group5/app/control/DbWrapper.java index b767a49..cafd03f 100644 --- a/src/main/java/edu/group5/app/control/DbWrapper.java +++ b/src/main/java/edu/group5/app/control/DbWrapper.java @@ -10,12 +10,12 @@ public class DbWrapper { private static final String DB_SCRIPT = "INIT=RUNSCRIPT FROM 'classpath:"; private String connectionString; - public DbWrapper(String script) { - this(script, "mem:test;"); - } - - public DbWrapper(String script, String dbPath) { - this.connectionString = CONNECTION_TYPE + dbPath + DB_SCRIPT + script + "'"; + public DbWrapper(boolean test) { + if (test) { + this.connectionString = CONNECTION_TYPE + "mem:test;" + DB_SCRIPT + "test_init.sql'"; + } else { + this.connectionString = CONNECTION_TYPE + "file:./help-me-help;" + DB_SCRIPT + "init.sql'"; + } } public boolean connect() throws SQLException { @@ -26,5 +26,4 @@ public boolean connect() throws SQLException { } return connection.isValid(0); } - }