github.com/prysmaticlabs/prysm@v1.4.4/beacon-chain/operations/slashings/mock.go (about)

     1  package slashings
     2  
     3  import (
     4  	"context"
     5  
     6  	iface "github.com/prysmaticlabs/prysm/beacon-chain/state/interface"
     7  	ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
     8  )
     9  
    10  // PoolMock is a fake implementation of PoolManager.
    11  type PoolMock struct {
    12  	PendingAttSlashings  []*ethpb.AttesterSlashing
    13  	PendingPropSlashings []*ethpb.ProposerSlashing
    14  }
    15  
    16  // PendingAttesterSlashings --
    17  func (m *PoolMock) PendingAttesterSlashings(_ context.Context, _ iface.ReadOnlyBeaconState, _ bool) []*ethpb.AttesterSlashing {
    18  	return m.PendingAttSlashings
    19  }
    20  
    21  // PendingProposerSlashings --
    22  func (m *PoolMock) PendingProposerSlashings(_ context.Context, _ iface.ReadOnlyBeaconState, _ bool) []*ethpb.ProposerSlashing {
    23  	return m.PendingPropSlashings
    24  }
    25  
    26  // InsertAttesterSlashing --
    27  func (m *PoolMock) InsertAttesterSlashing(_ context.Context, _ iface.ReadOnlyBeaconState, slashing *ethpb.AttesterSlashing) error {
    28  	m.PendingAttSlashings = append(m.PendingAttSlashings, slashing)
    29  	return nil
    30  }
    31  
    32  // InsertProposerSlashing --
    33  func (m *PoolMock) InsertProposerSlashing(_ context.Context, _ iface.BeaconState, slashing *ethpb.ProposerSlashing) error {
    34  	m.PendingPropSlashings = append(m.PendingPropSlashings, slashing)
    35  	return nil
    36  }
    37  
    38  // MarkIncludedAttesterSlashing --
    39  func (m *PoolMock) MarkIncludedAttesterSlashing(_ *ethpb.AttesterSlashing) {
    40  	panic("implement me")
    41  }
    42  
    43  // MarkIncludedProposerSlashing --
    44  func (m *PoolMock) MarkIncludedProposerSlashing(_ *ethpb.ProposerSlashing) {
    45  	panic("implement me")
    46  }