prysm-pulse/validator/client/fake_validator_test.go
Raul Jordan fcc18fac02
Treat Post & Pre-ChainStart Validators Equally in Assignments Fetching (#2074)
* treat post/prechainstart validators equally in assignment fetching

* Update validator/client/fake_validator_test.go

Co-Authored-By: rauljordan <raul@prysmaticlabs.com>

* status err code

* imports

* beyond activation epoch

* assigned to slot log

* gazelle
2019-03-26 09:03:05 -05:00

83 lines
2.2 KiB
Go

package client
import (
"context"
"time"
pb "github.com/prysmaticlabs/prysm/proto/beacon/rpc/v1"
"github.com/prysmaticlabs/prysm/shared/params"
)
var _ = Validator(&fakeValidator{})
type fakeValidator struct {
DoneCalled bool
CanonicalHeadSlotCalled bool
WaitForChainStartCalled 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
LogValidatorGainsAndLossesCalled bool
SlotDeadlineCalled bool
}
func (fv *fakeValidator) Done() {
fv.DoneCalled = true
}
func (fv *fakeValidator) WaitForChainStart(_ context.Context) error {
fv.WaitForChainStartCalled = true
return nil
}
func (fv *fakeValidator) CanonicalHeadSlot(_ context.Context) (uint64, error) {
fv.CanonicalHeadSlotCalled = true
return params.BeaconConfig().GenesisSlot, nil
}
func (fv *fakeValidator) SlotDeadline(_ uint64) time.Time {
fv.SlotDeadlineCalled = true
return time.Now()
}
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) LogValidatorGainsAndLosses(_ context.Context, slot uint64) error {
fv.LogValidatorGainsAndLossesCalled = true
return nil
}
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
}