mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-24 12:27:18 +00:00
df33ce3309
* slasher beacon node changes * remaining beacon node items * moar changes * gaz * flag fix * rem slashable * builds * imports * fix up * pruning faster test * deepsource * fix wrong item * node node feature flags * broken test * preston review * more preston comments * comment Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
35 lines
866 B
Go
35 lines
866 B
Go
package testing
|
|
|
|
import (
|
|
"context"
|
|
|
|
eth "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
)
|
|
|
|
// MockProtector mocks the protector.
|
|
type MockProtector struct {
|
|
AllowAttestation bool
|
|
AllowBlock bool
|
|
VerifyAttestationCalled bool
|
|
VerifyBlockCalled bool
|
|
StatusCalled bool
|
|
}
|
|
|
|
// CheckAttestationSafety returns bool with allow attestation value.
|
|
func (mp MockProtector) CheckAttestationSafety(_ context.Context, _ *eth.IndexedAttestation) bool {
|
|
mp.VerifyAttestationCalled = true
|
|
return mp.AllowAttestation
|
|
}
|
|
|
|
// CheckBlockSafety returns bool with allow block value.
|
|
func (mp MockProtector) CheckBlockSafety(_ context.Context, _ *eth.SignedBeaconBlockHeader) bool {
|
|
mp.VerifyBlockCalled = true
|
|
return mp.AllowBlock
|
|
}
|
|
|
|
// Status returns nil.
|
|
func (mp MockProtector) Status() error {
|
|
mp.StatusCalled = true
|
|
return nil
|
|
}
|