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.
119 lines
3.5 KiB
119 lines
3.5 KiB
package consensus
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/harmony-one/harmony/crypto"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
mock_host "github.com/harmony-one/harmony/p2p/host/mock"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/harmony-one/harmony/p2p/p2pimpl"
|
|
|
|
consensus_proto "github.com/harmony-one/harmony/api/consensus"
|
|
"github.com/harmony-one/harmony/p2p"
|
|
)
|
|
|
|
var (
|
|
ip = "127.0.0.1"
|
|
)
|
|
|
|
func TestProcessMessageLeaderCommit(test *testing.T) {
|
|
ctrl := gomock.NewController(test)
|
|
defer ctrl.Finish()
|
|
|
|
leader := p2p.Peer{IP: ip, Port: "7777"}
|
|
_, leader.PubKey = utils.GenKey(leader.IP, leader.Port)
|
|
|
|
validators := make([]p2p.Peer, 3)
|
|
hosts := make([]p2p.Host, 3)
|
|
|
|
for i := 0; i < 3; i++ {
|
|
port := fmt.Sprintf("%d", 7788+i)
|
|
validators[i] = p2p.Peer{IP: ip, Port: port, ValidatorID: i + 1}
|
|
_, validators[i].PubKey = utils.GenKey(validators[i].IP, validators[i].Port)
|
|
}
|
|
|
|
m := mock_host.NewMockHost(ctrl)
|
|
// Asserts that the first and only call to Bar() is passed 99.
|
|
// Anything else will fail.
|
|
m.EXPECT().GetSelfPeer().Return(leader)
|
|
m.EXPECT().SendMessage(gomock.Any(), gomock.Any()).Times(3)
|
|
|
|
consensusLeader := New(m, "0", validators, leader)
|
|
consensusLeader.blockHash = [32]byte{}
|
|
|
|
consensusValidators := make([]*Consensus, 3)
|
|
for i := 0; i < 3; i++ {
|
|
priKey, _, _ := utils.GenKeyP2P(validators[i].IP, validators[i].Port)
|
|
host, err := p2pimpl.NewHost(&validators[i], priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost error: %v", err)
|
|
}
|
|
hosts[i] = host
|
|
|
|
consensusValidators[i] = New(hosts[i], "0", validators, leader)
|
|
consensusValidators[i].blockHash = [32]byte{}
|
|
_, msg := consensusValidators[i].constructCommitMessage(consensus_proto.MessageType_COMMIT)
|
|
consensusLeader.ProcessMessageLeader(msg[1:])
|
|
}
|
|
|
|
assert.Equal(test, ChallengeDone, consensusLeader.state)
|
|
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
|
|
func TestProcessMessageLeaderResponse(test *testing.T) {
|
|
ctrl := gomock.NewController(test)
|
|
defer ctrl.Finish()
|
|
|
|
leader := p2p.Peer{IP: ip, Port: "8889"}
|
|
_, leader.PubKey = utils.GenKey(leader.IP, leader.Port)
|
|
|
|
validators := make([]p2p.Peer, 3)
|
|
hosts := make([]p2p.Host, 3)
|
|
|
|
for i := 0; i < 3; i++ {
|
|
port := fmt.Sprintf("%d", 8788+i)
|
|
validators[i] = p2p.Peer{IP: ip, Port: port, ValidatorID: i + 1}
|
|
_, validators[i].PubKey = utils.GenKey(validators[i].IP, validators[i].Port)
|
|
}
|
|
|
|
m := mock_host.NewMockHost(ctrl)
|
|
// Asserts that the first and only call to Bar() is passed 99.
|
|
// Anything else will fail.
|
|
m.EXPECT().GetSelfPeer().Return(leader)
|
|
m.EXPECT().SendMessage(gomock.Any(), gomock.Any()).Times(6)
|
|
|
|
for i := 0; i < 3; i++ {
|
|
priKey, _, _ := utils.GenKeyP2P(validators[i].IP, validators[i].Port)
|
|
host, err := p2pimpl.NewHost(&validators[i], priKey)
|
|
if err != nil {
|
|
test.Fatalf("newhost error: %v", err)
|
|
}
|
|
hosts[i] = host
|
|
}
|
|
|
|
consensusLeader := New(m, "0", validators, leader)
|
|
consensusLeader.blockHash = [32]byte{}
|
|
|
|
consensusValidators := make([]*Consensus, 3)
|
|
for i := 0; i < 3; i++ {
|
|
consensusValidators[i] = New(hosts[i], "0", validators, leader)
|
|
consensusValidators[i].blockHash = [32]byte{}
|
|
_, msg := consensusValidators[i].constructCommitMessage(consensus_proto.MessageType_COMMIT)
|
|
consensusLeader.ProcessMessageLeader(msg[1:])
|
|
}
|
|
|
|
for i := 0; i < 3; i++ {
|
|
msg := consensusValidators[i].constructResponseMessage(consensus_proto.MessageType_RESPONSE, crypto.Ed25519Curve.Scalar().One())
|
|
consensusLeader.ProcessMessageLeader(msg[1:])
|
|
}
|
|
|
|
assert.Equal(test, CollectiveSigDone, consensusLeader.state)
|
|
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
|