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.
116 lines
3.4 KiB
116 lines
3.4 KiB
package consensus
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
|
|
"github.com/harmony-one/harmony/crypto/bls"
|
|
common2 "github.com/harmony-one/harmony/internal/common"
|
|
|
|
msg_pb "github.com/harmony-one/harmony/api/proto/message"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
"github.com/harmony-one/harmony/p2p"
|
|
"github.com/harmony-one/harmony/p2p/p2pimpl"
|
|
)
|
|
|
|
func TestGetPeerFromID(t *testing.T) {
|
|
leaderPriKey := bls.RandPrivateKey()
|
|
leaderPubKey := leaderPriKey.GetPublicKey()
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902", ConsensusPubKey: leaderPubKey}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
t.Fatalf("newhost failure: %v", err)
|
|
}
|
|
consensus, err := New(host, 0, leader, leaderPriKey)
|
|
if err != nil {
|
|
t.Fatalf("Cannot craeate consensus: %v", err)
|
|
}
|
|
leaderAddress := utils.GetAddressFromBlsPubKey(leader.ConsensusPubKey)
|
|
l := consensus.GetPeerByAddress(common2.MustAddressToBech32(leaderAddress))
|
|
if l.IP != leader.IP || l.Port != leader.Port {
|
|
t.Errorf("leader IP not equal")
|
|
}
|
|
}
|
|
|
|
func TestPopulateMessageFields(t *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
t.Fatalf("newhost failure: %v", err)
|
|
}
|
|
blsPriKey := bls.RandPrivateKey()
|
|
consensus, err := New(host, 0, leader, blsPriKey)
|
|
if err != nil {
|
|
t.Fatalf("Cannot craeate consensus: %v", err)
|
|
}
|
|
consensus.viewID = 2
|
|
blockHash := [32]byte{}
|
|
consensus.blockHash = blockHash
|
|
|
|
msg := &msg_pb.Message{
|
|
Request: &msg_pb.Message_Consensus{
|
|
Consensus: &msg_pb.ConsensusRequest{},
|
|
},
|
|
}
|
|
consensusMsg := msg.GetConsensus()
|
|
consensus.populateMessageFields(consensusMsg)
|
|
|
|
if consensusMsg.ViewId != 2 {
|
|
t.Errorf("Consensus ID is not populated correctly")
|
|
}
|
|
if !bytes.Equal(consensusMsg.BlockHash[:], blockHash[:]) {
|
|
t.Errorf("Block hash is not populated correctly")
|
|
}
|
|
if bytes.Compare(consensusMsg.SenderPubkey, blsPriKey.GetPublicKey().Serialize()) != 0 {
|
|
t.Errorf("Sender ID is not populated correctly")
|
|
}
|
|
}
|
|
|
|
func TestSignAndMarshalConsensusMessage(t *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
t.Fatalf("newhost failure: %v", err)
|
|
}
|
|
consensus, err := New(host, 0, leader, bls.RandPrivateKey())
|
|
if err != nil {
|
|
t.Fatalf("Cannot craeate consensus: %v", err)
|
|
}
|
|
consensus.viewID = 2
|
|
consensus.blockHash = [32]byte{}
|
|
consensus.SelfAddress = common.Address{}
|
|
|
|
msg := &msg_pb.Message{}
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(msg)
|
|
|
|
if err != nil || len(marshaledMessage) == 0 {
|
|
t.Errorf("Failed to sign and marshal the message: %s", err)
|
|
}
|
|
if len(msg.Signature) == 0 {
|
|
t.Error("No signature is signed on the consensus message.")
|
|
}
|
|
}
|
|
|
|
func TestSetViewID(t *testing.T) {
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9902"}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
t.Fatalf("newhost failure: %v", err)
|
|
}
|
|
consensus, err := New(host, 0, leader, bls.RandPrivateKey())
|
|
if err != nil {
|
|
t.Fatalf("Cannot craeate consensus: %v", err)
|
|
}
|
|
|
|
height := uint32(1000)
|
|
consensus.SetViewID(height)
|
|
if consensus.viewID != height {
|
|
t.Errorf("Cannot set consensus ID. Got: %v, Expected: %v", consensus.viewID, height)
|
|
}
|
|
}
|
|
|