prysm-pulse/validator/keymanager/direct_interop_test.go

73 lines
2.1 KiB
Go
Raw Normal View History

package keymanager_test
import (
"bytes"
"encoding/hex"
"testing"
"github.com/prysmaticlabs/prysm/shared/bls"
"github.com/prysmaticlabs/prysm/validator/keymanager"
)
func TestInteropListValidatingKeysZero(t *testing.T) {
_, err := keymanager.NewInterop(0, 0)
if err == nil {
t.Fatal("Missing expected error")
}
if err.Error() != "failed to generate keys: input length must be greater than 0" {
t.Errorf("Incorrect value for error; expected \"failed to generate keys: input length must be greater than 0\", received %d", err)
}
}
func TestInteropListValidatingKeysSingle(t *testing.T) {
direct, err := keymanager.NewInterop(1, 0)
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
keys, err := direct.FetchValidatingKeys()
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
if len(keys) != 1 {
t.Errorf("Incorrect number of keys returned; expected 1, received %d", len(keys))
}
pkBytes, err := hex.DecodeString("25295f0d1d592a90b333e26e85149708208e9f8e8bc18f6c77bd62f8ad7a6866")
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
privateKey, err := bls.SecretKeyFromBytes(pkBytes)
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
if !bytes.Equal(privateKey.PublicKey().Marshal(), keys[0][:]) {
t.Fatalf("Public k 0 incorrect; expected %x, received %x", privateKey.PublicKey().Marshal(), keys[0])
}
}
func TestInteropListValidatingKeysOffset(t *testing.T) {
direct, err := keymanager.NewInterop(1, 9)
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
keys, err := direct.FetchValidatingKeys()
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
if len(keys) != 1 {
t.Errorf("Incorrect number of keys returned; expected 1, received %d", len(keys))
}
pkBytes, err := hex.DecodeString("2b3b88a041168a1c4cd04bdd8de7964fd35238f95442dc678514f9dadb81ec34")
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
privateKey, err := bls.SecretKeyFromBytes(pkBytes)
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
if !bytes.Equal(privateKey.PublicKey().Marshal(), keys[0][:]) {
t.Fatalf("Public k 0 incorrect; expected %x, received %x", privateKey.PublicKey().Marshal(), keys[0])
}
}