[DuckDB] Add options for the max number of views, deletes, and updates

mrigger · mrigger · commit 53d57e94f13c · 2020-04-25T15:05:24.000+02:00
diff --git a/src/sqlancer/duckdb/DuckDBOptions.java b/src/sqlancer/duckdb/DuckDBOptions.java
@@ -57,6 +57,15 @@ public class DuckDBOptions {
 	@Parameter(names = "--test-indexes", description="Allow explicit (i.e. CREATE INDEX) and implicit (i.e., UNIQUE and PRIMARY KEY) indexes")
 	public boolean testIndexes = true;
 	
+	@Parameter(names = "--max-num-views", description="The maximum number of views that can be generated for a database")
+	public int maxNumViews = 1;
+	
+	@Parameter(names = "--max-num-deletes", description="The maximum number of DELETE statements that are issued for a database")
+	public int maxNumDeletes = 1;
+	
+	@Parameter(names = "--max-num-updates", description="The maximum number of UPDATE statements that are issued for a database")
+	public int maxNumUpdates = 5;
+	
 	@Parameter(names = "--oracle", converter = DBMSConverter.class)
 	public List<DuckDBOracle> oracle = Arrays.asList(DuckDBOracle.QUERY_PARTITIONING);
 
diff --git a/src/sqlancer/duckdb/DuckDBProvider.java b/src/sqlancer/duckdb/DuckDBProvider.java
@@ -70,13 +70,15 @@ private static int mapActions(DuckDBGlobalState globalState, Action a) {
 			}
 			// fall through
 		case UPDATE:
-			return r.getInteger(0, 4);
+			return r.getInteger(0, globalState.getDmbsSpecificOptions().maxNumUpdates + 1);
 		case VACUUM: // seems to be ignored
 		case ANALYZE:  // seems to be ignored
-		case DELETE:
 		case EXPLAIN:
-		case CREATE_VIEW:
 			return r.getInteger(0, 2);
+		case DELETE:
+			return r.getInteger(0, globalState.getDmbsSpecificOptions().maxNumDeletes + 1);
+		case CREATE_VIEW:
+			return r.getInteger(0, globalState.getDmbsSpecificOptions().maxNumViews + 1);
 		default:
 			throw new AssertionError(a);
 		}
