File tree Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -105,6 +105,7 @@ class StatementGenerator {
105105 unique_ptr<DropStatement> GenerateDrop ();
106106 unique_ptr<PrepareStatement> GeneratePrepare ();
107107 unique_ptr<ExportStatement> GenerateExport ();
108+ unique_ptr<InsertStatement> GenerateInsert ();
108109
109110 unique_ptr<OrderModifier> GenerateOrderBy ();
110111 unique_ptr<OrderModifier> GenerateOrderByAll ();
Original file line number Diff line number Diff line change @@ -142,6 +142,9 @@ unique_ptr<SQLStatement> StatementGenerator::GenerateStatement() {
142142 if (RandomPercentage (20 )) {
143143 return GenerateExport ();
144144 }
145+ if (RandomPercentage (20 )) {
146+ return GenerateInsert ();
147+ }
145148 return GenerateStatement (StatementType::CREATE_STATEMENT);
146149}
147150
@@ -171,7 +174,9 @@ unique_ptr<SQLStatement> StatementGenerator::GenerateStatement(StatementType typ
171174 case StatementType::DROP_STATEMENT:
172175 return GenerateDrop ();
173176 case StatementType::EXPORT_STATEMENT:
174- return GenerateDrop ();
177+ return GenerateExport ();
178+ case StatementType::INSERT_STATEMENT:
179+ return GenerateInsert ();
175180 default :
176181 throw InternalException (" Unsupported type" );
177182 }
@@ -364,7 +369,7 @@ unique_ptr<ExportStatement> StatementGenerator::GenerateExport() {
364369}
365370
366371// ===--------------------------------------------------------------------===//
367- // Export Statement
372+ // Insert Statement
368373// ===--------------------------------------------------------------------===//
369374
370375unique_ptr<InsertStatement> StatementGenerator::GenerateInsert () {
You can’t perform that action at this time.
0 commit comments