Patch "enable rounds", as well.

Author: andreibancioiu

multiversx_sdk_cli/localnet/constants.py

@@ -9,3 +9,4 @@
 
 # See https://github.com/multiversx/mx-chain-go/blob/master/cmd/node/config/config.toml.
 ROUNDS_PER_EPOCH_TO_MIN_ROUNDS_BETWEEN_EPOCHS_RATIO = 4
+NUM_ROUNDS_BETWEEN_SUPERNOVA_ACTIVATION_EPOCH_AND_ACTIVATION_ROUND = 20

multiversx_sdk_cli/localnet/node_config_toml.py

@@ -1,13 +1,18 @@
 from typing import Any, Dict, Optional
 
 from multiversx_sdk_cli.localnet.config_root import ConfigRoot
-from multiversx_sdk_cli.localnet.constants import ROUNDS_PER_EPOCH_TO_MIN_ROUNDS_BETWEEN_EPOCHS_RATIO
+from multiversx_sdk_cli.localnet.constants import (
+    NUM_ROUNDS_BETWEEN_SUPERNOVA_ACTIVATION_EPOCH_AND_ACTIVATION_ROUND,
+    ROUNDS_PER_EPOCH_TO_MIN_ROUNDS_BETWEEN_EPOCHS_RATIO,
+)
 from multiversx_sdk_cli.localnet.nodes_setup_json import CHAIN_ID
 
 ConfigDict = Dict[str, Any]
 
 
-def patch_config(data: ConfigDict, config: ConfigRoot, supernova_activation_epoch: Optional[int] = None):
+def patch_config(data: ConfigDict, config: ConfigRoot, enable_epochs_config: ConfigDict):
+    supernova_activation_epoch = enable_epochs_config["EnableEpochs"].get("SupernovaEnableEpoch", None)
+
     data["GeneralSettings"]["ChainID"] = CHAIN_ID
 
     # "--operation-mode=historical-balances" is not available for nodes,
@@ -94,3 +99,19 @@ def patch_enable_epochs(data: ConfigDict, config: ConfigRoot):
     last_entry["MaxNumNodes"] = (
         penultimate_entry["MaxNumNodes"] - (config.shards.num_shards + 1) * penultimate_entry["NodesToShufflePerShard"]
     )
+
+
+def patch_enable_rounds(data: ConfigDict, config: ConfigRoot, enable_epochs_config: ConfigDict):
+    supernova_activation_epoch = enable_epochs_config["EnableEpochs"].get("SupernovaEnableEpoch", None)
+
+    activations = data["RoundActivations"]
+    supernova_entry = activations.get("SupernovaEnableRound")
+
+    if supernova_entry:
+        # Epochs are zero-indexed.
+        supernova_computed_activation_round = (
+            config.general.rounds_per_epoch * supernova_activation_epoch
+            + NUM_ROUNDS_BETWEEN_SUPERNOVA_ACTIVATION_EPOCH_AND_ACTIVATION_ROUND
+        )
+
+        supernova_entry["Round"] = str(supernova_computed_activation_round)

multiversx_sdk_cli/localnet/step_config.py

@@ -103,23 +103,25 @@ def patch_node_config(config: ConfigRoot):
         node_config_file = node_config / "config.toml"
         api_config_file = node_config / "api.toml"
         enable_epochs_config_file = node_config / "enableEpochs.toml"
+        enable_rounds_config_file = node_config / "enableRounds.toml"
         genesis_smart_contracts_file = node_config / "genesisSmartContracts.json"
 
         node_config_data = utils.read_toml_file(node_config_file)
         api_config_data = utils.read_toml_file(api_config_file)
         enable_epochs_config_data = utils.read_toml_file(enable_epochs_config_file)
+        enable_rounds_config_data = utils.read_toml_file(enable_rounds_config_file)
         genesis_smart_contracts_data = utils.read_json_file(genesis_smart_contracts_file)
 
-        supernova_activation_epoch = enable_epochs_config_data["EnableEpochs"].get("SupernovaEnableEpoch", None)
-
-        node_config_toml.patch_config(node_config_data, config, supernova_activation_epoch)
+        node_config_toml.patch_config(node_config_data, config, enable_epochs_config_data)
         node_config_toml.patch_api(api_config_data, config)
         node_config_toml.patch_enable_epochs(enable_epochs_config_data, config)
+        node_config_toml.patch_enable_rounds(enable_rounds_config_data, config, enable_epochs_config_data)
         genesis_smart_contracts_json.patch(genesis_smart_contracts_data, config)
 
         utils.write_toml_file(node_config_file, node_config_data)
         utils.write_toml_file(api_config_file, api_config_data)
         utils.write_toml_file(enable_epochs_config_file, enable_epochs_config_data)
+        utils.write_toml_file(enable_rounds_config_file, enable_rounds_config_data)
         utils.write_json_file(genesis_smart_contracts_file, genesis_smart_contracts_data)