Fix flaky test

paradigmxyz · Apr 19, 2024 · 0f301a4 · 0f301a4
1 parent 74e7a24
commit 0f301a4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/net/discv5/src/lib.rs b/crates/net/discv5/src/lib.rs
@@ -806,6 +806,9 @@ mod tests {
         // bucket index ceiled to the next multiple of 4
         const fn expected_bucket_index(kbucket_index: usize) -> u64 {
             let log2distance = kbucket_index + 1;
+            if log2distance % 8 == 0 {
+                return log2distance as u64;
+            }
             let log2distance = log2distance / 8;
             ((log2distance + 1) * 8) as u64
         }