mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-06-21 14:14:30 +00:00
a4b362ae9a
* refactor initial and introduce getExtraField function * add test for initial * refactor snapshot * initial first snapshot only
123 lines
4 KiB
Go
123 lines
4 KiB
Go
package tests
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/XinFinOrg/XDPoSChain/common"
|
|
"github.com/XinFinOrg/XDPoSChain/consensus/XDPoS"
|
|
"github.com/XinFinOrg/XDPoSChain/consensus/XDPoS/utils"
|
|
"github.com/XinFinOrg/XDPoSChain/core/types"
|
|
"github.com/XinFinOrg/XDPoSChain/params"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestInitialFirstV2Blcok(t *testing.T) {
|
|
blockchain, _, currentBlock, _, _, _ := PrepareXDCTestBlockChainForV2Engine(t, 900, params.TestXDPoSMockChainConfig, 0)
|
|
adaptor := blockchain.Engine().(*XDPoS.XDPoS)
|
|
header := currentBlock.Header()
|
|
|
|
// snapshot should not be created before initial
|
|
snap, _ := adaptor.EngineV2.GetSnapshot(blockchain, currentBlock.Header())
|
|
assert.Nil(t, snap)
|
|
|
|
err := adaptor.EngineV2.Initial(blockchain, header)
|
|
assert.Nil(t, err)
|
|
|
|
round, _, highQC, _, _ := adaptor.EngineV2.GetProperties()
|
|
blockInfo := &utils.BlockInfo{
|
|
Hash: header.Hash(),
|
|
Round: utils.Round(0),
|
|
Number: header.Number,
|
|
}
|
|
expectedQuorumCert := &utils.QuorumCert{
|
|
ProposedBlockInfo: blockInfo,
|
|
Signatures: nil,
|
|
}
|
|
assert.Equal(t, utils.Round(1), round)
|
|
assert.Equal(t, expectedQuorumCert, highQC)
|
|
|
|
// Test snapshot
|
|
snap, err = adaptor.EngineV2.GetSnapshot(blockchain, currentBlock.Header())
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, uint64(450), snap.Number)
|
|
|
|
// Test Running channels
|
|
WaitPeriod := <-adaptor.WaitPeriodCh
|
|
assert.Equal(t, params.TestXDPoSMockChainConfig.XDPoS.V2.WaitPeriod, WaitPeriod)
|
|
|
|
t.Logf("Waiting %d secs for timeout to happen", params.TestXDPoSMockChainConfig.XDPoS.V2.TimeoutPeriod)
|
|
timeoutMsg := <-adaptor.EngineV2.BroadcastCh
|
|
assert.NotNil(t, timeoutMsg)
|
|
assert.Equal(t, utils.Round(1), timeoutMsg.(*utils.Timeout).Round)
|
|
}
|
|
|
|
func TestInitialOtherV2Block(t *testing.T) {
|
|
// insert new block with new extra fields
|
|
blockchain, _, currentBlock, signer, signFn, _ := PrepareXDCTestBlockChainForV2Engine(t, 900, params.TestXDPoSMockChainConfig, 0)
|
|
adaptor := blockchain.Engine().(*XDPoS.XDPoS)
|
|
|
|
blockCoinBase := "0x111000000000000000000000000000000123"
|
|
for blockNum := 901; blockNum <= 910; blockNum++ {
|
|
currentBlock = CreateBlock(blockchain, params.TestXDPoSMockChainConfig, currentBlock, blockNum, int64(blockNum-900), blockCoinBase, signer, signFn, nil)
|
|
err := blockchain.InsertBlock(currentBlock)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
// v2
|
|
blockInfo := &utils.BlockInfo{
|
|
Hash: currentBlock.Header().Hash(),
|
|
Round: utils.Round(10),
|
|
Number: big.NewInt(910),
|
|
}
|
|
quorumCert := &utils.QuorumCert{
|
|
ProposedBlockInfo: blockInfo,
|
|
Signatures: nil, // after decode it got default value []utils.Signature{}
|
|
}
|
|
extra := utils.ExtraFields_v2{
|
|
Round: 11,
|
|
QuorumCert: quorumCert,
|
|
}
|
|
extraBytes, err := extra.EncodeToBytes()
|
|
assert.Nil(t, err)
|
|
|
|
header := &types.Header{
|
|
Root: common.HexToHash("ea465415b60d88429f181fec9fae67c0f19cbf5a4fa10971d96d4faa57d96ffa"),
|
|
Number: big.NewInt(int64(911)),
|
|
ParentHash: currentBlock.Hash(),
|
|
Coinbase: common.HexToAddress("0x111000000000000000000000000000000123"),
|
|
}
|
|
header.Extra = extraBytes
|
|
|
|
block, err := createBlockFromHeader(blockchain, header, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
blockchain.InsertBlock(block)
|
|
// Initialise
|
|
err = adaptor.EngineV2.Initial(blockchain, block.Header())
|
|
assert.Nil(t, err)
|
|
|
|
round, _, highQC, _, _ := adaptor.EngineV2.GetProperties()
|
|
expectedQuorumCert := &utils.QuorumCert{
|
|
ProposedBlockInfo: blockInfo,
|
|
Signatures: []utils.Signature{},
|
|
}
|
|
assert.Equal(t, utils.Round(11), round)
|
|
assert.Equal(t, expectedQuorumCert, highQC)
|
|
|
|
// Test snapshot
|
|
snap, err := adaptor.EngineV2.GetSnapshot(blockchain, block.Header())
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, uint64(450), snap.Number)
|
|
}
|
|
|
|
func TestSnapshotShouldAlreadyCreatedByUpdateM1(t *testing.T) {
|
|
// insert new block with new extra fields
|
|
blockchain, _, currentBlock, _, _, _ := PrepareXDCTestBlockChainForV2Engine(t, 1800, params.TestXDPoSMockChainConfig, 0)
|
|
adaptor := blockchain.Engine().(*XDPoS.XDPoS)
|
|
|
|
snap, err := adaptor.EngineV2.GetSnapshot(blockchain, currentBlock.Header())
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, uint64(1350), snap.Number)
|
|
}
|