mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
98f8ca4e34
* adding reverting change * adding additional tests and checks * removing comment * updating based on review suggestions * fixing deepsource issues * attempting to resolve dependencies * removing dependency * accidently flipped if statement with deepsource fix * fixing unit test
86 lines
2.9 KiB
Go
86 lines
2.9 KiB
Go
package testing
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v4/api/client/builder"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/cache"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/db"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/blocks"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/interfaces"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
v1 "github.com/prysmaticlabs/prysm/v4/proto/engine/v1"
|
|
ethpb "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v4/time/slots"
|
|
)
|
|
|
|
// Config defines a config struct for dependencies into the service.
|
|
type Config struct {
|
|
BeaconDB db.HeadAccessDatabase
|
|
}
|
|
|
|
// MockBuilderService to mock builder.
|
|
type MockBuilderService struct {
|
|
HasConfigured bool
|
|
Payload *v1.ExecutionPayload
|
|
PayloadCapella *v1.ExecutionPayloadCapella
|
|
ErrSubmitBlindedBlock error
|
|
Bid *ethpb.SignedBuilderBid
|
|
BidCapella *ethpb.SignedBuilderBidCapella
|
|
RegistrationCache *cache.RegistrationCache
|
|
ErrGetHeader error
|
|
ErrRegisterValidator error
|
|
Cfg *Config
|
|
}
|
|
|
|
// Configured for mocking.
|
|
func (s *MockBuilderService) Configured() bool {
|
|
return s.HasConfigured
|
|
}
|
|
|
|
// SubmitBlindedBlock for mocking.
|
|
func (s *MockBuilderService) SubmitBlindedBlock(_ context.Context, _ interfaces.ReadOnlySignedBeaconBlock) (interfaces.ExecutionData, error) {
|
|
if s.Payload != nil {
|
|
w, err := blocks.WrappedExecutionPayload(s.Payload)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not wrap payload")
|
|
}
|
|
return w, s.ErrSubmitBlindedBlock
|
|
}
|
|
w, err := blocks.WrappedExecutionPayloadCapella(s.PayloadCapella, 0)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not wrap capella payload")
|
|
}
|
|
return w, s.ErrSubmitBlindedBlock
|
|
}
|
|
|
|
// GetHeader for mocking.
|
|
func (s *MockBuilderService) GetHeader(_ context.Context, slot primitives.Slot, _ [32]byte, _ [48]byte) (builder.SignedBid, error) {
|
|
if slots.ToEpoch(slot) >= params.BeaconConfig().CapellaForkEpoch || s.BidCapella != nil {
|
|
return builder.WrappedSignedBuilderBidCapella(s.BidCapella)
|
|
}
|
|
w, err := builder.WrappedSignedBuilderBid(s.Bid)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not wrap capella bid")
|
|
}
|
|
return w, s.ErrGetHeader
|
|
}
|
|
|
|
// RegistrationByValidatorID returns either the values from the cache or db.
|
|
func (s *MockBuilderService) RegistrationByValidatorID(ctx context.Context, id primitives.ValidatorIndex) (*ethpb.ValidatorRegistrationV1, error) {
|
|
if s.RegistrationCache != nil {
|
|
return s.RegistrationCache.RegistrationByIndex(id)
|
|
}
|
|
if s.Cfg.BeaconDB != nil {
|
|
return s.Cfg.BeaconDB.RegistrationByValidatorID(ctx, id)
|
|
}
|
|
return nil, cache.ErrNotFoundRegistration
|
|
}
|
|
|
|
// RegisterValidator for mocking.
|
|
func (s *MockBuilderService) RegisterValidator(context.Context, []*ethpb.SignedValidatorRegistrationV1) error {
|
|
return s.ErrRegisterValidator
|
|
}
|