package engine_v2

import (
	"github.com/XinFinOrg/XDPoSChain/common"
	"github.com/XinFinOrg/XDPoSChain/consensus"
	"github.com/XinFinOrg/XDPoSChain/core/types"
)

/*
	Testing tools
*/

func (x *XDPoS_v2) SetNewRoundFaker(blockChainReader consensus.ChainReader, newRound types.Round, resetTimer bool) {
	x.lock.Lock()
	defer x.lock.Unlock()
	// Reset a bunch of things
	if resetTimer {
		x.timeoutWorker.Reset(blockChainReader, 0, 0)
	}
	x.currentRound = newRound
}

// for test only
func (x *XDPoS_v2) ProcessQCFaker(chain consensus.ChainReader, qc *types.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() types.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 *types.Vote) int {
	return x.votePool.Size(vote.PoolKey())
}

// Utils for test to get Timeout Pool Size
func (x *XDPoS_v2) GetTimeoutPoolSizeFaker(timeout *types.Timeout) int {
	return x.timeoutPool.Size(timeout.PoolKey())
}

// WARN: This function is designed for testing purpose only!
// Utils for test to check currentRound values
func (x *XDPoS_v2) GetPropertiesFaker() (types.Round, *types.QuorumCert, *types.QuorumCert, *types.TimeoutCert, types.Round, *types.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 *types.QuorumCert, highestTC *types.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
}

func (x *XDPoS_v2) GetForensicsFaker() *Forensics {
	return x.ForensicsProcessor
}

// WARN: This function is designed for testing purpose only!
func (x *XDPoS_v2) GetTCEpochInfoFaker(chain consensus.ChainReader, timeoutRound types.Round) (*types.EpochSwitchInfo, error) {
	return x.getTCEpochInfo(chain, timeoutRound)
}