Skip to content
This repository was archived by the owner on Nov 15, 2023. It is now read-only.
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Address some more review comments
  • Loading branch information
vstakhov committed Mar 23, 2022
commit a942d31677a97182ff6cae76c86d46bf6366f737
30 changes: 24 additions & 6 deletions node/network/statement-distribution/src/tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2020,15 +2020,21 @@ fn handle_multiple_seconded_statements() {
handle.recv().await,
AllMessages::NetworkBridge(
NetworkBridgeMessage::ReportPeer(p, r)
) if p == peer_a && r == BENEFIT_VALID_STATEMENT_FIRST => {}
) => {
assert_eq!(p, peer_a);
assert_eq!(r, BENEFIT_VALID_STATEMENT_FIRST);
}
);

// After the first valid statement, we expect messages to be circulated
assert_matches!(
handle.recv().await,
AllMessages::CandidateBacking(
CandidateBackingMessage::Statement(r, s)
) if r == relay_parent_hash && s == statement => {}
) => {
assert_eq!(r, relay_parent_hash);
assert_eq!(s, statement);
}
);

assert_matches!(
Expand Down Expand Up @@ -2066,7 +2072,10 @@ fn handle_multiple_seconded_statements() {
handle.recv().await,
AllMessages::NetworkBridge(
NetworkBridgeMessage::ReportPeer(p, r)
) if p == peer_b && r == BENEFIT_VALID_STATEMENT => {}
) => {
assert_eq!(p, peer_b);
assert_eq!(r, BENEFIT_VALID_STATEMENT);
}
);

// Create a `Valid` statement
Expand Down Expand Up @@ -2114,14 +2123,20 @@ fn handle_multiple_seconded_statements() {
handle.recv().await,
AllMessages::NetworkBridge(
NetworkBridgeMessage::ReportPeer(p, r)
) if p == peer_a && r == BENEFIT_VALID_STATEMENT_FIRST => {}
) => {
assert_eq!(p, peer_a);
assert_eq!(r, BENEFIT_VALID_STATEMENT_FIRST);
}
);

assert_matches!(
handle.recv().await,
AllMessages::CandidateBacking(
CandidateBackingMessage::Statement(r, s)
) if r == relay_parent_hash && s == statement => {}
) => {
assert_eq!(r, relay_parent_hash);
assert_eq!(s, statement);
}
);

assert_matches!(
Expand Down Expand Up @@ -2161,7 +2176,10 @@ fn handle_multiple_seconded_statements() {
handle.recv().await,
AllMessages::NetworkBridge(
NetworkBridgeMessage::ReportPeer(p, r)
) if p == peer_b && r == BENEFIT_VALID_STATEMENT => {}
) => {
assert_eq!(p, peer_b);
assert_eq!(r, BENEFIT_VALID_STATEMENT);
}
);

handle.send(FromOverseer::Signal(OverseerSignal::Conclude)).await;
Expand Down