diff --git a/tower-batch/tests/ed25519.rs b/tower-batch/tests/ed25519.rs index e28bbeff..bfef9307 100644 --- a/tower-batch/tests/ed25519.rs +++ b/tower-batch/tests/ed25519.rs @@ -106,10 +106,9 @@ fn install_tracing() { }) } -async fn sign_and_verify(mut verifier: V, n: usize) +async fn sign_and_verify(mut verifier: V, n: usize) -> Result<(), V::Error> where V: Service, - >::Error: Into>, { let mut results = FuturesUnordered::new(); for i in 0..n { @@ -119,15 +118,15 @@ where let msg = b"BatchVerifyTest"; let sig = sk.sign(&msg[..]); - verifier.ready_and().await.map_err(|e| e.into()).unwrap(); + verifier.ready_and().await?; results.push(span.in_scope(|| verifier.call((vk_bytes, sig, msg).into()))) } while let Some(result) = results.next().await { - let result = result.map_err(|e| e.into()); - tracing::trace!(?result); - assert!(result.is_ok()); + result?; } + + Ok(()) } #[tokio::test] @@ -137,12 +136,8 @@ async fn batch_flushes_on_max_items() -> color_eyre::Result<()> { // Use a very long max_latency and a short timeout to check that // flushing is happening based on hitting max_items. - let verifier = Batch::<_, _, color_eyre::Report>::new( - Ed25519Verifier::new(), - 10, - Duration::from_secs(1000), - ); - Ok(timeout(Duration::from_secs(1), sign_and_verify(verifier, 100)).await?) + let verifier = Batch::new(Ed25519Verifier::new(), 10, Duration::from_secs(1000)); + timeout(Duration::from_secs(1), sign_and_verify(verifier, 100)).await? } #[tokio::test] @@ -152,10 +147,6 @@ async fn batch_flushes_on_max_latency() -> color_eyre::Result<()> { // Use a very high max_items and a short timeout to check that // flushing is happening based on hitting max_latency. - let verifier = Batch::<_, _, color_eyre::Report>::new( - Ed25519Verifier::new(), - 100, - Duration::from_millis(500), - ); - Ok(timeout(Duration::from_secs(1), sign_and_verify(verifier, 10)).await?) + let verifier = Batch::new(Ed25519Verifier::new(), 100, Duration::from_millis(500)); + timeout(Duration::from_secs(1), sign_and_verify(verifier, 10)).await? }