prysm-pulse/slasher/detection/attestations/mock_spanner.go
Radosław Kapka 4b14fa4317
Better receiver names in validator and slasher modules (#8296)
* validator

* slasher

* rename db to s for store

Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
2021-01-20 14:39:07 +00:00

79 lines
3.0 KiB
Go

package attestations
import (
"context"
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
"github.com/prysmaticlabs/prysm/slasher/detection/attestations/iface"
"github.com/prysmaticlabs/prysm/slasher/detection/attestations/types"
)
var _ iface.SpanDetector = (*MockSpanDetector)(nil)
// MockSpanDetector defines a struct which can detect slashable
// attestation offenses by tracking validator min-max
// spans from validators.
type MockSpanDetector struct{}
// DetectSlashingsForAttestation mocks a detected slashing, if the sent attestation data
// has a source epoch of 0, nothing will be detected. If the sent attestation data has a target
// epoch equal to or greater than 6, it will "detect" a surrounded vote for the target epoch + 1.
// If the target epoch is greater than 12, it will "detect" a surrounding vote for target epoch - 1.
// Lastly, if it has a target epoch less than 6, it will "detect" a double vote for the target epoch.
func (m *MockSpanDetector) DetectSlashingsForAttestation(
_ context.Context,
att *ethpb.IndexedAttestation,
) ([]*types.DetectionResult, error) {
var detections []*types.DetectionResult
switch {
// If the source epoch is 0, don't find a slashing.
case att.Data.Source.Epoch == 0:
return nil, nil
// If the target epoch is > 12, it will "detect" a surrounded saved attestation.
case att.Data.Target.Epoch > 12:
detections = append(detections, &types.DetectionResult{
Kind: types.SurroundVote,
SlashableEpoch: att.Data.Target.Epoch - 1,
SigBytes: [2]byte{1, 2},
})
return detections, nil
// If the target epoch is >= 6 < 12, it will "detect" a surrounding saved attestation.
case att.Data.Target.Epoch >= 6:
detections = append(detections, &types.DetectionResult{
Kind: types.SurroundVote,
SlashableEpoch: att.Data.Target.Epoch + 1,
SigBytes: [2]byte{1, 2},
})
return detections, nil
// If the target epoch is less than 6, it will "detect" a double vote.
default:
detections = append(detections, &types.DetectionResult{
Kind: types.DoubleVote,
SlashableEpoch: att.Data.Target.Epoch,
SigBytes: [2]byte{1, 2},
})
}
return detections, nil
}
// SpanForEpochByValidator returns the specific min-max span for a
// validator index in a given epoch.
func (m *MockSpanDetector) SpanForEpochByValidator(_ context.Context, _, _ uint64) (types.Span, error) {
return types.Span{MinSpan: 0, MaxSpan: 0, SigBytes: [2]byte{}, HasAttested: false}, nil
}
// ValidatorSpansByEpoch returns a list of all validator spans in a given epoch.
func (m *MockSpanDetector) ValidatorSpansByEpoch(_ context.Context, _ uint64) map[uint64]types.Span {
return make(map[uint64]types.Span)
}
// DeleteValidatorSpansByEpoch mocks the delete spans by epoch function.
func (m *MockSpanDetector) DeleteValidatorSpansByEpoch(_ context.Context, _, _ uint64) error {
return nil
}
// UpdateSpans is a mock for updating the spans for a given attestation..
func (m *MockSpanDetector) UpdateSpans(_ context.Context, _ *ethpb.IndexedAttestation) error {
return nil
}