fix: broken grammar of aggreate in SQLite (#651)

bajinsheng · web-flow · commit ec1cbe243eba · 2022-12-17T20:09:32.000+08:00
diff --git a/src/sqlancer/sqlite3/oracle/tlp/SQLite3TLPAggregateOracle.java b/src/sqlancer/sqlite3/oracle/tlp/SQLite3TLPAggregateOracle.java
@@ -66,7 +66,9 @@ public void check() throws SQLException {
         SQLite3Select leftSelect = getSelect(aggregate, from, whereClause);
         SQLite3Select middleSelect = getSelect(aggregate, from, negatedClause);
         SQLite3Select rightSelect = getSelect(aggregate, from, notNullClause);
-        String metamorphicText = "SELECT " + aggregate.getFunc().toString() + "(aggr) FROM (";
+        String aggreateMethod = aggregate.getFunc() == SQLite3AggregateFunction.COUNT_ALL
+                ? SQLite3AggregateFunction.COUNT.toString() : aggregate.getFunc().toString();
+        String metamorphicText = "SELECT " + aggreateMethod + "(aggr) FROM (";
         metamorphicText += SQLite3Visitor.asString(leftSelect) + " UNION ALL " + SQLite3Visitor.asString(middleSelect)
                 + " UNION ALL " + SQLite3Visitor.asString(rightSelect);
         metamorphicText += ")";
