diff --git a/signer/src/testing/transaction_signer.rs b/signer/src/testing/transaction_signer.rs index b53b38f4..d4fb5a2a 100644 --- a/signer/src/testing/transaction_signer.rs +++ b/signer/src/testing/transaction_signer.rs @@ -93,8 +93,12 @@ where C: Context, { /// Wait for `expected` instances of the given event `msg`, timing out after `timeout`. - pub async fn wait_for_events(&mut self, msg: TxSignerEvent, expected: u16, timeout: Duration) -> Result<(), Elapsed> { - + pub async fn wait_for_events( + &mut self, + msg: TxSignerEvent, + expected: u16, + timeout: Duration, + ) -> Result<(), Elapsed> { let future = async { let mut n = 0; loop { @@ -112,8 +116,7 @@ where } }; - tokio::time::timeout(timeout, future) - .await + tokio::time::timeout(timeout, future).await } } @@ -326,7 +329,8 @@ where // Wait for the expected number of decisions to be received by each signer. for handle in event_loop_handles.iter_mut() { let msg = TxSignerEvent::ReceivedDepositDecision; - handle.wait_for_events(msg, num_expected_decisions, Duration::from_secs(10)) + handle + .wait_for_events(msg, num_expected_decisions, Duration::from_secs(10)) .await .expect("timed out waiting for events"); } diff --git a/signer/src/transaction_signer.rs b/signer/src/transaction_signer.rs index cfd3aa75..01f2926d 100644 --- a/signer/src/transaction_signer.rs +++ b/signer/src/transaction_signer.rs @@ -842,8 +842,13 @@ mod tests { use crate::testing::context::*; fn test_environment() -> testing::transaction_signer::TestEnvironment< - TestContext, WrappedMock, WrappedMock, - >> { + TestContext< + SharedStore, + WrappedMock, + WrappedMock, + WrappedMock, + >, + > { let test_model_parameters = testing::storage::model::Params { num_bitcoin_blocks: 20, num_stacks_blocks_per_bitcoin_block: 3, diff --git a/signer/tests/integration/transaction_signer.rs b/signer/tests/integration/transaction_signer.rs index 12937461..d61dc54a 100644 --- a/signer/tests/integration/transaction_signer.rs +++ b/signer/tests/integration/transaction_signer.rs @@ -15,7 +15,12 @@ async fn test_environment( mut signer_connections: Vec, signing_threshold: u32, ) -> TestEnvironment< - TestContext, WrappedMock, WrappedMock>, + TestContext< + PgStore, + WrappedMock, + WrappedMock, + WrappedMock, + >, > { let context_window = 3;