mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-06-22 06:34:32 +00:00
49cecaa9af
* fix bug in isEpochSwitchAtRound, fix penalty test TestHookPenaltyV2Mining * fix authorised test * fix things * revert a test
84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
package engine_v2
|
|
|
|
import (
|
|
"github.com/XinFinOrg/XDPoSChain/common"
|
|
"github.com/XinFinOrg/XDPoSChain/consensus"
|
|
"github.com/XinFinOrg/XDPoSChain/consensus/XDPoS/utils"
|
|
)
|
|
|
|
/*
|
|
Testing tools
|
|
*/
|
|
|
|
func (x *XDPoS_v2) SetNewRoundFaker(blockChainReader consensus.ChainReader, newRound utils.Round, resetTimer bool) {
|
|
x.lock.Lock()
|
|
defer x.lock.Unlock()
|
|
// Reset a bunch of things
|
|
if resetTimer {
|
|
x.timeoutWorker.Reset(blockChainReader)
|
|
}
|
|
x.currentRound = newRound
|
|
}
|
|
|
|
// for test only
|
|
func (x *XDPoS_v2) ProcessQCFaker(chain consensus.ChainReader, qc *utils.QuorumCert) error {
|
|
x.lock.Lock()
|
|
defer x.lock.Unlock()
|
|
return x.processQC(chain, qc)
|
|
}
|
|
|
|
// Utils for test to check currentRound value
|
|
func (x *XDPoS_v2) GetCurrentRoundFaker() utils.Round {
|
|
x.lock.RLock()
|
|
defer x.lock.RUnlock()
|
|
return x.currentRound
|
|
}
|
|
|
|
// Utils for test to get current Pool size
|
|
func (x *XDPoS_v2) GetVotePoolSizeFaker(vote *utils.Vote) int {
|
|
return x.votePool.Size(vote)
|
|
}
|
|
|
|
// Utils for test to get Timeout Pool Size
|
|
func (x *XDPoS_v2) GetTimeoutPoolSizeFaker(timeout *utils.Timeout) int {
|
|
return x.timeoutPool.Size(timeout)
|
|
}
|
|
|
|
// WARN: This function is designed for testing purpose only!
|
|
// Utils for test to check currentRound values
|
|
func (x *XDPoS_v2) GetPropertiesFaker() (utils.Round, *utils.QuorumCert, *utils.QuorumCert, *utils.TimeoutCert, utils.Round, *utils.BlockInfo) {
|
|
x.lock.RLock()
|
|
defer x.lock.RUnlock()
|
|
return x.currentRound, x.lockQuorumCert, x.highestQuorumCert, x.highestTimeoutCert, x.highestVotedRound, x.highestCommitBlock
|
|
}
|
|
|
|
// WARN: This function is designed for testing purpose only!
|
|
// Utils for tests to set engine specific values
|
|
func (x *XDPoS_v2) SetPropertiesFaker(highestQC *utils.QuorumCert, highestTC *utils.TimeoutCert) {
|
|
x.highestQuorumCert = highestQC
|
|
x.highestTimeoutCert = highestTC
|
|
}
|
|
|
|
func (x *XDPoS_v2) HygieneVotePoolFaker() {
|
|
x.hygieneVotePool()
|
|
}
|
|
|
|
func (x *XDPoS_v2) GetVotePoolKeyListFaker() []string {
|
|
return x.votePool.PoolObjKeysList()
|
|
}
|
|
|
|
func (x *XDPoS_v2) HygieneTimeoutPoolFaker() {
|
|
x.hygieneTimeoutPool()
|
|
}
|
|
|
|
func (x *XDPoS_v2) GetTimeoutPoolKeyListFaker() []string {
|
|
return x.timeoutPool.PoolObjKeysList()
|
|
}
|
|
|
|
// Fake the signer address, the signing function is incompatible
|
|
func (x *XDPoS_v2) AuthorizeFaker(signer common.Address) {
|
|
x.signLock.Lock()
|
|
defer x.signLock.Unlock()
|
|
|
|
x.signer = signer
|
|
}
|