mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-04 00:44:27 +00:00
f61f02e59b
* cmd tests * deposit util tests * feature config tests * hashutil tests * htr util tests * interop tests * ip util tests * Update BLS * Update cmd * Update bytesutil and depositutil * Update event * Update keystore * Update mathutil * Update mputil * Update pagination * Update params * Upate prome * Update testutil * Update trieutil * Merge branch 'master' of github.com:prysmaticlabs/prysm into testutil-shared * Sync with master * Mod * Typo * Revert * gazelle * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Gaz * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * fixes build error * Merge refs/heads/master into testutil-shared
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package keystore
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/pborman/uuid"
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestMarshalAndUnmarshal(t *testing.T) {
|
|
testID := uuid.NewRandom()
|
|
blsKey := bls.RandKey()
|
|
|
|
key := &Key{
|
|
ID: testID,
|
|
SecretKey: blsKey,
|
|
PublicKey: blsKey.PublicKey(),
|
|
}
|
|
marshalledObject, err := key.MarshalJSON()
|
|
require.NoError(t, err)
|
|
newKey := &Key{
|
|
ID: []byte{},
|
|
SecretKey: blsKey,
|
|
PublicKey: blsKey.PublicKey(),
|
|
}
|
|
|
|
err = newKey.UnmarshalJSON(marshalledObject)
|
|
require.NoError(t, err)
|
|
require.Equal(t, true, bytes.Equal(newKey.ID, testID))
|
|
}
|
|
|
|
func TestStoreRandomKey(t *testing.T) {
|
|
tempDir, teardown := setupTempKeystoreDir(t)
|
|
defer teardown()
|
|
ks := &Store{
|
|
keysDirPath: tempDir,
|
|
scryptN: LightScryptN,
|
|
scryptP: LightScryptP,
|
|
}
|
|
require.NoError(t, storeNewRandomKey(ks, "password"))
|
|
}
|
|
|
|
func TestNewKeyFromBLS(t *testing.T) {
|
|
b := []byte("hi")
|
|
b32 := bytesutil.ToBytes32(b)
|
|
blskey, err := bls.SecretKeyFromBytes(b32[:])
|
|
require.NoError(t, err)
|
|
key, err := NewKeyFromBLS(blskey)
|
|
require.NoError(t, err)
|
|
|
|
expected := blskey.Marshal()
|
|
require.Equal(t, true, bytes.Equal(expected, key.SecretKey.Marshal()))
|
|
_, err = NewKey()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestWriteFile(t *testing.T) {
|
|
tempDir, teardown := setupTempKeystoreDir(t)
|
|
defer teardown()
|
|
|
|
testKeystore := []byte{'t', 'e', 's', 't'}
|
|
|
|
err := writeKeyFile(tempDir, testKeystore)
|
|
require.NoError(t, err)
|
|
|
|
keystore, err := ioutil.ReadFile(tempDir)
|
|
require.NoError(t, err)
|
|
require.Equal(t, true, bytes.Equal(keystore, testKeystore))
|
|
}
|