client/beefy: fix incorrect BEEFY justifications import test (#12593)

Signed-off-by: Adrian Catangiu <adrian@parity.io>
This commit is contained in:
Adrian Catangiu
2022-10-31 17:41:01 +02:00
committed by GitHub
parent a1a3b923ab
commit c68bd397d8
+27 -8
View File
@@ -755,10 +755,16 @@ fn beefy_importing_blocks() {
block_on(block_import.import_block(params(block, None), HashMap::new())).unwrap(),
ImportResult::AlreadyInChain
);
// Verify no justifications present:
// Verify no BEEFY justifications present:
{
// none in backend,
assert!(full_client.justifications(&block_id).unwrap().is_none());
assert_eq!(
full_client
.justifications(&block_id)
.unwrap()
.and_then(|j| j.get(BEEFY_ENGINE_ID).cloned()),
None
);
// and none sent to BEEFY worker.
block_on(poll_fn(move |cx| {
assert_eq!(justif_recv.poll_next_unpin(cx), Poll::Pending);
@@ -787,11 +793,18 @@ fn beefy_importing_blocks() {
..Default::default()
}),
);
// Verify justification successfully imported:
// Verify BEEFY justification successfully imported:
{
// available in backend,
assert!(full_client.justifications(&BlockId::Number(block_num)).unwrap().is_some());
// and also sent to BEEFY worker.
// still not in backend (worker is responsible for appending to backend),
assert_eq!(
full_client
.justifications(&block_id)
.unwrap()
.and_then(|j| j.get(BEEFY_ENGINE_ID).cloned()),
None
);
// but sent to BEEFY worker
// (worker will append it to backend when all previous mandatory justifs are there as well).
block_on(poll_fn(move |cx| {
match justif_recv.poll_next_unpin(cx) {
Poll::Ready(Some(_justification)) => (),
@@ -823,10 +836,16 @@ fn beefy_importing_blocks() {
..Default::default()
}),
);
// Verify bad justifications was not imported:
// Verify bad BEEFY justifications was not imported:
{
// none in backend,
assert!(full_client.justifications(&block_id).unwrap().is_none());
assert_eq!(
full_client
.justifications(&BlockId::Number(block_num))
.unwrap()
.and_then(|j| j.get(BEEFY_ENGINE_ID).cloned()),
None
);
// and none sent to BEEFY worker.
block_on(poll_fn(move |cx| {
assert_eq!(justif_recv.poll_next_unpin(cx), Poll::Pending);