Write all blocks in transfer_snapshot_between_two_replicas at once

To make the test go faster. Especially now with throttling RootNode
updates the test was running with the speed of one block per throttle
period (1s), which caused the test to time out.

lib/src/network/tests.rs

@@ -109,7 +109,7 @@ async fn transfer_snapshot_between_two_replicas_case(
 // TODO: Make it faster and increase the cases.
 #[proptest(cases = 8)]
 fn transfer_blocks_between_two_replicas(
-    #[strategy(1usize..32)] block_count: usize,
+    #[strategy(1usize..62)] block_count: usize,
     #[strategy(test_utils::rng_seed_strategy())] rng_seed: u64,
 ) {
     test_utils::run(transfer_blocks_between_two_replicas_case(
@@ -145,8 +145,10 @@ async fn transfer_blocks_between_two_replicas_case(block_count: usize, rng_seed:
 
             a_vault.receive_block(block, Some(promise)).await.unwrap();
             tracing::info!(?id, "write block");
+        }
 
-            // Then wait until replica B receives and writes it too.
+        // Then wait until replica B receives and writes it too.
+        for (id, _block) in snapshot.blocks() {
             wait_until_block_exists(&b_vault, id).await;
         }
     };