prysm-pulse/beacon-chain/rpc/testutil/mock_state_fetcher.go
Radosław Kapka 77d3ccb9ad
Clean up state types (#11916)
* Clean up state types

* rename package
2023-01-26 14:40:12 +00:00

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
}