49 lines
1.3 KiB
Rust
49 lines
1.3 KiB
Rust
//! Mock [`ChainTip`]s for use in tests.
|
|
|
|
use std::sync::Arc;
|
|
|
|
use tokio::sync::watch;
|
|
|
|
use crate::{block, chain_tip::ChainTip, transaction};
|
|
|
|
/// A sender that sets the `best_tip_height` of a [`MockChainTip`].]
|
|
pub type MockChainTipSender = watch::Sender<Option<block::Height>>;
|
|
|
|
/// A mock [`ChainTip`] implementation that allows setting the `best_tip_height` externally.
|
|
#[derive(Clone, Debug)]
|
|
pub struct MockChainTip {
|
|
best_tip_height: watch::Receiver<Option<block::Height>>,
|
|
}
|
|
|
|
impl MockChainTip {
|
|
/// Create a new [`MockChainTip`].
|
|
///
|
|
/// Returns the [`MockChainTip`] instance and the endpoint to modiy the current best tip
|
|
/// height.
|
|
///
|
|
/// Initially, the best tip height is [`None`].
|
|
pub fn new() -> (Self, MockChainTipSender) {
|
|
let (sender, receiver) = watch::channel(None);
|
|
|
|
let mock_chain_tip = MockChainTip {
|
|
best_tip_height: receiver,
|
|
};
|
|
|
|
(mock_chain_tip, sender)
|
|
}
|
|
}
|
|
|
|
impl ChainTip for MockChainTip {
|
|
fn best_tip_height(&self) -> Option<block::Height> {
|
|
*self.best_tip_height.borrow()
|
|
}
|
|
|
|
fn best_tip_hash(&self) -> Option<block::Hash> {
|
|
unreachable!("Method not used in tests");
|
|
}
|
|
|
|
fn best_tip_mined_transaction_ids(&self) -> Arc<[transaction::Hash]> {
|
|
unreachable!("Method not used in tests");
|
|
}
|
|
}
|