Only override private BLS key in test mode if TestPrivateBlsFlag is d…

…efined
Layr-Labs · Feb 7, 2025 · 3c532bf · 3c532bf
1 parent 3db790b
commit 3c532bf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/node/config.go b/node/config.go
@@ -212,7 +212,7 @@ func NewConfig(ctx *cli.Context) (*Config, error) {
 			TLSCertFilePath:  ctx.GlobalString(flags.BLSSignerCertFileFlag.Name),
 			CerberusAPIKey:   blsSignerAPIKey,
 		}
-	} else {
+	} else if testMode && ctx.GlobalString(flags.TestPrivateBlsFlag.Name) != "" {
 		privateBls := ctx.GlobalString(flags.TestPrivateBlsFlag.Name)
 		blsSignerConfig = blssignerTypes.SignerConfig{
 			SignerType: blssignerTypes.PrivateKey,

diff --git a/node/node.go b/node/node.go
@@ -158,6 +158,7 @@ func NewNode(
 	var blockStaleMeasure, storeDurationBlocks uint32
 	if config.EnableTestMode && config.OverrideBlockStaleMeasure > 0 {
 		blockStaleMeasure = uint32(config.OverrideBlockStaleMeasure)
+		logger.Info("Test Mode Override!", "blockStaleMeasure", blockStaleMeasure)
 	} else {
 		staleMeasure, err := tx.GetBlockStaleMeasure(context.Background())
 		if err != nil {
@@ -167,6 +168,7 @@ func NewNode(
 	}
 	if config.EnableTestMode && config.OverrideStoreDurationBlocks > 0 {
 		storeDurationBlocks = uint32(config.OverrideStoreDurationBlocks)
+		logger.Info("Test Mode Override!", "storeDurationBlocks", storeDurationBlocks)
 	} else {
 		storeDuration, err := tx.GetStoreDurationBlocks(context.Background())
 		if err != nil {