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.
121 lines
3.6 KiB
121 lines
3.6 KiB
package consensus
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/harmony-one/harmony/crypto/bls"
|
|
|
|
bls_core "github.com/harmony-one/bls/ffi/go/bls"
|
|
harmony_bls "github.com/harmony-one/harmony/crypto/bls"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBasicViewChanging(t *testing.T) {
|
|
_, _, consensus, _, err := GenerateConsensusForTesting()
|
|
assert.NoError(t, err)
|
|
|
|
state := State{mode: Normal}
|
|
|
|
// Change Mode
|
|
assert.Equal(t, state.mode, consensus.current.mode)
|
|
assert.Equal(t, state.Mode(), consensus.current.Mode())
|
|
|
|
consensus.current.SetMode(ViewChanging)
|
|
assert.Equal(t, ViewChanging, consensus.current.mode)
|
|
assert.Equal(t, ViewChanging, consensus.current.Mode())
|
|
|
|
// Change ViewID
|
|
assert.Equal(t, state.GetViewChangingID(), consensus.current.GetViewChangingID())
|
|
|
|
newViewID := consensus.current.GetViewChangingID() + 1
|
|
consensus.SetViewIDs(newViewID)
|
|
assert.Equal(t, newViewID, consensus.current.GetViewChangingID())
|
|
}
|
|
|
|
func TestPhaseSwitching(t *testing.T) {
|
|
type phaseSwitch struct {
|
|
start FBFTPhase
|
|
end FBFTPhase
|
|
}
|
|
|
|
phases := []FBFTPhase{FBFTAnnounce, FBFTPrepare, FBFTCommit}
|
|
|
|
_, _, consensus, _, err := GenerateConsensusForTesting()
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, FBFTAnnounce, consensus.phase) // It's a new consensus, we should be at the FBFTAnnounce phase
|
|
|
|
switches := []phaseSwitch{
|
|
{start: FBFTAnnounce, end: FBFTPrepare},
|
|
{start: FBFTPrepare, end: FBFTCommit},
|
|
{start: FBFTCommit, end: FBFTAnnounce},
|
|
}
|
|
|
|
for _, sw := range switches {
|
|
testPhaseGroupSwitching(t, consensus, phases, sw.start, sw.end)
|
|
}
|
|
|
|
for _, sw := range switches {
|
|
testPhaseGroupSwitching(t, consensus, phases, sw.start, sw.end)
|
|
}
|
|
|
|
switches = []phaseSwitch{
|
|
{start: FBFTAnnounce, end: FBFTCommit},
|
|
{start: FBFTPrepare, end: FBFTAnnounce},
|
|
{start: FBFTCommit, end: FBFTPrepare},
|
|
}
|
|
|
|
for _, sw := range switches {
|
|
testPhaseGroupSwitching(t, consensus, phases, sw.start, sw.end)
|
|
}
|
|
}
|
|
|
|
func testPhaseGroupSwitching(t *testing.T, consensus *Consensus, phases []FBFTPhase, startPhase FBFTPhase, desiredPhase FBFTPhase) {
|
|
for range phases {
|
|
consensus.switchPhase("test", desiredPhase)
|
|
assert.Equal(t, desiredPhase, consensus.phase)
|
|
}
|
|
|
|
assert.Equal(t, desiredPhase, consensus.phase)
|
|
|
|
return
|
|
}
|
|
|
|
func TestGetNextLeaderKeyShouldFailForStandardGeneratedConsensus(t *testing.T) {
|
|
_, _, consensus, _, err := GenerateConsensusForTesting()
|
|
assert.NoError(t, err)
|
|
|
|
// The below results in: "panic: runtime error: integer divide by zero"
|
|
// This happens because there's no check for if there are any participants or not in https://github.com/harmony-one/harmony/blob/main/consensus/quorum/quorum.go#L188-L197
|
|
assert.Panics(t, func() { consensus.GetNextLeaderKey(uint64(1)) })
|
|
}
|
|
|
|
func TestGetNextLeaderKeyShouldSucceed(t *testing.T) {
|
|
_, _, consensus, _, err := GenerateConsensusForTesting()
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, int64(0), consensus.Decider.ParticipantsCount())
|
|
|
|
blsKeys := []*bls_core.PublicKey{}
|
|
wrappedBLSKeys := []bls.PublicKeyWrapper{}
|
|
|
|
keyCount := int64(5)
|
|
for i := int64(0); i < keyCount; i++ {
|
|
blsKey := harmony_bls.RandPrivateKey()
|
|
blsPubKey := blsKey.GetPublicKey()
|
|
bytes := bls.SerializedPublicKey{}
|
|
bytes.FromLibBLSPublicKey(blsPubKey)
|
|
wrapped := bls.PublicKeyWrapper{Object: blsPubKey, Bytes: bytes}
|
|
|
|
blsKeys = append(blsKeys, blsPubKey)
|
|
wrappedBLSKeys = append(wrappedBLSKeys, wrapped)
|
|
}
|
|
|
|
consensus.Decider.UpdateParticipants(wrappedBLSKeys)
|
|
assert.Equal(t, keyCount, consensus.Decider.ParticipantsCount())
|
|
|
|
consensus.LeaderPubKey = &wrappedBLSKeys[0]
|
|
nextKey := consensus.GetNextLeaderKey(uint64(1))
|
|
|
|
assert.Equal(t, nextKey, &wrappedBLSKeys[1])
|
|
}
|
|
|