mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 07:16:47 +00:00
8500acd86b
* [NOD-1548] Add TestDifficulty and remove PoW check from tests * [NOD-1548] Add TestSkipProofOfWork * [NOD-1548] Remove TestDifficulty
61 lines
1.9 KiB
Go
61 lines
1.9 KiB
Go
package consensus
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/consensusserialization"
|
|
)
|
|
|
|
type testConsensus struct {
|
|
*consensus
|
|
|
|
testBlockBuilder model.TestBlockBuilder
|
|
testReachabilityManager model.TestReachabilityManager
|
|
testConsensusStateManager model.TestConsensusStateManager
|
|
}
|
|
|
|
func (tc *testConsensus) BuildBlockWithParents(parentHashes []*externalapi.DomainHash, coinbaseData *externalapi.DomainCoinbaseData,
|
|
transactions []*externalapi.DomainTransaction) (*externalapi.DomainBlock, error) {
|
|
|
|
// Require write lock because BuildBlockWithParents stages temporary data
|
|
tc.lock.Lock()
|
|
defer tc.lock.Unlock()
|
|
|
|
return tc.testBlockBuilder.BuildBlockWithParents(parentHashes, coinbaseData, transactions)
|
|
}
|
|
|
|
func (tc *testConsensus) AddBlock(parentHashes []*externalapi.DomainHash, coinbaseData *externalapi.DomainCoinbaseData,
|
|
transactions []*externalapi.DomainTransaction) (*externalapi.DomainHash, error) {
|
|
|
|
// Require write lock because BuildBlockWithParents stages temporary data
|
|
tc.lock.Lock()
|
|
defer tc.lock.Unlock()
|
|
|
|
block, err := tc.testBlockBuilder.BuildBlockWithParents(parentHashes, coinbaseData, transactions)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = tc.blockProcessor.ValidateAndInsertBlock(block)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return consensusserialization.BlockHash(block), nil
|
|
}
|
|
|
|
func (tc *testConsensus) DiscardAllStores() {
|
|
tc.AcceptanceDataStore().Discard()
|
|
tc.BlockHeaderStore().Discard()
|
|
tc.BlockRelationStore().Discard()
|
|
tc.BlockStatusStore().Discard()
|
|
tc.BlockStore().Discard()
|
|
tc.ConsensusStateStore().Discard()
|
|
tc.GHOSTDAGDataStore().Discard()
|
|
tc.HeaderTipsStore().Discard()
|
|
tc.MultisetStore().Discard()
|
|
tc.PruningStore().Discard()
|
|
tc.ReachabilityDataStore().Discard()
|
|
tc.UTXODiffStore().Discard()
|
|
}
|