prysm-pulse/validator/client/fake_validator_test.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
}