vacuum statement generator

Author: hmeriann

src/include/statement_generator.hpp

@@ -106,6 +106,7 @@ class StatementGenerator {
 	unique_ptr<PrepareStatement> GeneratePrepare();
 	unique_ptr<ExportStatement> GenerateExport();
 	unique_ptr<InsertStatement> GenerateInsert();
+	unique_ptr<VacuumStatement> GenerateVacuum();
 
 	unique_ptr<OrderModifier> GenerateOrderBy();
 	unique_ptr<OrderModifier> GenerateOrderByAll();

src/statement_generator.cpp

@@ -37,6 +37,7 @@
 #include "duckdb/parser/statement/export_statement.hpp"
 #include "duckdb/parser/parsed_data/copy_info.hpp"
 #include "duckdb/parser/statement/insert_statement.hpp"
+#include "duckdb/parser/statement/vacuum_statement.hpp"
 
 namespace duckdb {
 
@@ -145,6 +146,9 @@ unique_ptr<SQLStatement> StatementGenerator::GenerateStatement() {
 	if (RandomPercentage(20)) {
 		return GenerateInsert();
 	}
+	if (RandomPercentage(20)) {
+		return GenerateVacuum();
+	}
 	return GenerateStatement(StatementType::CREATE_STATEMENT);
 }
 
@@ -177,6 +181,8 @@ unique_ptr<SQLStatement> StatementGenerator::GenerateStatement(StatementType typ
 		return GenerateExport();
 	case StatementType::INSERT_STATEMENT:
 		return GenerateInsert();
+	case StatementType::VACUUM_STATEMENT:
+		return GenerateVacuum();
 	default:
 		throw InternalException("Unsupported type");
 	}
@@ -388,6 +394,15 @@ unique_ptr<InsertStatement> StatementGenerator::GenerateInsert() {
     return stmt;
 }
 
+//===--------------------------------------------------------------------===//
+// Vacuum Statement
+//===--------------------------------------------------------------------===//
+
+unique_ptr<VacuumStatement> StatementGenerator::GenerateVacuum() {
+	duckdb::VacuumOptions opts;
+    return make_uniq<VacuumStatement>(opts);
+}
+
 //===--------------------------------------------------------------------===//
 // Generate Detach Info
 //===--------------------------------------------------------------------===//