Correct flakey CI tests (#6646)

* Correct flakey CI tests

* Correct clippy

* Extend timeout for events
This commit is contained in:
Age Manning
2024-12-10 15:24:55 +11:00
committed by GitHub
parent e9ec67e78a
commit 3b8254a8ec
2 changed files with 6 additions and 3 deletions

View File

@@ -213,6 +213,7 @@ impl<T: BeaconChainTypes> SubnetService<T> {
#[cfg(test)]
pub(crate) fn is_subscribed(&self, subnet: &Subnet) -> bool {
self.subscriptions.contains_key(subnet)
|| self.permanent_attestation_subscriptions.contains(subnet)
}
/// Processes a list of validator subscriptions.

View File

@@ -225,7 +225,7 @@ mod test {
let mut committee_count = 1;
let mut subnet = Subnet::Attestation(
SubnetId::compute_subnet::<MainnetEthSpec>(
current_slot,
subscription_slot,
committee_index,
committee_count,
&subnet_service.beacon_chain.spec,
@@ -250,7 +250,7 @@ mod test {
let subscriptions = vec![get_subscription(
committee_index,
current_slot,
subscription_slot,
committee_count,
true,
)];
@@ -556,7 +556,8 @@ mod test {
subnet_service.validator_subscriptions(vec![sub1, sub2].into_iter());
// Unsubscription event should happen at the end of the slot.
let events = get_events(&mut subnet_service, None, 1).await;
// We wait for 2 slots, to avoid timeout issues
let events = get_events(&mut subnet_service, None, 2).await;
let expected_subscription =
SubnetServiceMessage::Subscribe(Subnet::Attestation(subnet_id1));
@@ -567,6 +568,7 @@ mod test {
assert_eq!(expected_subscription, events[0]);
assert_eq!(expected_unsubscription, events[2]);
}
// Check that there are no more subscriptions
assert_eq!(subnet_service.subscriptions().count(), 0);
println!("{events:?}");