Add functions to test persistence

crates/chain/Cargo.toml

@@ -29,6 +29,8 @@ rand = "0.8"
 proptest = "1.2.0"
 bdk_testenv = { path = "../testenv" }
 criterion = { version = "0.7" }
+tempfile = "3"
+anyhow = "1"
 
 [features]
 default = ["std", "miniscript"]

crates/chain/tests/test_rusqlite_impl.rs

@@ -0,0 +1,91 @@
+#![cfg(feature = "rusqlite")]
+use anyhow::anyhow;
+use bdk_chain::{keychain_txout, local_chain, tx_graph, ConfirmationBlockTime};
+use bdk_testenv::persist_test_utils::{
+    assert_persist_changesets, keychain_txout_changesets, local_chain_changesets,
+    tx_graph_changesets,
+};
+
+#[test]
+fn txgraph_is_persisted() -> anyhow::Result<()> {
+    let temp_dir = tempfile::tempdir().unwrap();
+    let changesets = tx_graph_changesets();
+    assert_persist_changesets(
+        || {
+            let mut db = rusqlite::Connection::open(temp_dir.path().join("wallet.sqlite"))?;
+            let db_tx = db.transaction()?;
+            tx_graph::ChangeSet::<ConfirmationBlockTime>::init_sqlite_tables(&db_tx)?;
+            db_tx.commit()?;
+            Ok(db)
+        },
+        |db| {
+            let db_tx = db.transaction()?;
+            let changeset = tx_graph::ChangeSet::<ConfirmationBlockTime>::from_sqlite(&db_tx)?;
+            Ok(changeset)
+        },
+        |db, changeset| {
+            let db_tx = db.transaction()?;
+            changeset.persist_to_sqlite(&db_tx)?;
+            db_tx.commit()?;
+            Ok(())
+        },
+        &changesets,
+    )
+    .map_err(|err| anyhow!(err))
+}
+
+#[test]
+fn indexer_is_persisted() -> anyhow::Result<()> {
+    let temp_dir = tempfile::tempdir().unwrap();
+    let changesets = keychain_txout_changesets();
+    assert_persist_changesets(
+        || {
+            let mut db = rusqlite::Connection::open(temp_dir.path().join("wallet.sqlite"))?;
+            let db_tx = db.transaction()?;
+            keychain_txout::ChangeSet::init_sqlite_tables(&db_tx)?;
+            db_tx.commit()?;
+            Ok(db)
+        },
+        |db| {
+            let db_tx = db.transaction()?;
+            let changeset = keychain_txout::ChangeSet::from_sqlite(&db_tx)?;
+            Ok(changeset)
+        },
+        |db, changeset| {
+            let db_tx = db.transaction()?;
+            changeset.persist_to_sqlite(&db_tx)?;
+            db_tx.commit()?;
+            Ok(())
+        },
+        &changesets,
+    )
+    .map_err(|err| anyhow!(err))
+}
+
+#[test]
+fn local_chain_is_persisted() -> anyhow::Result<()> {
+    let temp_dir = tempfile::tempdir().unwrap();
+    let changesets = local_chain_changesets();
+    assert_persist_changesets(
+        || {
+            let mut db = rusqlite::Connection::open(temp_dir.path().join("wallet.sqlite"))?;
+            let db_tx = db.transaction()?;
+            local_chain::ChangeSet::init_sqlite_tables(&db_tx)?;
+            db_tx.commit()?;
+            Ok(db)
+        },
+        |db| {
+            let db_tx = db.transaction()?;
+            let changeset = local_chain::ChangeSet::from_sqlite(&db_tx)?;
+            Ok(changeset)
+        },
+        |db, changeset| {
+            let db_tx = db.transaction()?;
+            changeset.persist_to_sqlite(&db_tx)?;
+            db_tx.commit()?;
+            Ok(())
+        },
+        &changesets,
+    )
+    .map_err(|err| anyhow!(err))
+}

crates/file_store/Cargo.toml

@@ -21,3 +21,6 @@ serde = { version = "1", features = ["derive"] }
 
 [dev-dependencies]
 tempfile = "3"
