Fix flaky test and error reporting (#7282)

* Fix flaky test and error reporting

* Address review comments and adjust recv timeouts
This commit is contained in:
Vsevolod Stakhov
2023-05-24 16:50:00 +01:00
committed by GitHub
parent adad06ef16
commit 75b9456e69
2 changed files with 12 additions and 7 deletions
@@ -172,7 +172,9 @@ async fn overseer_signal(
.send(FromOrchestra::Signal(signal))
.timeout(TIMEOUT)
.await
.expect("10ms is more than enough for sending signals.");
.unwrap_or_else(|| {
panic!("{}ms is more than enough for sending signals.", TIMEOUT.as_millis())
});
}
async fn overseer_send(
@@ -184,7 +186,9 @@ async fn overseer_send(
.send(FromOrchestra::Communication { msg })
.timeout(TIMEOUT)
.await
.expect("10ms is more than enough for sending messages.");
.unwrap_or_else(|| {
panic!("{}ms is more than enough for sending messages.", TIMEOUT.as_millis())
});
}
async fn overseer_recv(
@@ -41,11 +41,12 @@ use sp_keystore::{testing::MemoryKeystore, Keystore, KeystorePtr};
use std::{iter::FromIterator as _, sync::Arc, time::Duration};
const TIMEOUT: Duration = Duration::from_millis(50);
macro_rules! launch {
($fut:expr) => {
$fut.timeout(Duration::from_millis(10))
.await
.expect("10ms is more than enough for sending messages.")
$fut.timeout(TIMEOUT).await.unwrap_or_else(|| {
panic!("{}ms is more than enough for sending messages.", TIMEOUT.as_millis())
});
};
}
@@ -220,7 +221,7 @@ fn receive_invalid_signature() {
));
// reputation doesn't change due to one_job_per_validator check
assert!(handle.recv().timeout(Duration::from_millis(10)).await.is_none());
assert!(handle.recv().timeout(TIMEOUT).await.is_none());
launch!(handle_network_msg(
&mut ctx,
@@ -523,7 +524,7 @@ fn do_not_relay_message_twice() {
);
// There shouldn't be any other message
assert!(handle.recv().timeout(Duration::from_millis(10)).await.is_none());
assert!(handle.recv().timeout(TIMEOUT).await.is_none());
});
}