mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-10 19:51:20 +00:00
30 lines
753 B
Go
30 lines
753 B
Go
package testutil
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/state"
|
|
"github.com/prysmaticlabs/prysm/v3/consensus-types/primitives"
|
|
)
|
|
|
|
// MockFetcher is a fake implementation of statefetcher.Fetcher.
|
|
type MockFetcher struct {
|
|
BeaconState state.BeaconState
|
|
BeaconStateRoot []byte
|
|
StatesBySlot map[primitives.Slot]state.BeaconState
|
|
}
|
|
|
|
// State --
|
|
func (m *MockFetcher) State(context.Context, []byte) (state.BeaconState, error) {
|
|
return m.BeaconState, nil
|
|
}
|
|
|
|
// StateRoot --
|
|
func (m *MockFetcher) StateRoot(context.Context, []byte) ([]byte, error) {
|
|
return m.BeaconStateRoot, nil
|
|
}
|
|
|
|
func (m *MockFetcher) StateBySlot(_ context.Context, s primitives.Slot) (state.BeaconState, error) {
|
|
return m.StatesBySlot[s], nil
|
|
}
|