test(rpc): add test_expect_tx_evicted

Author: ValuedMammal

crates/bitcoind_rpc/tests/test_emitter.rs

@@ -731,3 +731,109 @@ fn no_agreement_point() -> anyhow::Result<()> {
 
     Ok(())
 }
+
+#[test]
+fn test_expect_tx_evicted() -> anyhow::Result<()> {
+    use bdk_bitcoind_rpc::bitcoincore_rpc::bitcoin;
+    use bdk_bitcoind_rpc::bitcoincore_rpc::bitcoincore_rpc_json::CreateRawTransactionInput;
+    use bdk_chain::miniscript;
+    use bdk_chain::spk_txout::SpkTxOutIndex;
+    use bitcoin::constants::genesis_block;
+    use bitcoin::secp256k1::Secp256k1;
+    use bitcoin::Network;
+    use std::collections::HashMap;
+    let env = TestEnv::new()?;
+
+    let s = bdk_testenv::utils::DESCRIPTORS[0];
+    let desc = miniscript::Descriptor::parse_descriptor(&Secp256k1::new(), s)
+        .unwrap()
+        .0;
+    let spk = desc.at_derivation_index(0)?.script_pubkey();
+
+    let mut chain = LocalChain::from_genesis_hash(genesis_block(Network::Regtest).block_hash()).0;
+    let chain_tip = chain.tip().block_id();
+
+    let mut index = SpkTxOutIndex::default();
+    index.insert_spk((), spk.clone());
+    let mut graph = IndexedTxGraph::<BlockId, _>::new(index);
+
+    // Receive tx1.
+    let _ = env.mine_blocks(100, None)?;
+    let txid_1 = env.send(
+        &Address::from_script(&spk, Network::Regtest)?,
+        Amount::ONE_BTC,
+    )?;
+
+    let mut emitter = Emitter::new(env.rpc_client(), chain.tip(), 1, HashSet::from([txid_1]));
+    while let Some(emission) = emitter.next_block()? {
+        let height = emission.block_height();
+        chain.apply_update(CheckPoint::from_header(&emission.block.header, height))?;
+    }
+
+    let changeset = graph.batch_insert_unconfirmed(emitter.mempool()?.new_txs);
+    assert!(changeset
+        .tx_graph
+        .txs
+        .iter()
+        .any(|tx| tx.compute_txid() == txid_1));
+    let seen_at = graph
+        .graph()
+        .get_tx_node(txid_1)
+        .unwrap()
+        .last_seen_unconfirmed
+        .unwrap();
+
+    // Double spend tx1.
+
+    // Get `prevout` from core.
+    let core = env.rpc_client();
+    let tx1 = &core.get_raw_transaction(&txid_1, None)?;
+    let txin = &tx1.input[0];
+    let op = txin.previous_output;
+
+    // Create `tx1b` using the previous output from tx1.
+    let utxo = CreateRawTransactionInput {
+        txid: op.txid,
+        vout: op.vout,
+        sequence: None,
+    };
+    let addr = core.get_new_address(None, None)?.assume_checked();
+    let tx = core.create_raw_transaction(
+        &[utxo],
+        &HashMap::from([(addr.to_string(), Amount::from_btc(49.99)?)]),
+        None,
+        None,
+    )?;
+    let res = core.sign_raw_transaction_with_wallet(&tx, None, None)?;
+    let tx1b = res.transaction()?;
+
+    // Send the tx.
+    let _txid_2 = core.send_raw_transaction(&tx1b)?;
+
+    // Retrieve the expected unconfirmed txids and spks from the graph.
+    let exp_spk_txids = graph
+        .list_expected_spk_txids(&chain, chain_tip, ..)
+        .collect::<Vec<_>>();
+    assert_eq!(exp_spk_txids, vec![(spk, txid_1)]);
+
+    // Check that mempool emission contains evicted txid.
+    let mempool_event = emitter.mempool()?;
+    assert!(mempool_event.evicted_txids.contains(&txid_1));
+
+    // Update graph with evicted tx.
+    let exp_txids = exp_spk_txids.into_iter().map(|(_, txid)| txid);
+    let evicted_txids = mempool_event.evicted_txids(exp_txids);
+    for txid in evicted_txids {
+        let _ = graph.insert_evicted_at(txid, seen_at);
+    }
+
+    let canonical_txids = graph
+        .graph()
+        .list_canonical_txs(&chain, chain_tip)
+        .map(|tx| tx.tx_node.compute_txid())
+        .collect::<Vec<_>>();
+    // tx1 should no longer be canonical.
+    assert!(!canonical_txids.contains(&txid_1));
+
+    Ok(())
+}