feat(torture):optional checks after commit & crash

Author: gabriele-0201

torture/src/supervisor/cli.rs

@@ -70,4 +70,14 @@ pub struct WorkloadParams {
     #[clap(value_parser=clap::value_parser!(u8).range(0..=100))]
     #[arg(long = "workload-commit-crash")]
     pub crash: u8,
+
+    /// Whether to ensure the correct application of the changest after every commit.
+    #[clap(default_value = "false")]
+    #[arg(long = "ensure-changeset")]
+    pub ensure_changeset: bool,
+
+    /// Whether to ensure the correctness of the state after every crash.
+    #[clap(default_value = "false")]
+    #[arg(long = "ensure-snapshot")]
+    pub ensure_snapshot: bool,
 }

torture/src/supervisor/workload.rs

@@ -202,6 +202,10 @@ pub struct Workload {
     /// Data collected to evaluate the average commit time.
     tot_commit_time: u64,
     n_successfull_commit: u64,
+    /// Whether to ensure the correct application of the changest after every commit.
+    ensure_changeset: bool,
+    /// Whether to ensure the correctness of the state after every crash.
+    ensure_snapshot: bool,
 }
 
 impl Workload {
@@ -240,6 +244,8 @@ impl Workload {
             bitbox_seed,
             tot_commit_time: 0,
             n_successfull_commit: 0,
+            ensure_changeset: workload_params.ensure_changeset,
+            ensure_snapshot: workload_params.ensure_snapshot,
         }
     }
 
@@ -322,6 +328,10 @@ impl Workload {
         rr: &comms::RequestResponse,
         changeset: &Vec<KeyValueChange>,
     ) -> anyhow::Result<()> {
+        if !self.ensure_changeset {
+            return Ok(());
+        }
+
         for change in changeset {
             match change {
                 KeyValueChange::Insert(key, value)
@@ -400,6 +410,10 @@ impl Workload {
         rr: &comms::RequestResponse,
         changeset: &Vec<KeyValueChange>,
     ) -> anyhow::Result<()> {
+        if !self.ensure_changeset {
+            return Ok(());
+        }
+
         // Given a reverted changeset, we need to ensure that each modified/deleted key
         // is equal to its previous state and that each new key is not avaible.
         for change in changeset {
@@ -435,6 +449,10 @@ impl Workload {
     }
 
     async fn ensure_snapshot_validity(&self, rr: &comms::RequestResponse) -> anyhow::Result<()> {
+        if !self.ensure_snapshot {
+            return Ok(());
+        }
+
         for (i, (key, expected_value)) in (self.state.committed.state.iter()).enumerate() {
             let value = rr.send_request_query(*key).await?;
             if &value != expected_value {