feat: add utility to check local_chain persistence

Author: 110CodingP

crates/testenv/src/persist_test_utils.rs

@@ -1,13 +1,14 @@
 use crate::block_id;
 use bdk_chain::bitcoin::{self, hashes::Hash};
+use bdk_chain::miniscript::{Descriptor, DescriptorPublicKey};
 use bdk_chain::{
     bitcoin::{absolute, transaction, Amount, OutPoint, ScriptBuf, Transaction, TxIn, TxOut, Txid},
     indexer::keychain_txout,
-    tx_graph, ConfirmationBlockTime, Merge, DescriptorExt
+    local_chain, tx_graph, ConfirmationBlockTime, DescriptorExt, Merge
 };
+use crate::hash;
 use std::path::Path;
 use std::sync::Arc;
-use bdk_chain::miniscript::{Descriptor, DescriptorPublicKey};
 
 pub fn create_one_inp_one_out_tx(txid: Txid, amount: u64) -> Transaction {
     Transaction {
@@ -143,7 +144,6 @@ pub fn persist_indexer_changeset<Db, CreateDb, Initialize, Persist>(
     let changeset = initialize(&mut db).expect("should load empty changeset");
     assert_eq!(changeset, keychain_txout::ChangeSet::default());
 
-
     persist(&mut db, &keychain_txout_changeset).expect("should persist keychain_txout");
 
     let changeset = initialize(&mut db).expect("should load persisted changeset");
@@ -167,4 +167,50 @@ pub fn persist_indexer_changeset<Db, CreateDb, Initialize, Persist>(
     assert_eq!(changeset_new, keychain_txout_changeset);
 }
 
+pub fn persist_local_chain_changeset<Db, CreateDb, Initialize, Persist>(
+    filename: &str,
+    create_db: CreateDb,
+    initialize: Initialize,
+    persist: Persist,
+) where
+    CreateDb: Fn(&Path) -> anyhow::Result<Db>,
+    Initialize: Fn(&mut Db) -> anyhow::Result<local_chain::ChangeSet>,
+    Persist: Fn(&mut Db, &local_chain::ChangeSet) -> anyhow::Result<()>,
+{
+    let temp_dir = tempfile::tempdir().expect("must create tempdir");
+    let file_path = temp_dir.path().join(filename);
+    let mut db = create_db(&file_path).expect("db should get created");
+
+    // create a local_chain_changeset, persist that and read it
+    let local_chain_changeset = local_chain::ChangeSet {
+        blocks: [
+            (0, Some(hash!("B"))),
+            (1, Some(hash!("D"))),
+        ]
+        .into(),
+    };
+    let changeset = initialize(&mut db).expect("should load empty changeset");
+    assert_eq!(changeset, local_chain::ChangeSet::default());
+
+    persist(&mut db, &local_chain_changeset).expect("should persist changeset");
+
+    let changeset = initialize(&mut db).expect("should load persisted changeset");
+    assert_eq!(local_chain_changeset, changeset);
+
+    // create another local_chain_changeset, persist that and read it
+    let local_chain_changeset = local_chain::ChangeSet {
+        blocks: [(2, Some(hash!("K")))].into(),
+    };
+
+    persist(&mut db, &local_chain_changeset).expect("should persist changeset");
+
+    let changeset = initialize(&mut db).unwrap();
+
+    let local_chain_changeset = local_chain::ChangeSet {
+        blocks: [(0, Some(hash!("B"))), (1, Some(hash!("D"))), (2, Some(hash!("K")))].into(),
+    };
+
+    assert_eq!(local_chain_changeset, changeset);
+}
+
 // perhaps add test for file_store, sqlite, redb here.