From f4b5613fd3c6ce8e45fb7b99d10b33bec12ad43c Mon Sep 17 00:00:00 2001
From: Felix Kehrer <felix.kehrer@gmail.com>
Date: Sun, 6 May 2018 14:50:50 +0200
Subject: Copied changes from branch employee_list to allow for DAO Tests

---
 .../assignment/groupphase/util/JDBCConnectionManager.java     | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src/main/java/at/ac/tuwien')

diff --git a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/util/JDBCConnectionManager.java b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/util/JDBCConnectionManager.java
index 5494471..6eb15ec 100644
--- a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/util/JDBCConnectionManager.java
+++ b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/util/JDBCConnectionManager.java
@@ -12,12 +12,17 @@ import org.springframework.stereotype.Component;
 public class JDBCConnectionManager {
 
     private static final Logger LOG = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
-    private static final String CONNECTION_URL =
+    private static final String DEFAULT_CONNECTION_URL =
             "jdbc:h2:~/sepm;INIT=RUNSCRIPT FROM 'classpath:sql/database.sql'";
-
+    private String connectionUrl;
     private Connection connection;
 
     public JDBCConnectionManager() {
+        this(DEFAULT_CONNECTION_URL);
+    }
+
+    public JDBCConnectionManager(String connectionUrl) {
+        this.connectionUrl = connectionUrl;
         try {
             Class.forName("org.h2.Driver");
         } catch (ClassNotFoundException e) {
@@ -27,7 +32,7 @@ public class JDBCConnectionManager {
     }
 
     public Connection getConnection() throws SQLException {
-        if (connection == null) connection = DriverManager.getConnection(CONNECTION_URL);
+        if (connection == null) connection = DriverManager.getConnection(connectionUrl);
 
         return connection;
     }
-- 
cgit v1.2.3-70-g09d2