test: Add multiple wallet syncing

This test takes the first wallet to a stale block, then shuts down the
node. A new wallet is introduced, and we assert that the first wallet
still detects and applies the reorganization.

Author: rustaceanrob

tests/client.rs

@@ -1,5 +1,7 @@
 // #![allow(unused)]
 use bdk_kyoto::{Idle, Single};
+use bdk_wallet::chain::{DescriptorExt, DescriptorId};
+use std::collections::BTreeMap;
 use std::net::IpAddr;
 use std::path::PathBuf;
 use std::time::Duration;
@@ -17,6 +19,8 @@ use bdk_wallet::Update;
 
 const EXTERNAL_DESCRIPTOR: &str = "tr([7d94197e/86'/1'/0']tpubDCyQVJj8KzjiQsFjmb3KwECVXPvMwvAxxZGCP9XmWSopmjW3bCV3wD7TgxrUhiGSueDS1MU5X1Vb1YjYcp8jitXc5fXfdC1z68hDDEyKRNr/0/*)";
 const INTERNAL_DESCRIPTOR: &str = "tr([7d94197e/86'/1'/0']tpubDCyQVJj8KzjiQsFjmb3KwECVXPvMwvAxxZGCP9XmWSopmjW3bCV3wD7TgxrUhiGSueDS1MU5X1Vb1YjYcp8jitXc5fXfdC1z68hDDEyKRNr/1/*)";
+const RECV_TWO: &str = "wpkh([9122d9e0/84'/1'/0']tpubDCYVtmaSaDzTxcgvoP5AHZNbZKZzrvoNH9KARep88vESc6MxRqAp4LmePc2eeGX6XUxBcdhAmkthWTDqygPz2wLAyHWisD299Lkdrj5egY6/0/*)";
+const CHANGE_TWO: &str = "wpkh([9122d9e0/84'/1'/0']tpubDCYVtmaSaDzTxcgvoP5AHZNbZKZzrvoNH9KARep88vESc6MxRqAp4LmePc2eeGX6XUxBcdhAmkthWTDqygPz2wLAyHWisD299Lkdrj5egY6/1/*)";
 
 fn testenv() -> anyhow::Result<TestEnv> {
     use bdk_testenv::Config;
@@ -239,3 +243,92 @@ async fn update_handles_dormant_wallet() -> anyhow::Result<()> {
 
     Ok(())
 }
+
+#[tokio::test]
+async fn two_wallets_can_update() -> anyhow::Result<()> {
+    let env = testenv()?;
+
+    let mut wallet = CreateParams::new(EXTERNAL_DESCRIPTOR, INTERNAL_DESCRIPTOR)
+        .network(Network::Regtest)
+        .create_wallet_no_persist()?;
+    let addr = wallet.peek_address(KeychainKind::External, 0).address;
+
+    let tempdir = tempfile::tempdir()?.path().join("kyoto-data");
+    let client = init_node(&env, &wallet, tempdir.clone())?;
+    let (client, _, mut update_subscriber) = client.subscribe();
+    let client = client.start();
+    let requester = client.requester();
+
+    // mine blocks
+    let miner = env
+        .rpc_client()
+        .get_new_address(None, None)?
+        .assume_checked();
+    let _hashes = env.mine_blocks(100, Some(miner.clone()))?;
+    wait_for_height(&env, 101).await?;
+
+    // send tx
+    let amt = Amount::from_btc(0.21)?;
+    let txid = env.send(&addr, amt)?;
+    let hashes = env.mine_blocks(1, Some(miner.clone()))?;
+    let blockhash = hashes[0];
+    wait_for_height(&env, 102).await?;
+
+    // get update
+    let res = update_subscriber.update().await?;
+    let (anchor, anchor_txid) = *res.tx_update.anchors.iter().next().unwrap();
+    assert_eq!(anchor.block_id.hash, blockhash);
+    assert_eq!(anchor_txid, txid);
+    wallet.apply_update(res).unwrap();
+
+    // shut down then reorg
+    requester.shutdown()?;
+
+    let hashes = env.reorg(1)?; // 102
+    let new_blockhash = hashes[0];
+    _ = env.mine_blocks(20, Some(miner))?; // 122
+    wait_for_height(&env, 122).await?;
+
+    // add a new wallet to the sync request
+    let wallet_two = CreateParams::new(RECV_TWO, CHANGE_TWO)
+        .network(Network::Regtest)
+        .create_wallet_no_persist()?;
+    let peer = env.bitcoind.params.p2p_socket.unwrap();
+    let ip: IpAddr = (*peer.ip()).into();
+    let port = peer.port();
+    let mut peer = TrustedPeer::from_ip(ip);
+    peer.port = Some(port);
+    let client = Builder::new(Network::Regtest)
+        .add_peer(peer)
+        .data_dir(tempdir)
+        .required_peers(1)
+        .build_with_wallets(vec![
+            (&wallet, ScanType::Sync),
+            (&wallet_two, ScanType::Sync),
+        ])?;
+    let (client, _, mut update_subscriber) = client.subscribe();
+    let client = client.start();
+    let requester = client.requester();
+
+    // expect tx to confirm at same height but different blockhash
+    let results = update_subscriber.updates().await?;
+    let res = results
+        .collect::<BTreeMap<DescriptorId, Update>>()
+        .get(
+            &wallet
+                .public_descriptor(KeychainKind::External)
+                .descriptor_id(),
+        )
+        .unwrap()
+        .clone();
+    let (anchor, anchor_txid) = *res.tx_update.anchors.iter().next().unwrap();
+    assert_eq!(anchor_txid, txid);
+    assert_eq!(anchor.block_id.height, 102);
+    assert_ne!(anchor.block_id.hash, blockhash);
+    assert_eq!(anchor.block_id.hash, new_blockhash);
+    wallet.apply_update(res).unwrap();
+
+    requester.shutdown()?;
+
+    Ok(())
+}