mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-10 19:51:20 +00:00
551b03d6e6
* more descriptive password validation error * include change wallet password fixes * balance and jwt improvements * allow for different wallet dirs specified on startup * ensure wallet password is always validated * fix up prysm tests * gaz * test pass * pass balances tests * wrap up fixes * radek feedback * fix up tests * cors fix * add tests for validator status * pass tests * fix n * skip content type test * package level cache and send over feed * package level cache for derived * all tests passing * gofmt Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
package direct
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/event"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
mock "github.com/prysmaticlabs/prysm/validator/accounts/v2/testing"
|
|
)
|
|
|
|
func TestDirectKeymanager_reloadAccountsFromKeystore(t *testing.T) {
|
|
password := "Passw03rdz293**%#2"
|
|
wallet := &mock.Wallet{
|
|
Files: make(map[string]map[string][]byte),
|
|
AccountPasswords: make(map[string]string),
|
|
WalletPassword: password,
|
|
}
|
|
dr := &Keymanager{
|
|
wallet: wallet,
|
|
accountsChangedFeed: new(event.Feed),
|
|
}
|
|
|
|
numAccounts := 20
|
|
privKeys := make([][]byte, numAccounts)
|
|
pubKeys := make([][]byte, numAccounts)
|
|
for i := 0; i < numAccounts; i++ {
|
|
privKey := bls.RandKey()
|
|
privKeys[i] = privKey.Marshal()
|
|
pubKeys[i] = privKey.PublicKey().Marshal()
|
|
}
|
|
|
|
accountsStore, err := dr.createAccountsKeystore(context.Background(), privKeys, pubKeys)
|
|
require.NoError(t, err)
|
|
require.NoError(t, dr.reloadAccountsFromKeystore(accountsStore))
|
|
|
|
// Check that the public keys were added to the public keys cache.
|
|
for i, keyBytes := range pubKeys {
|
|
require.Equal(t, bytesutil.ToBytes48(keyBytes), orderedPublicKeys[i])
|
|
}
|
|
|
|
// Check that the secret keys were added to the secret keys cache.
|
|
lock.RLock()
|
|
defer lock.RUnlock()
|
|
for i, keyBytes := range privKeys {
|
|
privKey, ok := secretKeysCache[bytesutil.ToBytes48(pubKeys[i])]
|
|
require.Equal(t, true, ok)
|
|
require.Equal(t, bytesutil.ToBytes48(keyBytes), bytesutil.ToBytes48(privKey.Marshal()))
|
|
}
|
|
|
|
// Check the key was added to the global accounts store.
|
|
require.Equal(t, numAccounts, len(dr.accountsStore.PublicKeys))
|
|
require.Equal(t, numAccounts, len(dr.accountsStore.PrivateKeys))
|
|
assert.DeepEqual(t, dr.accountsStore.PublicKeys[0], pubKeys[0])
|
|
}
|