[refactor]: Custom genesis wait time for tps benchmark

Signed-off-by: Shanin Roman <[email protected]>

client/benches/tps/config.json

@@ -3,5 +3,6 @@
     "interval_us_per_tx": 0,
     "max_txs_per_block": 1024,
     "blocks": 15,
-    "sample_size": 10
+    "sample_size": 10,
+    "genesis_max_retries": 30
 }

client/benches/tps/utils.rs

@@ -31,6 +31,7 @@ pub struct Config {
     pub max_txs_per_block: u32,
     pub blocks: u32,
     pub sample_size: u32,
+    pub genesis_max_retries: u32,
 }
 
 impl fmt::Display for Config {
@@ -58,7 +59,7 @@ impl Config {
         // READY
         let (_rt, network, client) = Network::start_test_with_runtime(self.peers, None);
         let clients = network.clients();
-        wait_for_genesis_committed(&clients, 0);
+        wait_for_genesis_committed_with_max_retries(&clients, 0, self.genesis_max_retries);
 
         client.submit_all_blocking(
             ParametersBuilder::new()

core/test_network/src/lib.rs

@@ -327,10 +327,23 @@ impl Network {
 /// # Panics
 /// When unsuccessful after `MAX_RETRIES`.
 pub fn wait_for_genesis_committed(clients: &[Client], offline_peers: u32) {
-    const POLL_PERIOD: Duration = Duration::from_millis(1000);
     const MAX_RETRIES: u32 = 40;
+    wait_for_genesis_committed_with_max_retries(clients, offline_peers, MAX_RETRIES)
+}
+
+/// Wait for peers to have committed genesis block for specified amount of retries.
+/// Each retry once per second.
+///
+/// # Panics
+/// When unsuccessful after `max_retries`.
+pub fn wait_for_genesis_committed_with_max_retries(
+    clients: &[Client],
+    offline_peers: u32,
+    max_retries: u32,
+) {
+    const POLL_PERIOD: Duration = Duration::from_millis(1000);
 
-    for _ in 0..MAX_RETRIES {
+    for _ in 0..max_retries {
         let without_genesis_peers = clients.iter().fold(0_u32, |acc, client| {
             client.get_status().map_or(
                 acc + 1,
@@ -344,7 +357,7 @@ pub fn wait_for_genesis_committed(clients: &[Client], offline_peers: u32) {
     }
     panic!(
         "Failed to wait for online peers to commit genesis block. Total wait time: {:?}",
-        POLL_PERIOD * MAX_RETRIES
+        POLL_PERIOD * max_retries
     );
 }