Slightly improve code coverage

h2/src/main/org/h2/server/pg/PgServerThread.java

@@ -115,7 +115,7 @@ public class PgServerThread implements Runnable {
         return dataIn.readInt();
     }
 
-    private int readShort() throws IOException {
+    private short readShort() throws IOException {
         return dataIn.readShort();
     }
 
@@ -143,7 +143,7 @@ public class PgServerThread implements Runnable {
         byte[] data = DataUtils.newBytes(len);
         dataInRaw.readFully(data, 0, len);
         dataIn = new DataInputStream(new ByteArrayInputStream(data, 0, len));
-        switchBlock: switch (x) {
+        switch (x) {
         case 0:
             server.trace("Init");
             int version = readInt();
@@ -269,7 +269,7 @@ public class PgServerThread implements Runnable {
                 }
             } catch (Exception e) {
                 sendErrorResponse(e);
-                break switchBlock;
+                break;
             }
             int resultCodeCount = readShort();
             portal.resultColumnFormat = new int[resultCodeCount];
@@ -509,11 +509,11 @@ public class PgServerThread implements Runnable {
             switch (pgType) {
             case PgServer.PG_TYPE_INT2:
                 writeInt(2);
-                dataOut.writeShort(rs.getShort(column));
+                writeShort(rs.getShort(column));
                 break;
             case PgServer.PG_TYPE_INT4:
                 writeInt(4);
-                dataOut.writeInt(rs.getInt(column));
+                writeInt(rs.getInt(column));
                 break;
             case PgServer.PG_TYPE_INT8:
                 writeInt(8);
@@ -565,33 +565,23 @@ public class PgServerThread implements Runnable {
             // binary
             switch (pgType) {
             case PgServer.PG_TYPE_INT2:
-                if (paramLen != 4) {
-                    throw DbException.getInvalidValueException("paramLen", paramLen);
-                }
-                prep.setShort(col, dataIn.readShort());
+                checkParamLength(4, paramLen);
+                prep.setShort(col, readShort());
                 break;
             case PgServer.PG_TYPE_INT4:
-                if (paramLen != 4) {
-                    throw DbException.getInvalidValueException("paramLen", paramLen);
-                }
-                prep.setInt(col, dataIn.readInt());
+                checkParamLength(4, paramLen);
+                prep.setInt(col, readInt());
                 break;
             case PgServer.PG_TYPE_INT8:
-                if (paramLen != 8) {
-                    throw DbException.getInvalidValueException("paramLen", paramLen);
-                }
+                checkParamLength(8, paramLen);
                 prep.setLong(col, dataIn.readLong());
                 break;
             case PgServer.PG_TYPE_FLOAT4:
-                if (paramLen != 4) {
-                    throw DbException.getInvalidValueException("paramLen", paramLen);
-                }
+                checkParamLength(4, paramLen);
                 prep.setFloat(col, dataIn.readFloat());
                 break;
             case PgServer.PG_TYPE_FLOAT8:
-                if (paramLen != 8) {
-                    throw DbException.getInvalidValueException("paramLen", paramLen);
-                }
+                checkParamLength(8, paramLen);
                 prep.setDouble(col, dataIn.readDouble());
                 break;
             case PgServer.PG_TYPE_BYTEA:
@@ -607,6 +597,12 @@ public class PgServerThread implements Runnable {
             }
         }
     }
+    
+    private static void checkParamLength(int expected, int got) {
+        if (expected != got) {
+            throw DbException.getInvalidValueException("paramLen", got);
+        }
+    }
 
     private void sendErrorResponse(Exception re) throws IOException {
         SQLException e = DbException.toSQLException(re);

h2/src/test/org/h2/test/unit/TestPgServer.java

@@ -41,13 +41,23 @@ public class TestPgServer extends TestBase {
 
     @Override
     public void test() throws Exception {
-        testPGAdapter();
+        testPgAdapter();
         testKeyAlias();
         testCancelQuery();
         testBinaryTypes();
     }
+    
+    private boolean getPgJdbcDriver() {
+        try {
+            Class.forName("org.postgresql.Driver");
+            return true;
+        } catch (ClassNotFoundException e) {
+            println("PostgreSQL JDBC driver not found - PgServer not tested");
+            return false;
+        }
+    }
 
-    private void testPGAdapter() throws SQLException {
+    private void testPgAdapter() throws SQLException {
         deleteDb("test");
         Server server = Server.createPgServer("-baseDir", getBaseDir(), "-pgPort", "5535", "-pgDaemon");
         assertEquals(5535, server.getPort());
@@ -55,23 +65,24 @@ public class TestPgServer extends TestBase {
         server.start();
         assertStartsWith(server.getStatus(), "PG server running at pg://");
         try {
-            Class.forName("org.postgresql.Driver");
-            testPgClient();
-        } catch (ClassNotFoundException e) {
-            println("PostgreSQL JDBC driver not found - PgServer not tested");
+            if (getPgJdbcDriver()) {
+                testPgClient();
+            }
         } finally {
             server.stop();
         }
     }
 
     private void testCancelQuery() throws Exception {
+        if (!getPgJdbcDriver()) {
+            return;
+        }
+
         Server server = Server.createPgServer("-pgPort", "5535", "-pgDaemon", "-key", "test", "mem:test");
         server.start();
 
         ExecutorService executor = Executors.newSingleThreadExecutor();
         try {
-            Class.forName("org.postgresql.Driver");
-
             Connection conn = DriverManager.getConnection("jdbc:postgresql://localhost:5535/test", "sa", "sa");
             final Statement stat = conn.createStatement();
             stat.execute("create alias sleep for \"java.lang.Thread.sleep\"");
@@ -101,8 +112,6 @@ public class TestPgServer extends TestBase {
             } finally {
                 conn.close();
             }
-        } catch (ClassNotFoundException e) {
-            println("PostgreSQL JDBC driver not found - PgServer not tested");
         } finally {
             server.stop();
             executor.shutdown();
@@ -119,6 +128,7 @@ public class TestPgServer extends TestBase {
         stat.execute("create table test(id int primary key, name varchar)");
         stat.execute("create index idx_test_name on test(name, id)");
         stat.execute("grant all on test to test");
+        stat.close();
         conn.close();
 
         conn = DriverManager.getConnection("jdbc:postgresql://localhost:5535/test", "test", "test");
@@ -131,7 +141,13 @@ public class TestPgServer extends TestBase {
         assertEquals(6, rs.getInt(1));
         stat.execute("deallocate test");
 
-        PreparedStatement prep = conn.prepareStatement("insert into test values(?, ?)");
+        PreparedStatement prep;
+        prep = conn.prepareStatement("select * from test where name = ?");
+        prep.setNull(1, Types.VARCHAR);
+        rs = prep.executeQuery();
+        assertFalse(rs.next());
+        
+        prep = conn.prepareStatement("insert into test values(?, ?)");
         ParameterMetaData meta = prep.getParameterMetaData();
         assertEquals(2, meta.getParameterCount());
         prep.setInt(1, 1);
@@ -272,11 +288,12 @@ public class TestPgServer extends TestBase {
     }
 
     private void testKeyAlias() throws SQLException {
+        if (!getPgJdbcDriver()) {
+            return;
+        }
         Server server = Server.createPgServer("-pgPort", "5535", "-pgDaemon", "-key", "test", "mem:test");
         server.start();
         try {
-            Class.forName("org.postgresql.Driver");
-
             Connection conn = DriverManager.getConnection("jdbc:postgresql://localhost:5535/test", "sa", "sa");
             Statement stat = conn.createStatement();
 
@@ -288,19 +305,19 @@ public class TestPgServer extends TestBase {
             assertEquals("MEMORY", rs.getString(1));
 
             conn.close();
-        } catch (ClassNotFoundException e) {
-            println("PostgreSQL JDBC driver not found - PgServer not tested");
         } finally {
             server.stop();
         }
     }
 
     private void testBinaryTypes() throws SQLException {
+        if (!getPgJdbcDriver()) {
+            return;
+        }
+        
         Server server = Server.createPgServer("-pgPort", "5535", "-pgDaemon", "-key", "test", "mem:test");
         server.start();
         try {
-            Class.forName("org.postgresql.Driver");
-
             Connection conn = DriverManager.getConnection("jdbc:postgresql://localhost:5535/test", "sa", "sa");
             Statement stat = conn.createStatement();
 
@@ -335,8 +352,6 @@ public class TestPgServer extends TestBase {
                     rs.getBytes(10));
 
             conn.close();
-        } catch (ClassNotFoundException e) {
-            println("PostgreSQL JDBC driver not found - PgServer not tested");
         } finally {
             server.stop();
         }

h2/src/test/org/h2/test/unit/TestTraceSystem.java

@@ -35,8 +35,11 @@ public class TestTraceSystem extends TestBase {
 
     private static void testAdapter() {
         TraceSystem ts = new TraceSystem(null);
+        ts.setName("test");
         ts.setLevelFile(TraceSystem.ADAPTER);
+        ts.getTrace("test").debug("test");
         ts.getTrace("test").info("test");
+        ts.getTrace("test").error(new Exception(), "test");
         ts.close();
     }