Fix default gas limit on execution config

attestantio · Oct 30, 2024 · 5d7ca19 · 5d7ca19
1 parent 6b370f5
commit 5d7ca19
Show file tree

Hide file tree

Showing 2 changed files with 61 additions and 1 deletion.
diff --git a/services/blockrelay/v2/executionconfig.go b/services/blockrelay/v2/executionconfig.go
@@ -192,7 +192,11 @@ func (e *ExecutionConfig) setInitialRelayOptions(_ context.Context,
 		} else {
 			configRelay.MinValue = *e.MinValue
 		}
-		setRelayConfig(configRelay, baseRelayConfig, config.FeeRecipient, fallbackGasLimit)
+		if e.GasLimit == nil {
+			setRelayConfig(configRelay, baseRelayConfig, config.FeeRecipient, fallbackGasLimit)
+		} else {
+			setRelayConfig(configRelay, baseRelayConfig, config.FeeRecipient, *e.GasLimit)
+		}
 		config.Relays = append(config.Relays, configRelay)
 	}
 }

diff --git a/services/blockrelay/v2/executionconfig_test.go b/services/blockrelay/v2/executionconfig_test.go
@@ -672,6 +672,62 @@ func TestConfig(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "ExecutionConfigDefaultGasLimit",
+			executionConfig: &v2.ExecutionConfig{
+				GasLimit:     &gasLimit4,
+				Proposers:    []*v2.ProposerConfig{},
+				FeeRecipient: &feeRecipient3,
+				Relays: map[string]*v2.BaseRelayConfig{
+					"https://relay1.com/": {},
+				},
+			},
+			account:              account1,
+			pubkey:               pubkey1,
+			fallbackFeeRecipient: feeRecipient1,
+			fallbackGasLimit:     gasLimit1,
+			expected: &beaconblockproposer.ProposerConfig{
+				FeeRecipient: feeRecipient3,
+				Relays: []*beaconblockproposer.RelayConfig{
+					{
+						Address:      "https://relay1.com/",
+						FeeRecipient: feeRecipient3,
+						GasLimit:     gasLimit4,
+						Grace:        grace0,
+						MinValue:     minValue0,
+					},
+				},
+			},
+		},
+		{
+			name: "ExecutionConfigDefaultAndRelayGasLimit",
+			executionConfig: &v2.ExecutionConfig{
+				GasLimit:     &gasLimit4,
+				Proposers:    []*v2.ProposerConfig{},
+				FeeRecipient: &feeRecipient3,
+				Relays: map[string]*v2.BaseRelayConfig{
+					"https://relay1.com/": {
+						GasLimit: &gasLimit3,
+					},
+				},
+			},
+			account:              account1,
+			pubkey:               pubkey1,
+			fallbackFeeRecipient: feeRecipient1,
+			fallbackGasLimit:     gasLimit1,
+			expected: &beaconblockproposer.ProposerConfig{
+				FeeRecipient: feeRecipient3,
+				Relays: []*beaconblockproposer.RelayConfig{
+					{
+						Address:      "https://relay1.com/",
+						FeeRecipient: feeRecipient3,
+						GasLimit:     gasLimit3,
+						Grace:        grace0,
+						MinValue:     minValue0,
+					},
+				},
+			},
+		},
 		{
 			name: "InvalidProposerConfig",
 			executionConfig: &v2.ExecutionConfig{