85 lines
2.4 KiB
Rust
85 lines
2.4 KiB
Rust
//! Tests for redjubjub signature verification
|
|
|
|
use super::*;
|
|
|
|
use std::time::Duration;
|
|
|
|
use color_eyre::eyre::{eyre, Result};
|
|
use futures::stream::{FuturesUnordered, StreamExt};
|
|
use tower::ServiceExt;
|
|
use tower_batch::Batch;
|
|
|
|
async fn sign_and_verify<V>(mut verifier: V, n: usize) -> Result<(), V::Error>
|
|
where
|
|
V: Service<Item, Response = ()>,
|
|
{
|
|
let rng = thread_rng();
|
|
let mut results = FuturesUnordered::new();
|
|
for i in 0..n {
|
|
let span = tracing::trace_span!("sig", i);
|
|
let msg = b"BatchVerifyTest";
|
|
|
|
match i % 2 {
|
|
0 => {
|
|
let sk = SigningKey::<SpendAuth>::new(rng);
|
|
let vk = VerificationKey::from(&sk);
|
|
let sig = sk.sign(rng, &msg[..]);
|
|
verifier.ready_and().await?;
|
|
results.push(span.in_scope(|| verifier.call((vk.into(), sig, msg).into())))
|
|
}
|
|
1 => {
|
|
let sk = SigningKey::<Binding>::new(rng);
|
|
let vk = VerificationKey::from(&sk);
|
|
let sig = sk.sign(rng, &msg[..]);
|
|
verifier.ready_and().await?;
|
|
results.push(span.in_scope(|| verifier.call((vk.into(), sig, msg).into())))
|
|
}
|
|
_ => panic!(),
|
|
}
|
|
}
|
|
|
|
while let Some(result) = results.next().await {
|
|
result?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn batch_flushes_on_max_items_test() -> Result<()> {
|
|
batch_flushes_on_max_items().await
|
|
}
|
|
|
|
#[spandoc::spandoc]
|
|
async fn batch_flushes_on_max_items() -> Result<()> {
|
|
use tokio::time::timeout;
|
|
|
|
// Use a very long max_latency and a short timeout to check that
|
|
// flushing is happening based on hitting max_items.
|
|
let verifier = Batch::new(Verifier::default(), 10, Duration::from_secs(1000));
|
|
timeout(Duration::from_secs(5), sign_and_verify(verifier, 100))
|
|
.await?
|
|
.map_err(|e| eyre!(e))?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn batch_flushes_on_max_latency_test() -> Result<()> {
|
|
batch_flushes_on_max_latency().await
|
|
}
|
|
|
|
#[spandoc::spandoc]
|
|
async fn batch_flushes_on_max_latency() -> Result<()> {
|
|
use tokio::time::timeout;
|
|
|
|
// Use a very high max_items and a short timeout to check that
|
|
// flushing is happening based on hitting max_latency.
|
|
let verifier = Batch::new(Verifier::default(), 100, Duration::from_millis(500));
|
|
timeout(Duration::from_secs(5), sign_and_verify(verifier, 10))
|
|
.await?
|
|
.map_err(|e| eyre!(e))?;
|
|
|
|
Ok(())
|
|
}
|