#2 - Support for java.sql.Connection

Author: Schubi1981

.gitignore

@@ -2,3 +2,4 @@
 /.project
 /.settings/
 /target/
+/.factorypath

pom.xml

@@ -55,7 +55,9 @@
     <slf4j.version>1.7.30</slf4j.version>
     <logback.version>1.2.3</logback.version>
     <assertj.version>3.11.1</assertj.version>
-    <junit.version>5.3.2</junit.version>
+    <junit.version>5.5.2</junit.version>
+    <powermock.version>2.0.2</powermock.version>
+    <easymock.version>4.0.2</easymock.version>
     <lombok.version>1.18.10</lombok.version>
     <jacoco.version>0.8.5</jacoco.version>
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
@@ -91,6 +93,36 @@
       <version>${junit.version}</version>
       <scope>test</scope>
     </dependency>
+    <dependency>
+      <groupId>org.junit.vintage</groupId>
+      <artifactId>junit-vintage-engine</artifactId>
+      <version>${junit.version}</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.easymock</groupId>
+      <artifactId>easymock</artifactId>
+      <version>${easymock.version}</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.powermock</groupId>
+      <artifactId>powermock-api-easymock</artifactId>
+      <version>${powermock.version}</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.powermock</groupId>
+      <artifactId>powermock-module-junit4</artifactId>
+      <version>${powermock.version}</version>
+      <scope>test</scope>
+      <exclusions>
+        <exclusion>
+          <groupId>junit</groupId>
+          <artifactId>junit</artifactId>
+        </exclusion>
+      </exclusions>
+    </dependency>
     <dependency>
       <groupId>org.jacoco</groupId>
       <artifactId>org.jacoco.agent</artifactId>

src/main/java/de/jaggl/sqlbuilder/queries/Query.java

@@ -3,6 +3,9 @@
 import static de.jaggl.sqlbuilder.utils.Indentation.disabled;
 
 import java.io.PrintStream;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
 
 import de.jaggl.sqlbuilder.dialect.Dialect;
 import de.jaggl.sqlbuilder.utils.Indentation;
@@ -155,4 +158,19 @@ default void println(PrintStream printStream, String dialectName, Indentation in
     {
         println(printStream, Dialect.forName(dialectName), indentation);
     }
+
+    default PreparedStatement prepare(Connection connection) throws SQLException
+    {
+        return connection.prepareStatement(build());
+    }
+
+    default PreparedStatement prepare(Connection connection, Dialect dialect) throws SQLException
+    {
+        return connection.prepareStatement(build(dialect));
+    }
+
+    default PreparedStatement prepare(Connection connection, String dialectName) throws SQLException
+    {
+        return connection.prepareStatement(build(dialectName));
+    }
 }

src/test/java/de/jaggl/sqlbuilder/queries/QueryTest.java

@@ -4,13 +4,22 @@
 import static de.jaggl.sqlbuilder.queries.Queries.select;
 import static de.jaggl.sqlbuilder.schema.Table.create;
 import static de.jaggl.sqlbuilder.utils.Indentation.enabled;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.easymock.EasyMock.expect;
+import static org.powermock.api.easymock.PowerMock.createStrictMock;
+import static org.powermock.api.easymock.PowerMock.replayAll;
+import static org.powermock.api.easymock.PowerMock.verifyAll;
 
-import org.junit.jupiter.api.Test;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
 
-class QueryTest
+import org.junit.Test;
+
+public class QueryTest
 {
     @Test
-    void testPrintAndPrintln()
+    public void testPrintAndPrintln()
     {
         var query = select().from(create("table"));
 
@@ -52,4 +61,55 @@ void testPrintAndPrintln()
         System.out.println();
         query.println(System.out, "MYSQL", enabled());
     }
+
+    @Test
+    public void testPrepare() throws SQLException
+    {
+        var query = select().from(create("table"));
+
+        try (var connection = createStrictMock(Connection.class))
+        {
+            expect(connection.prepareStatement("SELECT * FROM `table`")).andReturn(createStrictMock(PreparedStatement.class));
+            connection.close();
+
+            replayAll();
+            assertThat(query.prepare(connection)).isInstanceOf(PreparedStatement.class);
+        }
+
+        verifyAll();
+    }
+
+    @Test
+    public void testPrepareWithDialect() throws SQLException
+    {
+        var query = select().from(create("table"));
+
+        try (var connection = createStrictMock(Connection.class))
+        {
+            expect(connection.prepareStatement("SELECT * FROM `table`")).andReturn(createStrictMock(PreparedStatement.class));
+            connection.close();
+
+            replayAll();
+            assertThat(query.prepare(connection, MYSQL)).isInstanceOf(PreparedStatement.class);
+        }
+
+        verifyAll();
+    }
+
+    @Test
+    public void testPrepareWithDialectName() throws SQLException
+    {
+        var query = select().from(create("table"));
+
+        try (var connection = createStrictMock(Connection.class))
+        {
+            expect(connection.prepareStatement("SELECT * FROM `table`")).andReturn(createStrictMock(PreparedStatement.class));
+            connection.close();
+
+            replayAll();
+            assertThat(query.prepare(connection, "MYSQL")).isInstanceOf(PreparedStatement.class);
+        }
+
+        verifyAll();
+    }
 }