Increase code coverage.

orbisgis · Nov 27, 2019 · 15785e4 · 15785e4
1 parent d510df3
commit 15785e4
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/data-manager/src/test/java/org/orbisgis/datamanager/JdbcTableTest.java b/data-manager/src/test/java/org/orbisgis/datamanager/JdbcTableTest.java
@@ -561,6 +561,25 @@ public void testSqlBuilding() {
         assertEquals("SELECT * FROM ORBISGIS LIMIT 0", getTable().limit(0).toString().trim());
     }
 
+    /**
+     * Test the {@link JdbcTable#columns(String...)} and {@link JdbcTable#columns(List)} methods.
+     */
+    @Test
+    public void testColumns() {
+        JdbcTable table = getTable();
+        JdbcSpatialTable spatialTable = (JdbcSpatialTable)dataSource.getSpatialTable(TABLE_NAME);
+
+        List<String> columns = new ArrayList<>();
+        columns.add("TOTO");
+        columns.add("tata");
+        columns.add("TIti");
+
+        assertEquals("SELECT TOTO, tata, TIti FROM ORBISGIS WHERE toto", ((JdbcTable)table.columns("TOTO", "tata", "TIti")).where("toto").toString().trim());
+        assertEquals("SELECT TOTO, tata, TIti FROM ORBISGIS WHERE toto", ((JdbcTable)table.columns(columns)).where("toto").toString().trim());
+        assertEquals("SELECT TOTO, tata, TIti FROM ORBISGIS WHERE toto", ((JdbcTable)spatialTable.columns("TOTO", "tata", "TIti")).where("toto").toString().trim());
+        assertEquals("SELECT TOTO, tata, TIti FROM ORBISGIS WHERE toto",((JdbcTable) spatialTable.columns(columns)).where("toto").toString().trim());
+    }
+
     /**
      * Test the {@link JdbcTable#getTable()} and {@link JdbcTable#getSpatialTable()} methods.
      */