diff --git a/simulator/generation/plan.rs b/simulator/generation/plan.rs index 39247a029..42aab9df6 100644 --- a/simulator/generation/plan.rs +++ b/simulator/generation/plan.rs @@ -772,7 +772,7 @@ pub(crate) struct InteractionStats { pub rollback_count: u32, pub alter_table_count: u32, pub drop_index_count: u32, - pub pragma_count:u32, + pub pragma_count: u32, } impl Display for InteractionStats { diff --git a/simulator/generation/query.rs b/simulator/generation/query.rs index 9706ba4ca..443a67231 100644 --- a/simulator/generation/query.rs +++ b/simulator/generation/query.rs @@ -11,7 +11,8 @@ use sql_generation::{ generation::{Arbitrary, ArbitraryFrom, GenerationContext, query::SelectFree}, model::{ query::{ - Create, CreateIndex, Delete, DropIndex, Insert, Select, alter_table::AlterTable, + Create, CreateIndex, Delete, DropIndex, Insert, Select, + alter_table::AlterTable, pragma::{Pragma, VacuumMode}, update::Update, }, diff --git a/simulator/model/mod.rs b/simulator/model/mod.rs index 60ddbd2f3..1ffa5a161 100644 --- a/simulator/model/mod.rs +++ b/simulator/model/mod.rs @@ -83,7 +83,7 @@ impl Query { | Query::DropIndex(DropIndex { table_name: table, .. }) => IndexSet::from_iter([table.clone()]), -Query::Begin(_) + Query::Begin(_) | Query::Commit(_) | Query::Rollback(_) | Query::Placeholder @@ -248,7 +248,6 @@ impl QueryDiscriminants { QueryDiscriminants::AlterTable, QueryDiscriminants::DropIndex, QueryDiscriminants::Pragma, - ]; } diff --git a/simulator/profiles/query.rs b/simulator/profiles/query.rs index d9eb7bb86..46d8f23b5 100644 --- a/simulator/profiles/query.rs +++ b/simulator/profiles/query.rs @@ -43,7 +43,7 @@ impl Default for QueryProfile { drop_table_weight: 2, alter_table_weight: 2, drop_index: 2, - pragma_weight: 100,//TODO change this back to 2 + pragma_weight: 100, //TODO change this back to 2 } } } diff --git a/sql_generation/model/query/mod.rs b/sql_generation/model/query/mod.rs index 72de42497..f6aa57abe 100644 --- a/sql_generation/model/query/mod.rs +++ b/sql_generation/model/query/mod.rs @@ -13,8 +13,8 @@ pub mod delete; pub mod drop; pub mod drop_index; pub mod insert; +pub mod pragma; pub mod predicate; pub mod select; pub mod transaction; pub mod update; -pub mod pragma;