You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
209 lines
7.0 KiB
209 lines
7.0 KiB
package drand
|
|
|
|
import (
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
|
|
bls2 "github.com/harmony-one/harmony/crypto/bls"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/harmony-one/bls/ffi/go/bls"
|
|
msg_pb "github.com/harmony-one/harmony/api/proto/message"
|
|
"github.com/harmony-one/harmony/core/types"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
"github.com/harmony-one/harmony/p2p"
|
|
"github.com/harmony-one/harmony/p2p/p2pimpl"
|
|
)
|
|
|
|
func TestNew(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
|
|
if !dRand.IsLeader {
|
|
test.Error("dRand should belong to a leader")
|
|
}
|
|
}
|
|
|
|
func TestGetValidatorPeers(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902", ConsensusPubKey: bls2.RandPrivateKey().GetPublicKey()}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905", ConsensusPubKey: bls2.RandPrivateKey().GetPublicKey()}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
|
|
if !dRand.IsLeader {
|
|
test.Error("dRand should belong to a leader")
|
|
}
|
|
|
|
countValidatorPeers := len(dRand.GetValidatorPeers())
|
|
|
|
if countValidatorPeers != 2 {
|
|
test.Error("Count of validator peers doesn't match, got", countValidatorPeers)
|
|
}
|
|
}
|
|
|
|
func TestAddPeers(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902", ConsensusPubKey: bls2.RandPrivateKey().GetPublicKey()}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905", ConsensusPubKey: bls2.RandPrivateKey().GetPublicKey()}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
|
|
if !dRand.IsLeader {
|
|
test.Error("dRand should belong to a leader")
|
|
}
|
|
|
|
newPeer := p2p.Peer{IP: "127.0.0.1", Port: "9907"}
|
|
countValidatorPeers := dRand.AddPeers([]*p2p.Peer{&newPeer})
|
|
|
|
if countValidatorPeers != 1 {
|
|
test.Error("Unable to add new peer")
|
|
}
|
|
|
|
if len(dRand.GetValidatorPeers()) != 3 {
|
|
test.Errorf("Number of validators doesn't match, actual count = %d expected count = 3", len(dRand.GetValidatorPeers()))
|
|
}
|
|
}
|
|
|
|
func TestGetValidatorByPeerId(test *testing.T) {
|
|
leaderPriKey := bls2.RandPrivateKey()
|
|
leaderPubKey := bls2.RandPrivateKey().GetPublicKey()
|
|
validatorPubKey := bls2.RandPrivateKey().GetPublicKey()
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902", ConsensusPubKey: leaderPubKey}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905", ConsensusPubKey: validatorPubKey}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, leaderPriKey)
|
|
|
|
if !dRand.IsLeader {
|
|
test.Error("dRand should belong to a leader")
|
|
}
|
|
|
|
validatorAddress := utils.GetAddressFromBlsPubKey(validatorPubKey).Hex()
|
|
|
|
if dRand.getValidatorPeerByAddress(validatorAddress) == nil {
|
|
test.Error("Unable to get validator by Peerid")
|
|
}
|
|
|
|
if dRand.getValidatorPeerByAddress("random address") != nil {
|
|
test.Error("Found validator for absent validatorId")
|
|
}
|
|
}
|
|
|
|
func TestResetState(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
dRand.ResetState()
|
|
}
|
|
|
|
func TestSetLeaderPubKey(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
|
|
_, newPublicKey, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
newPublicKeyBytes, _ := newPublicKey.Bytes()
|
|
if dRand.SetLeaderPubKey(newPublicKeyBytes) == nil {
|
|
test.Error("Failed to Set Public Key of Leader")
|
|
}
|
|
}
|
|
|
|
func TestUpdatePublicKeys(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
|
|
pubKey1 := bls2.RandPrivateKey().GetPublicKey()
|
|
pubKey2 := bls2.RandPrivateKey().GetPublicKey()
|
|
|
|
publicKeys := []*bls.PublicKey{pubKey1, pubKey2}
|
|
|
|
if dRand.UpdatePublicKeys(publicKeys) != 2 {
|
|
test.Error("Count of public keys doesn't match")
|
|
}
|
|
|
|
for index, publicKey := range dRand.PublicKeys {
|
|
if strings.Compare(publicKey.GetHexString(), publicKeys[index].GetHexString()) != 0 {
|
|
test.Error("Public keys not updated succssfully")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestVerifyMessageSig(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
|
|
message := &msg_pb.Message{
|
|
ReceiverType: msg_pb.ReceiverType_VALIDATOR,
|
|
ServiceType: msg_pb.ServiceType_DRAND,
|
|
Type: msg_pb.MessageType_DRAND_INIT,
|
|
Request: &msg_pb.Message_Drand{
|
|
Drand: &msg_pb.DrandRequest{},
|
|
},
|
|
}
|
|
drandMsg := message.GetDrand()
|
|
drandMsg.SenderAddress = dRand.SelfAddress
|
|
drandMsg.BlockHash = dRand.blockHash[:]
|
|
|
|
dRand.signDRandMessage(message)
|
|
|
|
if verifyMessageSig(dRand.pubKey, message) != nil {
|
|
test.Error("Failed to verify the signature")
|
|
}
|
|
}
|
|
|
|
func TestVrf(test *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
validator := p2p.Peer{IP: "127.0.0.1", Port: "9905"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost failure: %v", err)
|
|
}
|
|
dRand := New(host, "0", []p2p.Peer{leader, validator}, leader, nil, true, bls2.RandPrivateKey())
|
|
tx1 := types.NewTransaction(1, common.BytesToAddress([]byte{0x11}), 0, big.NewInt(111), 1111, big.NewInt(11111), []byte{0x11, 0x11, 0x11})
|
|
txs := []*types.Transaction{tx1}
|
|
|
|
block := types.NewBlock(&types.Header{Number: big.NewInt(314)}, txs, nil)
|
|
blockHash := block.Hash()
|
|
|
|
dRand.vrf(blockHash)
|
|
}
|
|
|