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.
84 lines
2.4 KiB
84 lines
2.4 KiB
6 years ago
|
package consensus
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
protobuf "github.com/golang/protobuf/proto"
|
||
|
|
"github.com/harmony-one/harmony/api/proto"
|
||
|
|
msg_pb "github.com/harmony-one/harmony/api/proto/message"
|
||
|
|
"github.com/harmony-one/harmony/crypto/bls"
|
||
|
|
"github.com/harmony-one/harmony/internal/utils"
|
||
|
|
"github.com/harmony-one/harmony/p2p"
|
||
|
|
"github.com/harmony-one/harmony/p2p/p2pimpl"
|
||
|
|
)
|
||
|
|
|
||
|
|
func constructAnnounceMessage(t *testing.T) []byte {
|
||
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "19999"}
|
||
|
|
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.blockHash = [32]byte{}
|
||
|
|
|
||
|
|
msgBytes := consensus.constructAnnounceMessage()
|
||
|
|
msgPayload, _ := proto.GetConsensusMessagePayload(msgBytes)
|
||
|
|
return msgPayload
|
||
|
|
}
|
||
|
|
|
||
|
|
func getConsensusMessage(payload []byte) (*msg_pb.Message, error) {
|
||
|
|
msg := &msg_pb.Message{}
|
||
|
|
err := protobuf.Unmarshal(payload, msg)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return msg, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParsePbftMessage(t *testing.T) {
|
||
|
|
payload := constructAnnounceMessage(t)
|
||
|
|
msg, err := getConsensusMessage(payload)
|
||
|
|
if err != nil {
|
||
|
|
t.Error("create consensus message error")
|
||
|
|
}
|
||
|
|
_, err = ParsePbftMessage(msg)
|
||
|
|
if err != nil {
|
||
|
|
t.Error("unable to parse PbftMessage")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetMessagesByTypeSeqViewHash(t *testing.T) {
|
||
|
|
pbftMsg := PbftMessage{MessageType: msg_pb.MessageType_ANNOUNCE, SeqNum: 2, ConsensusID: 3, BlockHash: [32]byte{01, 02}}
|
||
|
|
log := NewPbftLog()
|
||
|
|
log.AddMessage(&pbftMsg)
|
||
|
|
|
||
|
|
found := log.GetMessagesByTypeSeqViewHash(msg_pb.MessageType_ANNOUNCE, 2, 3, [32]byte{01, 02})
|
||
|
|
if len(found) != 1 {
|
||
|
|
t.Error("cannot find existing message")
|
||
|
|
}
|
||
|
|
|
||
|
|
notFound := log.GetMessagesByTypeSeqViewHash(msg_pb.MessageType_ANNOUNCE, 2, 3, [32]byte{01, 03})
|
||
|
|
if len(notFound) > 0 {
|
||
|
|
t.Error("find message that not exist")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHasMatchingAnnounce(t *testing.T) {
|
||
|
|
pbftMsg := PbftMessage{MessageType: msg_pb.MessageType_ANNOUNCE, SeqNum: 2, ConsensusID: 3, BlockHash: [32]byte{01, 02}}
|
||
|
|
log := NewPbftLog()
|
||
|
|
log.AddMessage(&pbftMsg)
|
||
|
|
found := log.HasMatchingAnnounce(2, 3, [32]byte{01, 02})
|
||
|
|
if !found {
|
||
|
|
t.Error("found should be true")
|
||
|
|
}
|
||
|
|
|
||
|
|
notFound := log.HasMatchingAnnounce(2, 3, [32]byte{02, 02})
|
||
|
|
if notFound {
|
||
|
|
t.Error("notFound should be false")
|
||
|
|
}
|
||
|
|
}
|