Rework sqlite: Instead of only supported one schema (defined in `bdk_sqlite`), we have a schema per changeset type for more flexiblity. * rm `bdk_sqlite` crate (as we don't need `bdk_sqlite::Store` anymore). * add `sqlite` feature on `bdk_chain` which adds methods on each changeset type for initializing tables, loading the changeset and writing. Rework changesets: Some callers may want to use `KeychainTxOutIndex` where `K` may change per descriptor on every run. So we only want to persist the last revealed indices by `DescriptorId` (which uniquely-ish identifies the descriptor). * rm `keychain_added` field from `keychain_txout`'s changeset. * Add `keychain_added` to `CombinedChangeSet` (which is renamed to `WalletChangeSet`). Rework persistence: add back some safety and convenience when persisting our types. Working with changeset directly (as we were doing before) can be cumbersome. * Intoduce `struct Persisted<T>` which wraps a type `T` which stores staged changes to it. This adds safety when creating and or loading `T` from db. * `struct Persisted<T>` methods, `create`, `load` and `persist`, are avaliable if `trait PersistWith<Db>` is implemented for `T`. `Db` represents the database connection and `PersistWith` should be implemented per database-type. * For async, we have `trait PersistedAsyncWith<Db>`. * `Wallet` has impls of `PersistedWith<rusqlite::Connection>`, `PersistedWith<rusqlite::Transaction>` and `PersistedWith<bdk_file_store::Store>` by default. Rework wallet-construction: Before, we had multiple methods for loading and creating with different input-counts so it would be unwieldly to add more parameters in the future. This also makes it difficult to impl `PersistWith` (which has a single method for `load` that takes in `PersistWith::LoadParams` and a single method for `create` that takes in `PersistWith::CreateParams`). * Introduce a builder pattern when constructing a `Wallet`. For loading from persistence or `ChangeSet`, we have `LoadParams`. For creating a new wallet, we have `CreateParams`.
125 lines
3.8 KiB
Rust
125 lines
3.8 KiB
Rust
use bdk_chain::{spk_txout::SpkTxOutIndex, Indexer};
|
|
use bitcoin::{
|
|
absolute, transaction, Amount, OutPoint, ScriptBuf, SignedAmount, Transaction, TxIn, TxOut,
|
|
};
|
|
|
|
#[test]
|
|
fn spk_txout_sent_and_received() {
|
|
let spk1 = ScriptBuf::from_hex("001404f1e52ce2bab3423c6a8c63b7cd730d8f12542c").unwrap();
|
|
let spk2 = ScriptBuf::from_hex("00142b57404ae14f08c3a0c903feb2af7830605eb00f").unwrap();
|
|
|
|
let mut index = SpkTxOutIndex::default();
|
|
index.insert_spk(0, spk1.clone());
|
|
index.insert_spk(1, spk2.clone());
|
|
|
|
let tx1 = Transaction {
|
|
version: transaction::Version::TWO,
|
|
lock_time: absolute::LockTime::ZERO,
|
|
input: vec![],
|
|
output: vec![TxOut {
|
|
value: Amount::from_sat(42_000),
|
|
script_pubkey: spk1.clone(),
|
|
}],
|
|
};
|
|
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, ..),
|
|
(Amount::from_sat(0), Amount::from_sat(42_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, ..1),
|
|
(Amount::from_sat(0), Amount::from_sat(42_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, 1..),
|
|
(Amount::from_sat(0), Amount::from_sat(0))
|
|
);
|
|
assert_eq!(index.net_value(&tx1, ..), SignedAmount::from_sat(42_000));
|
|
index.index_tx(&tx1);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, ..),
|
|
(Amount::from_sat(0), Amount::from_sat(42_000)),
|
|
"shouldn't change after scanning"
|
|
);
|
|
|
|
let tx2 = Transaction {
|
|
version: transaction::Version::ONE,
|
|
lock_time: absolute::LockTime::ZERO,
|
|
input: vec![TxIn {
|
|
previous_output: OutPoint {
|
|
txid: tx1.compute_txid(),
|
|
vout: 0,
|
|
},
|
|
..Default::default()
|
|
}],
|
|
output: vec![
|
|
TxOut {
|
|
value: Amount::from_sat(20_000),
|
|
script_pubkey: spk2,
|
|
},
|
|
TxOut {
|
|
script_pubkey: spk1,
|
|
value: Amount::from_sat(30_000),
|
|
},
|
|
],
|
|
};
|
|
|
|
assert_eq!(
|
|
index.sent_and_received(&tx2, ..),
|
|
(Amount::from_sat(42_000), Amount::from_sat(50_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx2, ..1),
|
|
(Amount::from_sat(42_000), Amount::from_sat(30_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx2, 1..),
|
|
(Amount::from_sat(0), Amount::from_sat(20_000))
|
|
);
|
|
assert_eq!(index.net_value(&tx2, ..), SignedAmount::from_sat(8_000));
|
|
}
|
|
|
|
#[test]
|
|
fn mark_used() {
|
|
let spk1 = ScriptBuf::from_hex("001404f1e52ce2bab3423c6a8c63b7cd730d8f12542c").unwrap();
|
|
let spk2 = ScriptBuf::from_hex("00142b57404ae14f08c3a0c903feb2af7830605eb00f").unwrap();
|
|
|
|
let mut spk_index = SpkTxOutIndex::default();
|
|
spk_index.insert_spk(1, spk1.clone());
|
|
spk_index.insert_spk(2, spk2);
|
|
|
|
assert!(!spk_index.is_used(&1));
|
|
spk_index.mark_used(&1);
|
|
assert!(spk_index.is_used(&1));
|
|
spk_index.unmark_used(&1);
|
|
assert!(!spk_index.is_used(&1));
|
|
spk_index.mark_used(&1);
|
|
assert!(spk_index.is_used(&1));
|
|
|
|
let tx1 = Transaction {
|
|
version: transaction::Version::TWO,
|
|
lock_time: absolute::LockTime::ZERO,
|
|
input: vec![],
|
|
output: vec![TxOut {
|
|
value: Amount::from_sat(42_000),
|
|
script_pubkey: spk1,
|
|
}],
|
|
};
|
|
|
|
spk_index.index_tx(&tx1);
|
|
spk_index.unmark_used(&1);
|
|
assert!(
|
|
spk_index.is_used(&1),
|
|
"even though we unmark_used it doesn't matter because there was a tx scanned that used it"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn unmark_used_does_not_result_in_invalid_representation() {
|
|
let mut spk_index = SpkTxOutIndex::default();
|
|
assert!(!spk_index.unmark_used(&0));
|
|
assert!(!spk_index.unmark_used(&1));
|
|
assert!(!spk_index.unmark_used(&2));
|
|
assert!(spk_index.unused_spks(..).collect::<Vec<_>>().is_empty());
|
|
}
|