Extend YSQL error handling to include subquery-related and additional specific error cases across multiple generators

Author: qvad

src/sqlancer/yugabyte/ysql/YSQLErrors.java

@@ -20,6 +20,7 @@ public static void addKnownIssues(ExpectedErrors errors) {
         errors.add("marked for deletion");
         errors.add("hit the limit");
         errors.add("insufficient disk space");
+        errors.add("NullTest indexqual has wrong key");
     }
 
     public static void addCommonFetchErrors(ExpectedErrors errors) {

src/sqlancer/yugabyte/ysql/gen/YSQLDeleteGenerator.java

@@ -38,6 +38,7 @@ public static SQLQueryAdapter create(YSQLGlobalState globalState) {
         }
         YSQLErrors.addCommonExpressionErrors(errors);
         YSQLErrors.addTransactionErrors(errors);
+        YSQLErrors.addSubqueryErrors(errors);
         errors.add("out of range");
         errors.add("cannot cast");
         errors.add("invalid input syntax for");

src/sqlancer/yugabyte/ysql/gen/YSQLIndexGenerator.java

@@ -143,6 +143,7 @@ public static SQLQueryAdapter generate(YSQLGlobalState globalState) {
         errors.add("cannot use subquery in index expression");
         YSQLErrors.addCommonExpressionErrors(errors);
         YSQLErrors.addTransactionErrors(errors);
+        YSQLErrors.addSubqueryErrors(errors);
         return new SQLQueryAdapter(sb.toString(), errors);
     }
 

src/sqlancer/yugabyte/ysql/gen/YSQLMergeGenerator.java

@@ -201,6 +201,8 @@ public static SQLQueryAdapter create(YSQLTable targetTable, YSQLGlobalState glob
 
         YSQLErrors.addCommonExpressionErrors(errors);
         YSQLErrors.addCommonInsertUpdateErrors(errors);
+        YSQLErrors.addTransactionErrors(errors);
+        YSQLErrors.addSubqueryErrors(errors);
 
         return new SQLQueryAdapter(sb.toString(), errors);
     }

src/sqlancer/yugabyte/ysql/gen/YSQLTableGenerator.java

@@ -67,6 +67,7 @@ public YSQLTableGenerator(String tableName, boolean generateOnlyKnown, YSQLGloba
         YSQLErrors.addCommonExpressionErrors(errors);
         YSQLErrors.addCommonTableErrors(errors);
         YSQLErrors.addTransactionErrors(errors);
+        YSQLErrors.addSubqueryErrors(errors);
     }
 
     public static SQLQueryAdapter generate(String tableName, boolean generateOnlyKnown, YSQLGlobalState globalState) {

src/sqlancer/yugabyte/ysql/gen/YSQLUpdateGenerator.java

@@ -52,6 +52,7 @@ private SQLQueryAdapter generate() {
         errors.add("but expression is of type");
         YSQLErrors.addCommonExpressionErrors(errors);
         YSQLErrors.addTransactionErrors(errors);
+        YSQLErrors.addSubqueryErrors(errors);
         if (!Randomly.getBooleanWithSmallProbability()) {
             sb.append(" WHERE ");
             YSQLExpression where = YSQLExpressionGenerator.generateExpression(globalState, randomTable.getColumns(),

src/sqlancer/yugabyte/ysql/gen/YSQLViewGenerator.java

@@ -56,6 +56,10 @@ public static SQLQueryAdapter create(YSQLGlobalState globalState) {
         YSQLErrors.addViewErrors(errors);
         YSQLErrors.addCommonExpressionErrors(errors);
         YSQLErrors.addTransactionErrors(errors);
+        YSQLErrors.addWindowFunctionErrors(errors);
+        YSQLErrors.addCTEErrors(errors);
+        YSQLErrors.addSetOperationErrors(errors);
+        YSQLErrors.addSubqueryErrors(errors);
         return new SQLQueryAdapter(sb.toString(), errors, true);
     }