test(consensus-types): add more test for block_denebplus

mod/consensus-types/pkg/types/block_denebplus_test.go

@@ -0,0 +1,242 @@
+// SPDX-License-Identifier: BUSL-1.1
+//
+// Copyright (C) 2024, Berachain Foundation. All rights reserved.
+// Use of this software is governed by the Business Source License included
+// in the LICENSE file of this repository and at www.mariadb.com/bsl11.
+//
+// ANY USE OF THE LICENSED WORK IN VIOLATION OF THIS LICENSE WILL AUTOMATICALLY
+// TERMINATE YOUR RIGHTS UNDER THIS LICENSE FOR THE CURRENT AND ALL OTHER
+// VERSIONS OF THE LICENSED WORK.
+//
+// THIS LICENSE DOES NOT GRANT YOU ANY RIGHT IN ANY TRADEMARK OR LOGO OF
+// LICENSOR OR ITS AFFILIATES (PROVIDED THAT YOU MAY USE A TRADEMARK OR LOGO OF
+// LICENSOR AS EXPRESSLY REQUIRED BY THIS LICENSE).
+//
+// TO THE EXTENT PERMITTED BY APPLICABLE LAW, THE LICENSED WORK IS PROVIDED ON
+// AN “AS IS” BASIS. LICENSOR HEREBY DISCLAIMS ALL WARRANTIES AND CONDITIONS,
+// EXPRESS OR IMPLIED, INCLUDING (WITHOUT LIMITATION) WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NON-INFRINGEMENT, AND
+// TITLE.
+
+package types_test
+
+import (
+	"testing"
+
+	"github.com/berachain/beacon-kit/mod/consensus-types/pkg/types"
+	engineprimitives "github.com/berachain/beacon-kit/mod/engine-primitives/pkg/engine-primitives"
+	gethprimitives "github.com/berachain/beacon-kit/mod/geth-primitives"
+	"github.com/berachain/beacon-kit/mod/primitives/pkg/common"
+	"github.com/berachain/beacon-kit/mod/primitives/pkg/eip4844"
+	"github.com/berachain/beacon-kit/mod/primitives/pkg/version"
+	ssz "github.com/ferranbt/fastssz"
+	"github.com/stretchr/testify/require"
+)
+
+func generateBeaconBlockDenebPlus() *types.BeaconBlockDenebPlus {
+	var byteArray [256]byte
+	byteSlice := byteArray[:]
+	return &types.BeaconBlockDenebPlus{
+		BeaconBlockHeaderBase: types.BeaconBlockHeaderBase{
+			Slot:          12345,
+			ProposerIndex: 67890,
+			ParentBlockRoot: [32]byte{
+				1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
+				20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32},
+			StateRoot: [32]byte{
+				32, 31, 30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18,
+				17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1},
+		},
+		Body: &types.BeaconBlockBodyDenebPlus{
+			BeaconBlockBodyBase: types.BeaconBlockBodyBase{
+				RandaoReveal: [96]byte{
+					1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
+					20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36,
+					37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53,
+					54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70,
+					71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87,
+					88, 89, 90, 91, 92, 93, 94, 95, 96},
+				Eth1Data: &types.Eth1Data{
+					DepositRoot: common.Root{
+						0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09,
+						0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12, 0x13,
+						0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1A, 0x1B, 0x1C, 0x1D,
+						0x1E, 0x1F,
+					},
+					DepositCount: 12345,
+					BlockHash: gethprimitives.ExecutionHash{
+						0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09,
+						0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12, 0x13,
+						0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1A, 0x1B, 0x1C, 0x1D,
+						0x1E, 0x1F,
+					},
+				},
+				Graffiti: [32]byte{
+					1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
+					20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32,
+				},
+				Deposits: []*types.Deposit{},
+			},
+			ExecutionPayload: &types.ExecutableDataDeneb{
+				LogsBloom:    byteSlice,
+				ExtraData:    make([]byte, 0),
+				Transactions: make([][]byte, 0),
+				Withdrawals:  make([]*engineprimitives.Withdrawal, 0),
+			},
+			Attestations:       []*types.AttestationData{},
+			SlashingInfo:       []*types.SlashingInfo{},
+			BlobKzgCommitments: []eip4844.KZGCommitment{},
+		},
+	}
+}
+
+func TestBeaconBlockDenebPlus_Version(t *testing.T) {
+	block := generateBeaconBlockDenebPlus()
+	require.Equal(t, version.Deneb, block.Version())
+}
+
+func TestBeaconBlockDenebPlus_IsNil(t *testing.T) {
+	var block *types.BeaconBlockDenebPlus
+	require.True(t, block.IsNil())
+
+	block = generateBeaconBlockDenebPlus()
+	require.False(t, block.IsNil())
+}
+
+func TestBeaconBlockDenebPlus_SetStateRoot(t *testing.T) {
+	block := generateBeaconBlockDenebPlus()
+	newRoot := [32]uint8{
+		0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,
+		21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31}
+	block.SetStateRoot(newRoot)
+	require.Equal(t, newRoot, [32]uint8(block.StateRoot))
+}
+
+func TestBeaconBlockDenebPlus_GetBody(t *testing.T) {
+	block := generateBeaconBlockDenebPlus()
+	body := block.GetBody()
+	require.NotNil(t, body)
+	require.Equal(t, block.Body, body.RawBeaconBlockBody)
+}
+
+func TestBeaconBlockDenebPlus_GetHeader(t *testing.T) {
+	block := generateBeaconBlockDenebPlus()
+	header := block.GetHeader()
+	require.NotNil(t, header)
+	require.Equal(t, block.Slot, header.Slot)
+	require.Equal(t, block.ProposerIndex, header.ProposerIndex)
+	require.Equal(t, block.ParentBlockRoot, header.ParentBlockRoot)
+	require.Equal(t, block.StateRoot, header.StateRoot)
+}
+
+func TestBeaconBlockDenebPlus_MarshalSSZ_UnmarshalSSZ(t *testing.T) {
+	var byteArray [256]byte
+	byteSlice := byteArray[:]
+	testCases := []struct {
+		name     string
+		data     *types.BeaconBlockDenebPlus
+		expected *types.BeaconBlockDenebPlus
+		err      error
+	}{
+		{
+			name:     "Valid BeaconBlockDenebPlus",
+			data:     generateBeaconBlockDenebPlus(),
+			expected: generateBeaconBlockDenebPlus(),
+			err:      nil,
+		},
+		{
+			name: "Empty BeaconBlockDenebPlus",
+			data: &types.BeaconBlockDenebPlus{
+				BeaconBlockHeaderBase: types.BeaconBlockHeaderBase{
+					Slot:            0,
+					ProposerIndex:   0,
+					ParentBlockRoot: [32]byte{},
+					StateRoot:       [32]byte{},
+				},
+				Body: &types.BeaconBlockBodyDenebPlus{
+					BeaconBlockBodyBase: types.BeaconBlockBodyBase{
+						RandaoReveal: [96]byte{},
+						Eth1Data:     &types.Eth1Data{},
+						Graffiti:     [32]byte{},
+						Deposits:     []*types.Deposit{},
+					},
+					ExecutionPayload: &types.ExecutableDataDeneb{
+						LogsBloom:    byteSlice,
+						ExtraData:    make([]byte, 0),
+						Transactions: make([][]byte, 0),
+						Withdrawals:  make([]*engineprimitives.Withdrawal, 0),
+					},
+					Attestations:       []*types.AttestationData{},
+					SlashingInfo:       []*types.SlashingInfo{},
+					BlobKzgCommitments: []eip4844.KZGCommitment{},
+				},
+			},
+			expected: &types.BeaconBlockDenebPlus{
+				BeaconBlockHeaderBase: types.BeaconBlockHeaderBase{
+					Slot:            0,
+					ProposerIndex:   0,
+					ParentBlockRoot: [32]byte{},
+					StateRoot:       [32]byte{},
+				},
+				Body: &types.BeaconBlockBodyDenebPlus{
+					BeaconBlockBodyBase: types.BeaconBlockBodyBase{
+						RandaoReveal: [96]byte{},
+						Eth1Data:     &types.Eth1Data{},
+						Graffiti:     [32]byte{},
+						Deposits:     []*types.Deposit{},
+					},
+					ExecutionPayload: &types.ExecutableDataDeneb{
+						LogsBloom:    byteSlice,
+						ExtraData:    make([]byte, 0),
+						Transactions: make([][]byte, 0),
+						Withdrawals:  make([]*engineprimitives.Withdrawal, 0),
+					},
+					Attestations:       []*types.AttestationData{},
+					SlashingInfo:       []*types.SlashingInfo{},
+					BlobKzgCommitments: []eip4844.KZGCommitment{},
+				},
+			},
+			err: nil,
+		},
+		{
+			name:     "Invalid Buffer Size",
+			data:     generateBeaconBlockDenebPlus(),
+			expected: nil,
+			err:      ssz.ErrSize,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			data, err := tc.data.MarshalSSZ()
+			require.NoError(t, err)
+			require.NotNil(t, data)
+
+			var unmarshalled types.BeaconBlockDenebPlus
+			if tc.name == "Invalid Buffer Size" {
+				err = unmarshalled.UnmarshalSSZ(data[:32])
+				require.Error(t, err)
+				require.Equal(t, tc.err, err)
+			} else {
+				err = unmarshalled.UnmarshalSSZ(data)
+				require.NoError(t, err)
+				require.Equal(t, tc.expected, &unmarshalled)
+			}
+		})
+	}
+}
+
+func TestBeaconBlockDenebPlus_GetTree(t *testing.T) {
+	block := generateBeaconBlockDenebPlus()
+
+	tree, err := block.GetTree()
+	require.NoError(t, err)
+	require.NotNil(t, tree)
+
+	expectedRoot, err := block.HashTreeRoot()
+	require.NoError(t, err)
+
+	// Compare the tree root with the expected root
+	actualRoot := tree.Hash()
+	require.Equal(t, string(expectedRoot[:]), string(actualRoot))
+}