prysm-pulse/validator/client/fake_validator_test.go
Preston Van Loon 0becb6dc01 Simplified validator workflow - first pass (#1362)
* Add attestation p2p messages

* first pass refactor

* lint

* viz fix

* comment unused const for now

* Add opentracing spans

* PR feedback
2019-01-23 10:52:39 +08:00

65 lines
1.5 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
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) {
fv.UpdateAssignmentsCalled = true
fv.UpdateAssignmentsArg1 = slot
}
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
}