mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
8310d22a05
* initial commit * fixing variable rename * fixing unit test * adding based on review comments * renaming cache * simplifying logic on signed validator registrationRequest * adding unit tests * fixing linting * using wrong dependency
199 lines
5.0 KiB
Go
199 lines
5.0 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
types "github.com/prysmaticlabs/prysm/consensus-types/primitives"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/validator/accounts/iface"
|
|
iface2 "github.com/prysmaticlabs/prysm/validator/client/iface"
|
|
"github.com/prysmaticlabs/prysm/validator/keymanager"
|
|
)
|
|
|
|
// Wallet contains an in-memory, simulated wallet implementation.
|
|
type Wallet struct {
|
|
InnerAccountsDir string
|
|
Directories []string
|
|
Files map[string]map[string][]byte
|
|
EncryptedSeedFile []byte
|
|
AccountPasswords map[string]string
|
|
WalletPassword string
|
|
UnlockAccounts bool
|
|
lock sync.RWMutex
|
|
}
|
|
|
|
// AccountNames --
|
|
func (w *Wallet) AccountNames() ([]string, error) {
|
|
w.lock.RLock()
|
|
defer w.lock.RUnlock()
|
|
names := make([]string, 0)
|
|
for name := range w.AccountPasswords {
|
|
names = append(names, name)
|
|
}
|
|
return names, nil
|
|
}
|
|
|
|
// AccountsDir --
|
|
func (w *Wallet) AccountsDir() string {
|
|
return w.InnerAccountsDir
|
|
}
|
|
|
|
// Exists --
|
|
func (w *Wallet) Exists() (bool, error) {
|
|
return len(w.Directories) > 0, nil
|
|
}
|
|
|
|
// Password --
|
|
func (w *Wallet) Password() string {
|
|
return w.WalletPassword
|
|
}
|
|
|
|
// WriteFileAtPath --
|
|
func (w *Wallet) WriteFileAtPath(_ context.Context, pathName, fileName string, data []byte) error {
|
|
w.lock.Lock()
|
|
defer w.lock.Unlock()
|
|
if w.Files[pathName] == nil {
|
|
w.Files[pathName] = make(map[string][]byte)
|
|
}
|
|
w.Files[pathName][fileName] = data
|
|
return nil
|
|
}
|
|
|
|
// ReadFileAtPath --
|
|
func (w *Wallet) ReadFileAtPath(_ context.Context, pathName, fileName string) ([]byte, error) {
|
|
w.lock.RLock()
|
|
defer w.lock.RUnlock()
|
|
for f, v := range w.Files[pathName] {
|
|
if strings.Contains(fileName, f) {
|
|
return v, nil
|
|
}
|
|
}
|
|
return nil, errors.New("no files found")
|
|
}
|
|
|
|
// InitializeKeymanager --
|
|
func (_ *Wallet) InitializeKeymanager(_ context.Context, _ iface.InitKeymanagerConfig) (keymanager.IKeymanager, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
type MockValidator struct {
|
|
Km keymanager.IKeymanager
|
|
}
|
|
|
|
func (_ MockValidator) LogSyncCommitteeMessagesSubmitted() {}
|
|
|
|
func (_ MockValidator) Done() {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) WaitForChainStart(_ context.Context) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) WaitForSync(_ context.Context) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) WaitForActivation(_ context.Context, _ chan [][48]byte) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) CanonicalHeadSlot(_ context.Context) (types.Slot, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) NextSlot() <-chan types.Slot {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) SlotDeadline(_ types.Slot) time.Time {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) LogValidatorGainsAndLosses(_ context.Context, _ types.Slot) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) UpdateDuties(_ context.Context, _ types.Slot) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) RolesAt(_ context.Context, _ types.Slot) (map[[48]byte][]iface2.ValidatorRole, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) SubmitAttestation(_ context.Context, _ types.Slot, _ [48]byte) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) ProposeBlock(_ context.Context, _ types.Slot, _ [48]byte) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) SubmitAggregateAndProof(_ context.Context, _ types.Slot, _ [48]byte) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) SubmitSyncCommitteeMessage(_ context.Context, _ types.Slot, _ [48]byte) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) SubmitSignedContributionAndProof(_ context.Context, _ types.Slot, _ [48]byte) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) LogAttestationsSubmitted() {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) LogNextDutyTimeLeft(_ types.Slot) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) UpdateDomainDataCaches(_ context.Context, _ types.Slot) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) WaitForKeymanagerInitialization(_ context.Context) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) AllValidatorsAreExited(_ context.Context) (bool, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (m MockValidator) Keymanager() (keymanager.IKeymanager, error) {
|
|
return m.Km, nil
|
|
}
|
|
|
|
func (_ MockValidator) ReceiveBlocks(_ context.Context, _ chan<- error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) HandleKeyReload(_ context.Context, _ [][48]byte) (bool, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (_ MockValidator) CheckDoppelGanger(_ context.Context) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
// PushProposerSettings for mocking
|
|
func (_ MockValidator) PushProposerSettings(_ context.Context, _ keymanager.IKeymanager) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
// SetPubKeyToValidatorIndexMap for mocking
|
|
func (_ MockValidator) SetPubKeyToValidatorIndexMap(_ context.Context, _ keymanager.IKeymanager) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
// SignValidatorRegistrationRequest for mocking
|
|
func (_ MockValidator) SignValidatorRegistrationRequest(_ context.Context, _ iface2.SigningFunc, _ *ethpb.ValidatorRegistrationV1) (*ethpb.SignedValidatorRegistrationV1, error) {
|
|
panic("implement me")
|
|
}
|