46 lines
1.4 KiB
Rust
46 lines
1.4 KiB
Rust
use proptest::{arbitrary::any, collection::vec, prelude::*};
|
|
|
|
use crate::{block, LedgerState};
|
|
|
|
use super::{CoinbaseData, Input, OutPoint, Script};
|
|
|
|
impl Input {
|
|
/// Construct a strategy for creating validish vecs of Inputs.
|
|
pub fn vec_strategy(ledger_state: LedgerState, max_size: usize) -> BoxedStrategy<Vec<Self>> {
|
|
if ledger_state.is_coinbase {
|
|
let height = block::Height(ledger_state.tip_height.0 + 1);
|
|
Self::arbitrary_with(Some(height))
|
|
.prop_map(|input| vec![input])
|
|
.boxed()
|
|
} else {
|
|
vec(Self::arbitrary_with(None), max_size).boxed()
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Arbitrary for Input {
|
|
type Parameters = Option<block::Height>;
|
|
|
|
fn arbitrary_with(height: Self::Parameters) -> Self::Strategy {
|
|
if let Some(height) = height {
|
|
(vec(any::<u8>(), 0..95), any::<u32>())
|
|
.prop_map(move |(data, sequence)| Input::Coinbase {
|
|
height,
|
|
data: CoinbaseData(data),
|
|
sequence,
|
|
})
|
|
.boxed()
|
|
} else {
|
|
(any::<OutPoint>(), any::<Script>(), any::<u32>())
|
|
.prop_map(|(outpoint, unlock_script, sequence)| Input::PrevOut {
|
|
outpoint,
|
|
unlock_script,
|
|
sequence,
|
|
})
|
|
.boxed()
|
|
}
|
|
}
|
|
|
|
type Strategy = BoxedStrategy<Self>;
|
|
}
|