forked from bitcoindevkit/bdk
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
38ef170
commit a38f633
Showing
5 changed files
with
143 additions
and
137 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,117 +1 @@ | ||
| use bdk_chain::{ | ||
| bitcoin::Transaction, | ||
| keychain::{KeychainChangeSet, KeychainTracker}, | ||
| TxHeight, | ||
| }; | ||
| use bdk_file_store::{FileError, IterError, KeychainStore, MAGIC_BYTES, MAGIC_BYTES_LEN}; | ||
| use std::{ | ||
| io::{Read, Write}, | ||
| vec::Vec, | ||
| }; | ||
| use tempfile::NamedTempFile; | ||
| #[derive( | ||
| Debug, Clone, Copy, PartialOrd, Ord, PartialEq, Eq, Hash, serde::Serialize, serde::Deserialize, | ||
| )] | ||
| enum TestKeychain { | ||
| External, | ||
| Internal, | ||
| } | ||
|
|
||
| impl core::fmt::Display for TestKeychain { | ||
| fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { | ||
| match self { | ||
| Self::External => write!(f, "external"), | ||
| Self::Internal => write!(f, "internal"), | ||
| } | ||
| } | ||
| } | ||
|
|
||
| #[test] | ||
| fn magic_bytes() { | ||
| assert_eq!(&MAGIC_BYTES, "bdkfs0000000".as_bytes()); | ||
| } | ||
|
|
||
| #[test] | ||
| fn new_fails_if_file_is_too_short() { | ||
| let mut file = NamedTempFile::new().unwrap(); | ||
| file.write_all(&MAGIC_BYTES[..MAGIC_BYTES_LEN - 1]) | ||
| .expect("should write"); | ||
|
|
||
| match KeychainStore::<TestKeychain, TxHeight, Transaction>::new(file.reopen().unwrap()) { | ||
| Err(FileError::Io(e)) => assert_eq!(e.kind(), std::io::ErrorKind::UnexpectedEof), | ||
| unexpected => panic!("unexpected result: {:?}", unexpected), | ||
| }; | ||
| } | ||
|
|
||
| #[test] | ||
| fn new_fails_if_magic_bytes_are_invalid() { | ||
| let invalid_magic_bytes = "ldkfs0000000"; | ||
|
|
||
| let mut file = NamedTempFile::new().unwrap(); | ||
| file.write_all(invalid_magic_bytes.as_bytes()) | ||
| .expect("should write"); | ||
|
|
||
| match KeychainStore::<TestKeychain, TxHeight, Transaction>::new(file.reopen().unwrap()) { | ||
| Err(FileError::InvalidMagicBytes(b)) => { | ||
| assert_eq!(b, invalid_magic_bytes.as_bytes()) | ||
| } | ||
| unexpected => panic!("unexpected result: {:?}", unexpected), | ||
| }; | ||
| } | ||
|
|
||
| #[test] | ||
| fn append_changeset_truncates_invalid_bytes() { | ||
| use bdk_chain::miniscript; | ||
| use core::str::FromStr; | ||
| // initial data to write to file (magic bytes + invalid data) | ||
| let mut data = [255_u8; 2000]; | ||
| data[..MAGIC_BYTES_LEN].copy_from_slice(&MAGIC_BYTES); | ||
|
|
||
| let descriptor = miniscript::Descriptor::from_str("tr([73c5da0a/86'/0'/0']xpub6BgBgsespWvERF3LHQu6CnqdvfEvtMcQjYrcRzx53QJjSxarj2afYWcLteoGVky7D3UKDP9QyrLprQ3VCECoY49yfdDEHGCtMMj92pReUsQ/0/*)#rg247h69").unwrap(); | ||
| let mut tracker = KeychainTracker::<TestKeychain, TxHeight, Transaction>::default(); | ||
| tracker.add_keychain(TestKeychain::External, descriptor); | ||
| let changeset = KeychainChangeSet { | ||
| derivation_indices: tracker | ||
| .txout_index | ||
| .reveal_to_target(&TestKeychain::External, 21) | ||
| .1, | ||
| chain_graph: Default::default(), | ||
| }; | ||
|
|
||
| let mut file = NamedTempFile::new().unwrap(); | ||
| file.write_all(&data).expect("should write"); | ||
|
|
||
| let mut store = | ||
| KeychainStore::<TestKeychain, TxHeight, Transaction>::new(file.reopen().unwrap()) | ||
| .expect("should open"); | ||
| match store.iter_changesets().expect("seek should succeed").next() { | ||
| Some(Err(IterError::Bincode(_))) => {} | ||
| unexpected_res => panic!("unexpected result: {:?}", unexpected_res), | ||
| } | ||
|
|
||
| store.append_changeset(&changeset).expect("should append"); | ||
|
|
||
| drop(store); | ||
|
|
||
| let got_bytes = { | ||
| let mut buf = Vec::new(); | ||
| file.reopen() | ||
| .unwrap() | ||
| .read_to_end(&mut buf) | ||
| .expect("should read"); | ||
| buf | ||
| }; | ||
|
|
||
| let expected_bytes = { | ||
| let mut buf = MAGIC_BYTES.to_vec(); | ||
| bincode::encode_into_std_write( | ||
| bincode::serde::Compat(&changeset), | ||
| &mut buf, | ||
| bincode::config::standard(), | ||
| ) | ||
| .expect("should encode"); | ||
| buf | ||
| }; | ||
|
|
||
| assert_eq!(got_bytes, expected_bytes); | ||
| } |