mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-06 01:32:18 +00:00
956721fe43
* Add attestation p2p messages * first pass refactor * lint * viz fix * comment unused const for now * Add opentracing spans * PR feedback * added updateAssignments implementation * update proto * handle error in runner * only do updates on epoch start * gazelle * goimports
67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/rpc/v1"
|
|
)
|
|
|
|
var _ = Validator(&fakeValidator{})
|
|
|
|
type fakeValidator struct {
|
|
InitializeCalled bool
|
|
DoneCalled bool
|
|
WaitForActivationCalled bool
|
|
NextSlotRet <-chan uint64
|
|
NextSlotCalled bool
|
|
UpdateAssignmentsCalled bool
|
|
UpdateAssignmentsArg1 uint64
|
|
UpdateAssignmentsRet error
|
|
RoleAtCalled bool
|
|
RoleAtArg1 uint64
|
|
RoleAtRet pb.ValidatorRole
|
|
AttestToBlockHeadCalled bool
|
|
AttestToBlockHeadArg1 uint64
|
|
ProposeBlockCalled bool
|
|
ProposeBlockArg1 uint64
|
|
}
|
|
|
|
func (fv *fakeValidator) Initialize(_ context.Context) {
|
|
fv.InitializeCalled = true
|
|
}
|
|
|
|
func (fv *fakeValidator) Done() {
|
|
fv.DoneCalled = true
|
|
}
|
|
|
|
func (fv *fakeValidator) WaitForActivation(_ context.Context) {
|
|
fv.WaitForActivationCalled = true
|
|
}
|
|
|
|
func (fv *fakeValidator) NextSlot() <-chan uint64 {
|
|
fv.NextSlotCalled = true
|
|
return fv.NextSlotRet
|
|
}
|
|
|
|
func (fv *fakeValidator) UpdateAssignments(_ context.Context, slot uint64) error {
|
|
fv.UpdateAssignmentsCalled = true
|
|
fv.UpdateAssignmentsArg1 = slot
|
|
return fv.UpdateAssignmentsRet
|
|
}
|
|
|
|
func (fv *fakeValidator) RoleAt(slot uint64) pb.ValidatorRole {
|
|
fv.RoleAtCalled = true
|
|
fv.RoleAtArg1 = slot
|
|
return fv.RoleAtRet
|
|
}
|
|
|
|
func (fv *fakeValidator) AttestToBlockHead(_ context.Context, slot uint64) {
|
|
fv.AttestToBlockHeadCalled = true
|
|
fv.AttestToBlockHeadArg1 = slot
|
|
}
|
|
|
|
func (fv *fakeValidator) ProposeBlock(_ context.Context, slot uint64) {
|
|
fv.ProposeBlockCalled = true
|
|
fv.ProposeBlockArg1 = slot
|
|
}
|