mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 15:37:56 -05:00
4ff91bebf8
* Update gomock * Update mockgen * Gaz * Go mod * Cleanup * Regenerate gomock * Manually fix import
53 lines
1.7 KiB
Go
53 lines
1.7 KiB
Go
package beacon_api
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v5/api/server/structs"
|
|
rpctesting "github.com/prysmaticlabs/prysm/v5/beacon-chain/rpc/eth/shared/testing"
|
|
"github.com/prysmaticlabs/prysm/v5/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v5/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/client/beacon-api/mock"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
func TestProposeBeaconBlock_Deneb(t *testing.T) {
|
|
t.Skip("TODO: Fix this in the beacon-API PR")
|
|
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
jsonRestHandler := mock.NewMockJsonRestHandler(ctrl)
|
|
|
|
var blockContents structs.SignedBeaconBlockContentsDeneb
|
|
err := json.Unmarshal([]byte(rpctesting.DenebBlockContents), &blockContents)
|
|
require.NoError(t, err)
|
|
genericSignedBlock, err := blockContents.ToGeneric()
|
|
require.NoError(t, err)
|
|
|
|
denebBytes, err := json.Marshal(blockContents)
|
|
require.NoError(t, err)
|
|
// Make sure that what we send in the POST body is the marshalled version of the protobuf block
|
|
headers := map[string]string{"Eth-Consensus-Version": "deneb"}
|
|
jsonRestHandler.EXPECT().Post(
|
|
context.Background(),
|
|
"/eth/v1/beacon/blocks",
|
|
headers,
|
|
bytes.NewBuffer(denebBytes),
|
|
nil,
|
|
)
|
|
|
|
validatorClient := &beaconApiValidatorClient{jsonRestHandler: jsonRestHandler}
|
|
proposeResponse, err := validatorClient.proposeBeaconBlock(context.Background(), genericSignedBlock)
|
|
assert.NoError(t, err)
|
|
require.NotNil(t, proposeResponse)
|
|
|
|
expectedBlockRoot, err := genericSignedBlock.GetDeneb().Block.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
|
|
// Make sure that the block root is set
|
|
assert.DeepEqual(t, expectedBlockRoot[:], proposeResponse.BlockRoot)
|
|
}
|