fix random epochs test

near · Oct 4, 2024 · 4705111 · 4705111
1 parent 5fe7028
commit 4705111
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/chain/epoch-manager/src/tests/random_epochs.rs b/chain/epoch-manager/src/tests/random_epochs.rs
@@ -342,7 +342,10 @@ fn verify_block_stats(
                 aggregator.block_tracker.values().map(|value| value.expected).sum::<u64>();
             assert_eq!(sum_produced, blocks_in_epoch);
             assert_eq!(sum_expected, blocks_in_epoch_expected);
-            for shard_id in shard_layout.shard_ids() {
+            // TODO: The following sophisticated check doesn't do anything. The
+            // shard tracker is empty because the chunk mask in all block infos
+            // is empty.
+            for &shard_id in aggregator.shard_tracker.keys() {
                 let sum_produced = aggregator
                     .shard_tracker
                     .get(&shard_id)