db actions

Author: armaganyildirak

fuzz/fuzz_targets/standard/fuzz_database_states.rs

@@ -9,8 +9,8 @@ use ssv_types::{
     Cluster, ClusterId, OperatorId, Share, ValidatorIndex, ValidatorMetadata, ENCRYPTED_KEY_LENGTH,
 };
 use ssz::Decode;
-use tempfile::NamedTempFile;
 use std::{path::Path, sync::OnceLock};
+use tempfile::NamedTempFile;
 use types::{Address, Graffiti, PublicKeyBytes};
 
 const TESTING_KEY: &str = "-----BEGIN PRIVATE KEY-----
@@ -44,7 +44,15 @@ SPdvWXU4osCd7vgiJvAP4ek=
 
 static RSA_KEY: OnceLock<Rsa<openssl::pkey::Public>> = OnceLock::new();
 
-#[derive(Arbitrary)]
+#[derive(Arbitrary, Debug)]
+enum DbAction {
+    InsertValidator,
+    UpdateStatus(bool),
+    DeleteValidator,
+    BumpNonce,
+}
+
+#[derive(Arbitrary, Debug)]
 struct FuzzData {
     cluster_id: [u8; 32],
     owner: [u8; 20],
@@ -56,6 +64,7 @@ struct FuzzData {
     encrypted_private_key: [u8; ENCRYPTED_KEY_LENGTH],
     validator_index: usize,
     graffiti: [u8; 32],
+    actions: Vec<DbAction>,
 }
 
 fn fuzz_with_data(fuzz: FuzzData) {
@@ -117,22 +126,37 @@ fn fuzz_with_data(fuzz: FuzzData) {
         encrypted_private_key: fuzz.encrypted_private_key,
     };
 
-    // Test insert_validator
-    let Ok(conn) = old.connection() else {
-        return;
-    };
-    let Ok(tx) = conn.unchecked_transaction() else {
-        return;
-    };
+    for action in fuzz.actions {
+        let Ok(conn) = old.connection() else {
+            return;
+        };
+        let Ok(tx) = conn.unchecked_transaction() else {
+            return;
+        };
 
-    let _ = old.insert_validator(cluster, &validator, vec![share], &tx);
-    let _ = tx.commit();
+        match action {
+            DbAction::InsertValidator => {
+                let _ = old.insert_validator(cluster.clone(), &validator, vec![share.clone()], &tx);
+            }
+            DbAction::UpdateStatus(status) => {
+                let _ = old.update_status(ClusterId(fuzz.cluster_id), status, &tx);
+            }
+            DbAction::DeleteValidator => {
+                let _ = old.delete_validator(&validator.public_key, &tx);
+            }
+            DbAction::BumpNonce => {
+                let _ = old.bump_and_get_nonce(&cluster.owner, &tx);
+            }
+        }
+
+        let _ = tx.commit();
+    }
 
     let Ok(new) = NetworkDatabase::new(Path::new(&db_path), rsa_key, Default::default()) else {
         return;
     };
 
-    // Compare states. I'm not sure about this part.
+    // Compare states
     let old_state = old.state();
     let new_state = new.state();
 
@@ -152,7 +176,6 @@ fn fuzz_with_data(fuzz: FuzzData) {
     let old_clusters: Vec<_> = old_state.clusters().values().cloned().collect();
     let new_clusters: Vec<_> = new_state.clusters().values().cloned().collect();
     assert_eq!(old_clusters, new_clusters, "clusters differ");
-
 }
 
 fn main() {