+anyhow = { version = "1", default-features = false}
+bdk_testenv = {path = "../testenv"}
+bdk_chain = { path = "../chain", version = "0.23.1", default-features = false, features = ["serde"]}

crates/file_store/src/store.rs

@@ -291,10 +291,17 @@ mod test {
         io::{Seek, Write},
     };
 
+    use anyhow::anyhow;
+
     const TEST_MAGIC_BYTES_LEN: usize = 12;
     const TEST_MAGIC_BYTES: [u8; TEST_MAGIC_BYTES_LEN] =
         [98, 100, 107, 102, 115, 49, 49, 49, 49, 49, 49, 49];
 
+    use bdk_testenv::persist_test_utils::{
+        assert_persist_changesets, keychain_txout_changesets, local_chain_changesets,
+        tx_graph_changesets,
+    };
+
     type TestChangeSet = BTreeSet<String>;
 
     /// Check behavior of [`Store::create`] and [`Store::load`].
@@ -599,4 +606,43 @@ mod test {
         // current position matches EOF
         assert_eq!(current_pointer, expected_pointer);
     }
+
+    #[test]
+    fn txgraph_is_persisted() -> anyhow::Result<()> {
+        let temp_dir = tempfile::tempdir().unwrap();
+        let changesets = tx_graph_changesets();
+        assert_persist_changesets(
+            || Ok(Store::load_or_create(&TEST_MAGIC_BYTES, temp_dir.path().join("store.db"))?.0),
+            |db| Ok(db.dump().map(Option::unwrap_or_default)?),
+            |db, changeset| Ok(db.append(changeset)?),
+            &changesets,
+        )
+        .map_err(|err| anyhow!(err))
+    }
+
+    #[test]
+    fn indexer_is_persisted() -> anyhow::Result<()> {
+        let temp_dir = tempfile::tempdir().unwrap();
+        let changesets = keychain_txout_changesets();
+        assert_persist_changesets(
+            || Ok(Store::load_or_create(&TEST_MAGIC_BYTES, temp_dir.path().join("store.db"))?.0),
+            |db| Ok(db.dump().map(Option::unwrap_or_default)?),
+            |db, changeset| Ok(db.append(changeset)?),
+            &changesets,
+        )
+        .map_err(|err| anyhow!(err))
+    }
+
+    #[test]
+    fn local_chain_is_persisted() -> anyhow::Result<()> {
+        let temp_dir = tempfile::tempdir().unwrap();
+        let changesets = local_chain_changesets();
+        assert_persist_changesets(
+            || Ok(Store::load_or_create(&TEST_MAGIC_BYTES, temp_dir.path().join("store.db"))?.0),
+            |db| Ok(db.dump().map(Option::unwrap_or_default)?),
+            |db, changeset| Ok(db.append(changeset)?),
+            &changesets,
+        )
+        .map_err(|err| anyhow!(err))
+    }
 }

crates/testenv/Cargo.toml

@@ -24,10 +24,10 @@ bitcoin = { version = "0.32.0", default-features = false }
 bdk_testenv = { path = "." }
 
 [features]
-default = ["std", "download"]
+default = ["std", "download", "miniscript"]
 download = ["electrsd/bitcoind_download", "electrsd/bitcoind_28_2", "electrsd/esplora_a33e97e1"]
 std = ["bdk_chain/std", "bitcoin/rand-std"]
 serde = ["bdk_chain/serde"]
-
+miniscript = ["bdk_chain/miniscript"]
 [package.metadata.docs.rs]
 no-default-features = true

crates/testenv/src/lib.rs

@@ -1,5 +1,6 @@
 #![cfg_attr(coverage_nightly, feature(coverage_attribute))]
 
+pub mod persist_test_utils;
 pub mod utils;
 
 use anyhow::Context;
@@ -14,6 +15,8 @@ use core::time::Duration;
 use electrsd::bitcoind::mtype::GetBlockTemplate;
 use electrsd::bitcoind::{TemplateRequest, TemplateRules};
 
+extern crate alloc;
+
 pub use electrsd;
 pub use electrsd::bitcoind;
 pub use electrsd::bitcoind::anyhow;