db: single row insert shortcut

Author: MykolaGolubyev

webtau-core/src/main/java/org/testingisdocumenting/webtau/data/table/Record.java

@@ -142,7 +142,7 @@ public String toString() {
     }
 
     private static class MultiValuesUnwrapper {
-        private List<Record> result;
+        private final List<Record> result;
 
         MultiValuesUnwrapper() {
             this.result = new ArrayList<>();
@@ -173,7 +173,7 @@ void add(Record record) {
     private static class RecordFromStream {
         private boolean hasMultiValues;
         private boolean hasValueGenerators;
-        private List<Object> values;
+        private final List<Object> values;
 
         public RecordFromStream(Stream<Object> valuesStream) {
             values = new ArrayList<>();

webtau-db/src/main/java/org/testingisdocumenting/webtau/db/DatabaseTable.java

@@ -22,6 +22,7 @@
 import org.testingisdocumenting.webtau.reporter.MessageToken;
 
 import java.sql.SQLException;
+import java.util.Map;
 
 import static org.testingisdocumenting.webtau.reporter.IntegrationTestsMessageBuilder.*;
 import static org.testingisdocumenting.webtau.reporter.TokenizedMessage.tokenizedMessage;
@@ -42,7 +43,16 @@ public void insert(TableData tableData) {
                         INTO, createMessageId()),
                 () -> tokenizedMessage(action("inserted"), numberValue(tableData.numberOfRows()), action("row(s)"),
                         INTO, createMessageId()),
-                () -> insertStep(tableData));
+                () -> insertTableStep(tableData));
+    }
+
+    public void insert(Map<String, Object> row) {
+        createAndExecuteStep(
+                tokenizedMessage(action("inserting"), numberValue(1), action("row"),
+                        INTO, createMessageId()),
+                () -> tokenizedMessage(action("inserted"), numberValue(1), action("row"),
+                        INTO, createMessageId()),
+                () -> insertRowStep(row));
     }
 
     public TableData query() {
@@ -57,7 +67,7 @@ public DbQuery createQuery() {
         return QueryRunnerUtils.createQuery(dataSource, SqlQueriesGenerator.fullTable(name));
     }
 
-    private void insertStep(TableData tableData) {
+    private void insertTableStep(TableData tableData) {
         if (tableData.isEmpty()) {
             return;
         }
@@ -75,11 +85,25 @@ private void insertStep(TableData tableData) {
         }
     }
 
+    private void insertRowStep(Map<String, Object> row) {
+        QueryRunner run = new QueryRunner(dataSource.getDataSource());
+        try {
+            run.update(SqlQueriesGenerator.insert(name, row.keySet().stream(), row.values().stream()),
+                    row.values().toArray());
+        } catch (SQLException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
     private MessageToken createMessageId() {
         return id(dataSource.getLabel() + "." + name);
     }
 
     public void leftShift(TableData tableData) {
         insert(tableData);
     }
+
+    public void leftShift(Map<String, Object> row) {
+        insert(row);
+    }
 }

webtau-db/src/main/java/org/testingisdocumenting/webtau/db/gen/SqlQueriesGenerator.java

@@ -18,20 +18,24 @@
 
 import org.testingisdocumenting.webtau.data.table.Record;
 
+import java.util.Collection;
 import java.util.List;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 public class SqlQueriesGenerator {
     private SqlQueriesGenerator() {
     }
 
     public static String insert(String tableName, Record record) {
-        String columnNames = record.getHeader().getNamesStream().collect(Collectors.joining(", "));
+        return insert(tableName, record.getHeader().getNamesStream(), record.valuesStream());
+    }
 
-        List<Object> values = record.getValues();
-        String questionMarks = values.stream().map(v -> "?").collect(Collectors.joining(", "));
+    public static String insert(String tableName, Stream<String> columnNamesStream, Stream<Object> valuesStream) {
+        String enumeratedColumnNames = columnNamesStream.collect(Collectors.joining(", "));
+        String questionMarks = valuesStream.map(v -> "?").collect(Collectors.joining(", "));
 
-        return "INSERT INTO " + tableName + " (" + columnNames + ") VALUES (" + questionMarks + ")";
+        return "INSERT INTO " + tableName + " (" + enumeratedColumnNames + ") VALUES (" + questionMarks + ")";
     }
 
     public static String fullTable(String tableName) {

webtau-db/src/test/groovy/org/testingisdocumenting/webtau/db/DatabaseFacadeTest.groovy

@@ -36,9 +36,9 @@ class DatabaseFacadeTest extends DatabaseBaseTest {
                    "id2" | "another set" | 2000 }
 
         PRICES.createQuery().should == ["ID" | "DESCRIPTION" | "PRICE"] {
-                                 ___________________________________
-                                 "id1" | "nice set"    | 1000
-                                 "id2" | "another set" | 2000 }
+                                       ___________________________________
+                                       "id1" | "nice set"    | 1000
+                                       "id2" | "another set" | 2000 }
 
         db.createQuery("select * from PRICES where id='id2'").should ==
                 ["ID" | "DESCRIPTION" | "PRICE"] {
@@ -59,6 +59,18 @@ class DatabaseFacadeTest extends DatabaseBaseTest {
         PRICES.query().numberOfRows().should == 2
     }
 
+    @Test
+    void "should insert single row into table"() {
+        db.update("delete from PRICES")
+
+        def PRICES = db.table("PRICES")
+        PRICES << [id: "id1", description: "nice set", price: 1000]
+
+        PRICES.should == ["ID" | "DESCRIPTION" | "PRICE"] {
+                         ___________________________________
+                         "id1" | "nice set"    | 1000 }
+    }
+
     @Test
     void "use table data permute, above and guid to generate rows"() {
         db.update("delete from PRICES")

webtau-docs/znai/database/data-setup.md

@@ -11,6 +11,15 @@ The simplest way to set up a DB state is to use handcrafted [TableData](referenc
     commentsType: "inline"
 }
 
+:include-groovy: org/testingisdocumenting/webtau/db/DatabaseFacadeTest.groovy {
+    entry: "should insert single row into table",
+    title: "inserting single row",
+    bodyOnly: true,
+    startLine: "def PRICES",
+    endLine: "1000]",
+    commentsType: "inline"
+}
+ 
 # Semi-Auto Generated TableData
 
 [TableData](reference/table-data) has features like [permute](reference/table-data#permutations) and