gnosis/lighthouse
1
0
Fork 0
mirror of https://github.com/sigp/lighthouse.git synced 2026-03-22 14:24:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix incorrect prune test logic (#7999)

Browse Source 
I just noticed that one of the tests i added in #7915 is incorrect, after it was running flaky for a bit.
This PR fixes the scenario and ensure the outcome will always be the same.
This commit is contained in:
Jimmy Chen
2025-09-05 05:53:38 +10:00
committed by GitHub
parent 84ec209eba
commit 677de70025
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
beacon_node/lighthouse_network/src/peer_manager/mod.rs
View File

@@ -2885,11 +2885,10 @@ mod tests {
*peer_manager.network_globals.sampling_subnets.write() = HashSet::new();
// Create 12 peers:
// - 4 on custody subnet 0
// - 3 on subnet 1
//- 2 on subnet 2
// - 3 scattered.
// Every 4th peer (0,4,8) is on sync committee 0.
// * 4 on custody subnet 0, all on sync committee 0 subnet as well (should only prune up to 2 peers)
// * 3 on subnet 1
// * 2 on subnet 2
// * 3 scattered.
let mut peers = Vec::new();
for i in 0..12 {
let peer = PeerId::random();
@@ -2901,7 +2900,7 @@ mod tests {
7..9 => 2,
_ => i - 6,
};
let on_sync_committee = i % 4 == 0;
let on_sync_committee = i < 4;
{
let mut peers_db = peer_manager.network_globals.peers.write();
@@ -2949,10 +2948,10 @@ mod tests {
.cloned()
.collect();
let sync_committee_peers = [&peers[0], &peers[4], &peers[8]];
// only 2 peers should be pruned from the 4 peers in subnet 0.
let remaining_sync_peers = connected_peers
.iter()
.filter(|peer| sync_committee_peers.contains(peer))
.filter(|peer| peers[0..4].contains(peer))
.count();
assert_eq!(
remaining_sync_peers, 2,