Fixed tests

Author: JanHolger

src/test/java/org/javawebstack/orm/test/querybuilding/FromClauseTest.java

@@ -3,6 +3,7 @@
 import org.javawebstack.orm.Model;
 import org.javawebstack.orm.ORM;
 import org.javawebstack.orm.Repo;
+import org.javawebstack.orm.connection.pool.PooledSQL;
 import org.javawebstack.orm.exception.ORMConfigurationException;
 import org.javawebstack.orm.test.ORMTestCase;
 import org.javawebstack.orm.test.shared.models.tablenames.*;
@@ -68,6 +69,8 @@ void testOverwrittenTableName() throws ORMConfigurationException {
      */
     private String getBaseQuery(Class<? extends Model> clazz) throws ORMConfigurationException {
         ORM.register(clazz, sql());
-        return Repo.get(clazz).getConnection().builder().buildQuery(Repo.get(clazz).query()).getQuery();
+        try(PooledSQL sql = Repo.get(clazz).getPool().get()) {
+            return sql.builder().buildQuery(Repo.get(clazz).query()).getQuery();
+        }
     }
 }

src/test/java/org/javawebstack/orm/test/shared/verification/QueryVerification.java

@@ -1,5 +1,6 @@
 package org.javawebstack.orm.test.shared.verification;
 
+import org.javawebstack.orm.connection.pool.PooledSQL;
 import org.javawebstack.orm.query.Query;
 import org.javawebstack.orm.test.exception.SectionIndexOutOfBoundException;
 import org.javawebstack.orm.test.shared.util.QueryStringUtil;
@@ -145,8 +146,10 @@ public String getSection(String topLevelKeyword, int sectionIndex) throws Sectio
      * @return The order sensitive string list of inner sections.
      */
     public List<String> getSectionList(String topLevelKeyword) {
-        return new QueryStringUtil(this.query.getRepo().getConnection().builder().buildQuery(this.query).getQuery())
-                .getTopLevelSectionsByKeyword(topLevelKeyword);
+        try(PooledSQL sql = this.query.getRepo().getPool().get()) {
+            return new QueryStringUtil(sql.builder().buildQuery(this.query).getQuery())
+                    .getTopLevelSectionsByKeyword(topLevelKeyword);
+        }
     }
 
     private void failDueToSectionIndexOutOfBounds(SectionIndexOutOfBoundException exception) {