db: createCountQuery table shortcut

Author: MykolaGolubyev

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

@@ -49,8 +49,12 @@ public TableData query() {
         return createQuery().queryTableData();
     }
 
+    public DbQuery createCountQuery() {
+        return QueryRunnerUtils.createQuery(dataSource, SqlQueriesGenerator.count(name));
+    }
+
     public DbQuery createQuery() {
-        return QueryRunnerUtils.createQuery(dataSource, SqlQueriesGenerator.query(name));
+        return QueryRunnerUtils.createQuery(dataSource, SqlQueriesGenerator.fullTable(name));
     }
 
     private void insertStep(TableData tableData) {

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

@@ -34,7 +34,11 @@ public static String insert(String tableName, Record record) {
         return "INSERT INTO " + tableName + " (" + columnNames + ") VALUES (" + questionMarks + ")";
     }
 
-    public static String query(String tableName) {
+    public static String fullTable(String tableName) {
         return "SELECT * FROM " + tableName;
     }
+
+    public static String count(String tableName) {
+        return "SELECT count(*) FROM " + tableName;
+    }
 }

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

@@ -130,11 +130,26 @@ class DatabaseFacadeTest extends DatabaseBaseTest {
         setupPrices()
         // query with where clause start
         def prices = db.createQuery("select * from PRICES")
+
+        prices.shouldNot == []
         db.update("delete from PRICES")
         prices.should == []
         // query with where clause end
     }
 
+    @Test
+    void "lazy count query"() {
+        setupPrices()
+        // query with where clause start
+        def PRICES = db.table("PRICES")
+        def numberOfItems = PRICES.createCountQuery()
+
+        numberOfItems.shouldNot == 0
+        db.update("delete from PRICES")
+        numberOfItems.should == 0
+        // query with where clause end
+    }
+
     @Test
     void "query table to match one row and assert against map"() {
         setupPrices()

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

@@ -66,6 +66,15 @@ If your query uses a single unique placeholder name, you can pass a regular valu
     excludeStartEnd: true
 }
 
+:include-groovy: org/testingisdocumenting/webtau/db/DatabaseFacadeTest.groovy {
+    entry: "lazy count query",
+    title: "lazy count query",
+    bodyOnly: true,
+    startLine: "query with where clause start",
+    endLine: "query with where clause end",
+    excludeStartEnd: true
+}
+
 # Single Value
 
 :include-groovy: org/testingisdocumenting/webtau/db/DatabaseFacadeTest.groovy {