Browse Source

Fixed bug that still blocked spares

ltulloch 2 years ago
parent
commit
4525d50b54
1 changed files with 3 additions and 2 deletions
  1. 3 2
      src/tests.rs

+ 3 - 2
src/tests.rs

@@ -1396,12 +1396,13 @@ fn stats_test_blockage_migration_100() {
 /// bucket list.
 
 fn block_bridges(th: &mut TestHarness, percentage: usize, credentials: Vec<cred::Lox>) {
-    let to_block: usize = ((th.ba.bridge_table.buckets.len() - th.ba.bridge_table.spares.len()) * percentage / 100).into();
+    let blockable_bridges = th.ba.bridge_table.buckets.len() - th.ba.bridge_table.spares.len();
+    let to_block: usize = (blockable_bridges * percentage / 100).into();
     let mut block_index: HashSet<usize> = HashSet::new();
     let mut rng = rand::thread_rng();
 
     while block_index.len() <= to_block - 1 {
-        let rand_num = rng.gen_range(0, th.ba.bridge_table.buckets.len());
+        let rand_num = rng.gen_range(0, blockable_bridges);
         block_index.insert(rand_num);
     }