add tests for beacon-chain/state/state-native/getters_balance_deposit…

…s.go

beacon-chain/state/state-native/BUILD.bazel

@@ -89,6 +89,7 @@ go_test(
     name = "go_default_test",
     srcs = [
         "getters_attestation_test.go",
+        "getters_balance_deposits_test.go",
         "getters_block_test.go",
         "getters_checkpoint_test.go",
         "getters_consolidation_test.go",

beacon-chain/state/state-native/getters_balance_deposits_test.go

@@ -0,0 +1,48 @@
+package state_native_test
+
+import (
+	"testing"
+
+	state_native "github.com/prysmaticlabs/prysm/v5/beacon-chain/state/state-native"
+	eth "github.com/prysmaticlabs/prysm/v5/proto/prysm/v1alpha1"
+	"github.com/prysmaticlabs/prysm/v5/testing/require"
+)
+
+func TestDepositBalanceToConsume(t *testing.T) {
+	s, err := state_native.InitializeFromProtoElectra(&eth.BeaconStateElectra{
+		DepositBalanceToConsume: 44,
+	})
+	require.NoError(t, err)
+	dbtc, err := s.DepositBalanceToConsume()
+	require.NoError(t, err)
+	require.Equal(t, uint64(44), dbtc)
+
+	// Fails for older than electra state
+	s, err = state_native.InitializeFromProtoDeneb(&eth.BeaconStateDeneb{})
+	require.NoError(t, err)
+	_, err = s.DepositBalanceToConsume()
+	require.ErrorContains(t, "not supported", err)
+}
+
+func TestPendingBalanceDeposits(t *testing.T) {
+	s, err := state_native.InitializeFromProtoElectra(&eth.BeaconStateElectra{
+		PendingBalanceDeposits: []*eth.PendingBalanceDeposit{
+			{Index: 1, Amount: 2},
+			{Index: 3, Amount: 4},
+		},
+	})
+	require.NoError(t, err)
+	pbd, err := s.PendingBalanceDeposits()
+	require.NoError(t, err)
+	require.Equal(t, 2, len(pbd))
+	require.Equal(t, 1, pbd[0].Index)
+	require.Equal(t, 2, pbd[0].Amount)
+	require.Equal(t, 3, pbd[1].Index)
+	require.Equal(t, 4, pbd[1].Amount)
+
+	// Fails for older than electra state
+	s, err = state_native.InitializeFromProtoDeneb(&eth.BeaconStateDeneb{})
+	require.NoError(t, err)
+	_, err = s.DepositBalanceToConsume()
+	require.ErrorContains(t, "not supported", err)
+}