Add JUnit for DbState

gliptak · gliptak · commit 63667ffc4606 · 2016-06-04T13:41:30.000-04:00
diff --git a/core/build.gradle b/core/build.gradle
@@ -7,6 +7,8 @@ archivesBaseName = project.projectName
 
 dependencies {
     testCompile dependencyJunit
+    testCompile 'com.h2database:h2:1.4.191'
+    testCompile 'commons-dbutils:commons-dbutils:1.6'
 }
 
 performSigning(signingEnabled, signModule)
diff --git a/core/src/test/java/fj/control/db/TestDbState.java b/core/src/test/java/fj/control/db/TestDbState.java
@@ -0,0 +1,55 @@
+package fj.control.db;
+
+import fj.Unit;
+import fj.data.Option;
+import org.apache.commons.dbutils.DbUtils;
+import org.junit.Test;
+
+import java.sql.*;
+
+import static org.hamcrest.core.Is.is;
+import static org.junit.Assert.assertThat;
+
+public class TestDbState {
+    @Test
+    public void testWriter() throws SQLException {
+        final int TEN = 10;
+        DbState writer = DbState.writer(DbState.driverManager("jdbc:h2:mem:"));
+
+        DB<Unit> setup = new DB<Unit>() {
+            @Override
+            public Unit run(Connection c) throws SQLException {
+                Statement s = null;
+                try {
+                    s = c.createStatement();
+                    assertThat(s.executeUpdate("CREATE TABLE TEST(ID INT PRIMARY KEY, NAME VARCHAR(255))"), is(0));
+                    assertThat(s.executeUpdate("INSERT INTO TEST (ID, NAME) VALUES (" + TEN + ", 'FOO')"), is(1));
+                } finally {
+                    DbUtils.closeQuietly(s);
+                }
+                return Unit.unit();
+            }
+        };
+        DB<Option<Integer>> query = new DB<Option<Integer>>() {
+            @Override
+            public Option<Integer> run(Connection c) throws SQLException {
+                PreparedStatement ps = null;
+                ResultSet rs = null;
+                try {
+                    ps = c.prepareStatement("SELECT ID FROM TEST WHERE NAME = ?");
+                    ps.setString(1, "FOO");
+                    rs = ps.executeQuery();
+                    if (rs.next()) {
+                        return Option.some(rs.getInt("ID"));
+                    } else {
+                        return Option.none();
+                    }
+                } finally {
+                    DbUtils.closeQuietly(rs);
+                    DbUtils.closeQuietly(ps);
+                }
+            }
+        };
+        assertThat(writer.run(setup.bind(v -> query)).some(), is(TEN));
+    }
+}

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,8 @@ archivesBaseName = project.projectName`
`7`	`7`
`8`	`8`	`dependencies {`
`9`	`9`	`testCompile dependencyJunit`
	`10`	`+ testCompile 'com.h2database:h2:1.4.191'`
	`11`	`+ testCompile 'commons-dbutils:commons-dbutils:1.6'`
`10`	`12`	`}`
`11`	`13`
`12`	`14`	`performSigning(signingEnabled, signModule)`