mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-04-27 01:07:57 +00:00
enable disputes (#3478)
* initial integration and migration code * fix tests * fix counting test * assume the current version on missing file * use SelectRelayChain * remove duplicate metric * Update node/service/src/lib.rs Co-authored-by: Robert Habermeier <rphmeier@gmail.com> * remove ApprovalCheckingVotingRule * address my concern * never mode for StagnantCheckInterval * REVERTME: some logs * w00t * it's ugly but it works * Revert "REVERTME: some logs" This reverts commit e210505a2e83e31c381394924500b69277bb042e. * it's handle, not handler * fix a few typos Co-authored-by: Robert Habermeier <rphmeier@gmail.com>
This commit is contained in:
@@ -18,6 +18,7 @@ use std::sync::atomic;
|
||||
use std::collections::HashMap;
|
||||
use std::task::{Poll};
|
||||
use futures::{executor, pin_mut, select, FutureExt, pending, poll, stream};
|
||||
use futures::channel::mpsc;
|
||||
|
||||
use polkadot_primitives::v1::{CollatorPair, CandidateHash};
|
||||
use polkadot_node_primitives::{CollationResult, CollationGenerationConfig, PoV, BlockData};
|
||||
@@ -166,13 +167,14 @@ fn overseer_works() {
|
||||
.replace_candidate_validation(TestSubsystem1(s1_tx))
|
||||
.replace_candidate_backing(TestSubsystem2(s2_tx));
|
||||
|
||||
let (overseer, mut handler) = Overseer::new(
|
||||
let (overseer, handle) = Overseer::new(
|
||||
vec![],
|
||||
all_subsystems,
|
||||
None,
|
||||
MockSupportsParachains,
|
||||
spawner,
|
||||
).unwrap();
|
||||
let mut handle = Handle::Connected(handle);
|
||||
let overseer_fut = overseer.run().fuse();
|
||||
|
||||
pin_mut!(overseer_fut);
|
||||
@@ -188,7 +190,7 @@ fn overseer_works() {
|
||||
Some(msg) => {
|
||||
s1_results.push(msg);
|
||||
if s1_results.len() == 10 {
|
||||
handler.stop().await;
|
||||
handle.stop().await;
|
||||
}
|
||||
}
|
||||
None => break,
|
||||
@@ -236,21 +238,22 @@ fn overseer_metrics_work() {
|
||||
|
||||
let all_subsystems = AllSubsystems::<()>::dummy();
|
||||
let registry = prometheus::Registry::new();
|
||||
let (overseer, mut handler) = Overseer::new(
|
||||
let (overseer, handle) = Overseer::new(
|
||||
vec![first_block],
|
||||
all_subsystems,
|
||||
Some(®istry),
|
||||
MockSupportsParachains,
|
||||
spawner,
|
||||
).unwrap();
|
||||
let mut handle = Handle::Connected(handle);
|
||||
let overseer_fut = overseer.run().fuse();
|
||||
|
||||
pin_mut!(overseer_fut);
|
||||
|
||||
handler.block_imported(second_block).await;
|
||||
handler.block_imported(third_block).await;
|
||||
handler.send_msg_anon(AllMessages::CandidateValidation(test_candidate_validation_msg())).await;
|
||||
handler.stop().await;
|
||||
handle.block_imported(second_block).await;
|
||||
handle.block_imported(third_block).await;
|
||||
handle.send_msg_anon(AllMessages::CandidateValidation(test_candidate_validation_msg())).await;
|
||||
handle.stop().await;
|
||||
|
||||
select! {
|
||||
res = overseer_fut => {
|
||||
@@ -398,13 +401,14 @@ fn overseer_start_stop_works() {
|
||||
let all_subsystems = AllSubsystems::<()>::dummy()
|
||||
.replace_candidate_validation(TestSubsystem5(tx_5))
|
||||
.replace_candidate_backing(TestSubsystem6(tx_6));
|
||||
let (overseer, mut handler) = Overseer::new(
|
||||
let (overseer, handle) = Overseer::new(
|
||||
vec![first_block],
|
||||
all_subsystems,
|
||||
None,
|
||||
MockSupportsParachains,
|
||||
spawner,
|
||||
).unwrap();
|
||||
let mut handle = Handle::Connected(handle);
|
||||
|
||||
let overseer_fut = overseer.run().fuse();
|
||||
pin_mut!(overseer_fut);
|
||||
@@ -412,8 +416,8 @@ fn overseer_start_stop_works() {
|
||||
let mut ss5_results = Vec::new();
|
||||
let mut ss6_results = Vec::new();
|
||||
|
||||
handler.block_imported(second_block).await;
|
||||
handler.block_imported(third_block).await;
|
||||
handle.block_imported(second_block).await;
|
||||
handle.block_imported(third_block).await;
|
||||
|
||||
let expected_heartbeats = vec![
|
||||
OverseerSignal::ActiveLeaves(ActiveLeavesUpdate::start_work(ActivatedLeaf {
|
||||
@@ -463,7 +467,7 @@ fn overseer_start_stop_works() {
|
||||
|
||||
if ss5_results.len() == expected_heartbeats.len() &&
|
||||
ss6_results.len() == expected_heartbeats.len() {
|
||||
handler.stop().await;
|
||||
handle.stop().await;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -507,13 +511,14 @@ fn overseer_finalize_works() {
|
||||
.replace_candidate_backing(TestSubsystem6(tx_6));
|
||||
|
||||
// start with two forks of different height.
|
||||
let (overseer, mut handler) = Overseer::new(
|
||||
let (overseer, handle) = Overseer::new(
|
||||
vec![first_block, second_block],
|
||||
all_subsystems,
|
||||
None,
|
||||
MockSupportsParachains,
|
||||
spawner,
|
||||
).unwrap();
|
||||
let mut handle = Handle::Connected(handle);
|
||||
|
||||
let overseer_fut = overseer.run().fuse();
|
||||
pin_mut!(overseer_fut);
|
||||
@@ -522,7 +527,7 @@ fn overseer_finalize_works() {
|
||||
let mut ss6_results = Vec::new();
|
||||
|
||||
// this should stop work on both forks we started with earlier.
|
||||
handler.block_finalized(third_block).await;
|
||||
handle.block_finalized(third_block).await;
|
||||
|
||||
let expected_heartbeats = vec![
|
||||
OverseerSignal::ActiveLeaves(ActiveLeavesUpdate {
|
||||
@@ -569,7 +574,7 @@ fn overseer_finalize_works() {
|
||||
}
|
||||
|
||||
if ss5_results.len() == expected_heartbeats.len() && ss6_results.len() == expected_heartbeats.len() {
|
||||
handler.stop().await;
|
||||
handle.stop().await;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -607,21 +612,22 @@ fn do_not_send_empty_leaves_update_on_block_finalization() {
|
||||
let all_subsystems = AllSubsystems::<()>::dummy()
|
||||
.replace_candidate_backing(TestSubsystem6(tx_5));
|
||||
|
||||
let (overseer, mut handler) = Overseer::new(
|
||||
let (overseer, handle) = Overseer::new(
|
||||
Vec::new(),
|
||||
all_subsystems,
|
||||
None,
|
||||
MockSupportsParachains,
|
||||
spawner,
|
||||
).unwrap();
|
||||
let mut handle = Handle::Connected(handle);
|
||||
|
||||
let overseer_fut = overseer.run().fuse();
|
||||
pin_mut!(overseer_fut);
|
||||
|
||||
let mut ss5_results = Vec::new();
|
||||
|
||||
handler.block_finalized(finalized_block.clone()).await;
|
||||
handler.block_imported(imported_block.clone()).await;
|
||||
handle.block_finalized(finalized_block.clone()).await;
|
||||
handle.block_imported(imported_block.clone()).await;
|
||||
|
||||
let expected_heartbeats = vec![
|
||||
OverseerSignal::ActiveLeaves(ActiveLeavesUpdate {
|
||||
@@ -652,7 +658,7 @@ fn do_not_send_empty_leaves_update_on_block_finalization() {
|
||||
}
|
||||
|
||||
if ss5_results.len() == expected_heartbeats.len() {
|
||||
handler.stop().await;
|
||||
handle.stop().await;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -809,10 +815,35 @@ fn test_approval_voting_msg() -> ApprovalVotingMessage {
|
||||
ApprovalVotingMessage::ApprovedAncestor(Default::default(), 0, sender)
|
||||
}
|
||||
|
||||
fn test_dispute_coordinator_msg() -> DisputeCoordinatorMessage {
|
||||
let (sender, _) = oneshot::channel();
|
||||
DisputeCoordinatorMessage::RecentDisputes(sender)
|
||||
}
|
||||
|
||||
fn test_dispute_participation_msg() -> DisputeParticipationMessage {
|
||||
let (sender, _) = oneshot::channel();
|
||||
DisputeParticipationMessage::Participate {
|
||||
candidate_hash: Default::default(),
|
||||
candidate_receipt: Default::default(),
|
||||
session: 0,
|
||||
n_validators: 0,
|
||||
report_availability: sender,
|
||||
}
|
||||
}
|
||||
|
||||
fn test_dispute_distribution_msg() -> DisputeDistributionMessage {
|
||||
let (_, receiver) = mpsc::channel(1);
|
||||
DisputeDistributionMessage::DisputeSendingReceiver(receiver)
|
||||
}
|
||||
|
||||
fn test_chain_selection_msg() -> ChainSelectionMessage {
|
||||
ChainSelectionMessage::Approved(Default::default())
|
||||
}
|
||||
|
||||
// Checks that `stop`, `broadcast_signal` and `broadcast_message` are implemented correctly.
|
||||
#[test]
|
||||
fn overseer_all_subsystems_receive_signals_and_messages() {
|
||||
const NUM_SUBSYSTEMS: usize = 17;
|
||||
const NUM_SUBSYSTEMS: usize = 21;
|
||||
// -3 for BitfieldSigning, GossipSupport and AvailabilityDistribution
|
||||
const NUM_SUBSYSTEMS_MESSAGED: usize = NUM_SUBSYSTEMS - 3;
|
||||
|
||||
@@ -846,20 +877,25 @@ fn overseer_all_subsystems_receive_signals_and_messages() {
|
||||
approval_distribution: subsystem.clone(),
|
||||
approval_voting: subsystem.clone(),
|
||||
gossip_support: subsystem.clone(),
|
||||
dispute_coordinator: subsystem.clone(),
|
||||
dispute_participation: subsystem.clone(),
|
||||
dispute_distribution: subsystem.clone(),
|
||||
chain_selection: subsystem.clone(),
|
||||
};
|
||||
let (overseer, mut handler) = Overseer::new(
|
||||
let (overseer, handle) = Overseer::new(
|
||||
vec![],
|
||||
all_subsystems,
|
||||
None,
|
||||
MockSupportsParachains,
|
||||
spawner,
|
||||
).unwrap();
|
||||
let mut handle = Handle::Connected(handle);
|
||||
let overseer_fut = overseer.run().fuse();
|
||||
|
||||
pin_mut!(overseer_fut);
|
||||
|
||||
// send a signal to each subsystem
|
||||
handler.block_imported(BlockInfo {
|
||||
handle.block_imported(BlockInfo {
|
||||
hash: Default::default(),
|
||||
parent_hash: Default::default(),
|
||||
number: Default::default(),
|
||||
@@ -867,22 +903,26 @@ fn overseer_all_subsystems_receive_signals_and_messages() {
|
||||
|
||||
// send a msg to each subsystem
|
||||
// except for BitfieldSigning and GossipSupport as the messages are not instantiable
|
||||
handler.send_msg_anon(AllMessages::CandidateValidation(test_candidate_validation_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::CandidateBacking(test_candidate_backing_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::CollationGeneration(test_collator_generation_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::CollatorProtocol(test_collator_protocol_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::StatementDistribution(test_statement_distribution_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::AvailabilityRecovery(test_availability_recovery_msg())).await;
|
||||
// handler.send_msg_anon(AllMessages::BitfieldSigning(test_bitfield_signing_msg())).await;
|
||||
// handler.send_msg_anon(AllMessages::GossipSupport(test_bitfield_signing_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::BitfieldDistribution(test_bitfield_distribution_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::Provisioner(test_provisioner_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::RuntimeApi(test_runtime_api_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::AvailabilityStore(test_availability_store_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::NetworkBridge(test_network_bridge_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::ChainApi(test_chain_api_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::ApprovalDistribution(test_approval_distribution_msg())).await;
|
||||
handler.send_msg_anon(AllMessages::ApprovalVoting(test_approval_voting_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::CandidateValidation(test_candidate_validation_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::CandidateBacking(test_candidate_backing_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::CollationGeneration(test_collator_generation_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::CollatorProtocol(test_collator_protocol_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::StatementDistribution(test_statement_distribution_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::AvailabilityRecovery(test_availability_recovery_msg())).await;
|
||||
// handle.send_msg_anon(AllMessages::BitfieldSigning(test_bitfield_signing_msg())).await;
|
||||
// handle.send_msg_anon(AllMessages::GossipSupport(test_bitfield_signing_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::BitfieldDistribution(test_bitfield_distribution_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::Provisioner(test_provisioner_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::RuntimeApi(test_runtime_api_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::AvailabilityStore(test_availability_store_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::NetworkBridge(test_network_bridge_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::ChainApi(test_chain_api_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::ApprovalDistribution(test_approval_distribution_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::ApprovalVoting(test_approval_voting_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::DisputeCoordinator(test_dispute_coordinator_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::DisputeParticipation(test_dispute_participation_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::DisputeDistribution(test_dispute_distribution_msg())).await;
|
||||
handle.send_msg_anon(AllMessages::ChainSelection(test_chain_selection_msg())).await;
|
||||
|
||||
// Wait until all subsystems have received. Otherwise the messages might race against
|
||||
// the conclude signal.
|
||||
@@ -903,7 +943,7 @@ fn overseer_all_subsystems_receive_signals_and_messages() {
|
||||
}
|
||||
|
||||
// send a stop signal to each subsystems
|
||||
handler.stop().await;
|
||||
handle.stop().await;
|
||||
|
||||
let res = overseer_fut.await;
|
||||
assert_eq!(stop_signals_received.load(atomic::Ordering::SeqCst), NUM_SUBSYSTEMS);
|
||||
@@ -933,6 +973,10 @@ fn context_holds_onto_message_until_enough_signals_received() {
|
||||
let (approval_distribution_bounded_tx, _) = metered::channel(CHANNEL_CAPACITY);
|
||||
let (approval_voting_bounded_tx, _) = metered::channel(CHANNEL_CAPACITY);
|
||||
let (gossip_support_bounded_tx, _) = metered::channel(CHANNEL_CAPACITY);
|
||||
let (dispute_coordinator_bounded_tx, _) = metered::channel(CHANNEL_CAPACITY);
|
||||
let (dispute_participation_bounded_tx, _) = metered::channel(CHANNEL_CAPACITY);
|
||||
let (dispute_distribution_bounded_tx, _) = metered::channel(CHANNEL_CAPACITY);
|
||||
let (chain_selection_bounded_tx, _) = metered::channel(CHANNEL_CAPACITY);
|
||||
|
||||
let (candidate_validation_unbounded_tx, _) = metered::unbounded();
|
||||
let (candidate_backing_unbounded_tx, _) = metered::unbounded();
|
||||
@@ -951,6 +995,10 @@ fn context_holds_onto_message_until_enough_signals_received() {
|
||||
let (approval_distribution_unbounded_tx, _) = metered::unbounded();
|
||||
let (approval_voting_unbounded_tx, _) = metered::unbounded();
|
||||
let (gossip_support_unbounded_tx, _) = metered::unbounded();
|
||||
let (dispute_coordinator_unbounded_tx, _) = metered::unbounded();
|
||||
let (dispute_participation_unbounded_tx, _) = metered::unbounded();
|
||||
let (dispute_distribution_unbounded_tx, _) = metered::unbounded();
|
||||
let (chain_selection_unbounded_tx, _) = metered::unbounded();
|
||||
|
||||
let channels_out = ChannelsOut {
|
||||
candidate_validation: candidate_validation_bounded_tx.clone(),
|
||||
@@ -970,6 +1018,10 @@ fn context_holds_onto_message_until_enough_signals_received() {
|
||||
approval_distribution: approval_distribution_bounded_tx.clone(),
|
||||
approval_voting: approval_voting_bounded_tx.clone(),
|
||||
gossip_support: gossip_support_bounded_tx.clone(),
|
||||
dispute_coordinator: dispute_coordinator_bounded_tx.clone(),
|
||||
dispute_participation: dispute_participation_bounded_tx.clone(),
|
||||
dispute_distribution: dispute_distribution_bounded_tx.clone(),
|
||||
chain_selection: chain_selection_bounded_tx.clone(),
|
||||
|
||||
candidate_validation_unbounded: candidate_validation_unbounded_tx.clone(),
|
||||
candidate_backing_unbounded: candidate_backing_unbounded_tx.clone(),
|
||||
@@ -988,6 +1040,10 @@ fn context_holds_onto_message_until_enough_signals_received() {
|
||||
approval_distribution_unbounded: approval_distribution_unbounded_tx.clone(),
|
||||
approval_voting_unbounded: approval_voting_unbounded_tx.clone(),
|
||||
gossip_support_unbounded: gossip_support_unbounded_tx.clone(),
|
||||
dispute_coordinator_unbounded: dispute_coordinator_unbounded_tx.clone(),
|
||||
dispute_participation_unbounded: dispute_participation_unbounded_tx.clone(),
|
||||
dispute_distribution_unbounded: dispute_distribution_unbounded_tx.clone(),
|
||||
chain_selection_unbounded: chain_selection_unbounded_tx.clone(),
|
||||
};
|
||||
|
||||
let (mut signal_tx, signal_rx) = metered::channel(CHANNEL_CAPACITY);
|
||||
|
||||
Reference in New Issue
Block a user