From 038e3e2ee103aafbf6736ae1583705520bf368aa Mon Sep 17 00:00:00 2001
From: Nazar Hussain <nazarhussain@gmail.com>
Date: Mon, 5 Feb 2024 16:46:51 +0100
Subject: [PATCH] Compute effective genesis delay

---
 .../cli/test/sim/backup_eth_provider.test.ts  |  1 +
 packages/cli/test/sim/deneb.test.ts           |  1 +
 packages/cli/test/sim/endpoints.test.ts       |  1 +
 packages/cli/test/sim/mixed_client.test.ts    |  1 +
 packages/cli/test/sim/multi_fork.test.ts      |  1 +
 .../cli/test/utils/simulation/utils/index.ts  | 22 +++++++++++++++++--
 6 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/packages/cli/test/sim/backup_eth_provider.test.ts b/packages/cli/test/sim/backup_eth_provider.test.ts
index 40ad7d4661de..5a38a27dc8f0 100644
--- a/packages/cli/test/sim/backup_eth_provider.test.ts
+++ b/packages/cli/test/sim/backup_eth_provider.test.ts
@@ -16,6 +16,7 @@ const {estimatedTimeoutMs, forkConfig} = defineSimTestConfig({
   ALTAIR_FORK_EPOCH: altairForkEpoch,
   BELLATRIX_FORK_EPOCH: bellatrixForkEpoch,
   runTillEpoch: runTillEpoch + syncWaitEpoch,
+  initialNodes: 3,
 });
 
 const env = await SimulationEnvironment.initWithDefaults(
diff --git a/packages/cli/test/sim/deneb.test.ts b/packages/cli/test/sim/deneb.test.ts
index 94d165bb937c..b66252b06892 100644
--- a/packages/cli/test/sim/deneb.test.ts
+++ b/packages/cli/test/sim/deneb.test.ts
@@ -18,6 +18,7 @@ const {estimatedTimeoutMs, forkConfig} = defineSimTestConfig({
   ALTAIR_FORK_EPOCH: altairForkEpoch,
   BELLATRIX_FORK_EPOCH: bellatrixForkEpoch,
   runTillEpoch: runTillEpoch + syncWaitEpoch,
+  initialNodes: 2,
 });
 
 const env = await SimulationEnvironment.initWithDefaults(
diff --git a/packages/cli/test/sim/endpoints.test.ts b/packages/cli/test/sim/endpoints.test.ts
index ea55b70e26c1..a31b561e0578 100644
--- a/packages/cli/test/sim/endpoints.test.ts
+++ b/packages/cli/test/sim/endpoints.test.ts
@@ -16,6 +16,7 @@ const {estimatedTimeoutMs, forkConfig} = defineSimTestConfig({
   ALTAIR_FORK_EPOCH: altairForkEpoch,
   BELLATRIX_FORK_EPOCH: bellatrixForkEpoch,
   runTillEpoch: 2,
+  initialNodes: 1,
 });
 
 const env = await SimulationEnvironment.initWithDefaults(
diff --git a/packages/cli/test/sim/mixed_client.test.ts b/packages/cli/test/sim/mixed_client.test.ts
index ad2095f8c378..80c20471ede5 100644
--- a/packages/cli/test/sim/mixed_client.test.ts
+++ b/packages/cli/test/sim/mixed_client.test.ts
@@ -17,6 +17,7 @@ const {estimatedTimeoutMs, forkConfig} = defineSimTestConfig({
   BELLATRIX_FORK_EPOCH: bellatrixForkEpoch,
   CAPELLA_FORK_EPOCH: capellaForkEpoch,
   runTillEpoch: runTillEpoch + syncWaitEpoch,
+  initialNodes: 2,
 });
 
 const env = await SimulationEnvironment.initWithDefaults(
diff --git a/packages/cli/test/sim/multi_fork.test.ts b/packages/cli/test/sim/multi_fork.test.ts
index 24f42498c638..70f119609454 100644
--- a/packages/cli/test/sim/multi_fork.test.ts
+++ b/packages/cli/test/sim/multi_fork.test.ts
@@ -27,6 +27,7 @@ const {estimatedTimeoutMs, forkConfig} = defineSimTestConfig({
   BELLATRIX_FORK_EPOCH: bellatrixForkEpoch,
   CAPELLA_FORK_EPOCH: capellaForkEpoch,
   runTillEpoch: runTillEpoch + syncWaitEpoch,
+  initialNodes: 5,
 });
 
 const env = await SimulationEnvironment.initWithDefaults(
diff --git a/packages/cli/test/utils/simulation/utils/index.ts b/packages/cli/test/utils/simulation/utils/index.ts
index dfa66bed6580..94a97b6e33ff 100644
--- a/packages/cli/test/utils/simulation/utils/index.ts
+++ b/packages/cli/test/utils/simulation/utils/index.ts
@@ -17,18 +17,36 @@ export const avg = (arr: number[]): number => {
   return arr.length === 0 ? 0 : arr.reduce((p, c) => p + c, 0) / arr.length;
 };
 
+function getGenesisDelaySlots(initialNodes?: number): number {
+  if (process.env.GENESIS_DELAY_SLOTS) {
+    const genesisDelaySlots = parseInt(process.env.GENESIS_DELAY_SLOTS);
+    // If custom job is invoked and want to use default genesis delay then provider -1 as value
+    if (genesisDelaySlots >= 0) return genesisDelaySlots;
+  }
+
+  if (initialNodes == null) return 40;
+  // Considering each node consists of EN, BN, VC and KM
+  // EN - Execution Node - 5s
+  // BN - Beacon Node - 10s
+  // VC - Validator Client - 5s
+  // KM - Key Manager - 3s
+  // Initial script launch time - 10s
+  return Math.ceil(((5 + 10 + 5 + 3) * initialNodes + 10) / SIM_TESTS_SECONDS_PER_SLOT);
+}
+
 export function defineSimTestConfig(
   opts: Partial<ChainConfig> & {
     cliqueSealingPeriod?: number;
     additionalSlotsForTTD?: number;
     runTillEpoch: number;
+    // Used to calculate genesis delay
+    initialNodes?: number;
   }
 ): {
   estimatedTimeoutMs: number;
   forkConfig: ChainForkConfig;
 } {
-  const genesisDelaySeconds =
-    (process.env.GENESIS_DELAY_SLOTS ? parseInt(process.env.GENESIS_DELAY_SLOTS) : 40) * SIM_TESTS_SECONDS_PER_SLOT;
+  const genesisDelaySeconds = getGenesisDelaySlots(opts.initialNodes) * SIM_TESTS_SECONDS_PER_SLOT;
 
   const estimatedTimeoutMs =
     getEstimatedTimeInSecForRun({