diff --git a/chain-impl-mockchain/src/testing/builders/initial_builder.rs b/chain-impl-mockchain/src/testing/builders/initial_builder.rs index 43d09ee72..4b1e5e07c 100644 --- a/chain-impl-mockchain/src/testing/builders/initial_builder.rs +++ b/chain-impl-mockchain/src/testing/builders/initial_builder.rs @@ -1,5 +1,5 @@ use crate::{ - certificate::{Certificate, PoolUpdate, VoteCast, VotePlan}, + certificate::{Certificate, PoolUpdate, VoteCast, VotePlan, VoteTally}, fragment::Fragment, key::EitherEd25519SecretKey, ledger::ledger::OutputAddress, @@ -59,6 +59,16 @@ pub fn create_initial_vote_cast(vote_cast: &VoteCast, owners: &[Wallet]) -> Frag fragment(cert, keys, &[], &[]) } +pub fn create_initial_vote_tally(vote_tally: &VoteTally, owners: &[Wallet]) -> Fragment { + let cert: Certificate = vote_tally.clone().into(); + let keys: Vec = owners + .iter() + .cloned() + .map(|owner| owner.private_key()) + .collect(); + fragment(cert, keys, &[], &[]) +} + pub fn create_initial_transaction(wallet: &Wallet) -> Fragment { let tx = TxBuilder::new() .set_nopayload()