mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-20 22:06:42 +00:00

* Pruning headers p2p basic structure * Remove headers-first * Fix consensus tests except TestValidateAndInsertPruningPointWithSideBlocks and TestValidateAndInsertImportedPruningPoint * Add virtual genesis * Implement PruningPointAndItsAnticoneWithMetaData * Start fixing TestValidateAndInsertImportedPruningPoint * Fix TestValidateAndInsertImportedPruningPoint * Fix BlockWindow * Update p2p and gRPC * Fix all tests except TestHandleRelayInvs * Delete TestHandleRelayInvs parts that cover the old IBD flow * Fix lint errors * Add p2p_request_ibd_blocks.go * Clean code * Make MsgBlockWithMetaData implement its own representation * Remove redundant check if highest share block is below the pruning point * Fix TestCheckLockTimeVerifyConditionedByAbsoluteTimeWithWrongLockTime * Fix comments, errors ane names * Fix window size to the real value * Check reindex root after each block at TestUpdateReindexRoot * Remove irrelevant check * Renames and comments * Remove redundant argument from sendGetBlockLocator * Don't delete staging on non-recoverable errors * Renames and comments * Remove redundant code * Commit changes inside ResolveVirtual * Add comment to IsRecoverableError * Remove blocksWithMetaDataGHOSTDAGDataStore * Increase windows pagefile * Move DeleteStagingConsensus outside of defer * Get rid of mustAccepted in receiveBlockWithMetaData * Ban on invalid pruning point * Rename interface_datastructures_daawindowstore.go to interface_datastructures_blocks_with_meta_data_daa_window_store.go * * Change GetVirtualSelectedParentChainFromBlockResponseMessage and VirtualSelectedParentChainChangedNotificationMessage to show only added block hashes * Remove ResolveVirtual * Use externalapi.ConsensusWrapper inside MiningManager * Fix pruningmanager.blockwithmetadata * Set pruning point selected child when importing the pruning point UTXO set * Change virtual genesis hash * replace the selected parent with virtual genesis on removePrunedBlocksFromGHOSTDAGData * Get rid of low hash in block locators * Remove +1 from everywhere we use difficultyAdjustmentWindowSize and increase the default value by one * Add comments about consensus wrapper * Don't use separate staging area when resolving resolveBlockStatus * Fix netsync stability test * Fix checkResolveVirtual * Rename ConsensusWrapper->ConsensusReference * Get rid of blockHeapNode * Add comment to defaultDifficultyAdjustmentWindowSize * Add SelectedChild to DAGTraversalManager * Remove redundant copy * Rename blockWindowHeap->calculateBlockWindowHeap * Move isVirtualGenesisOnlyParent to utils * Change BlockWithMetaData->BlockWithTrustedData * Get rid of maxReasonLength * Split IBD to 100 blocks each time * Fix a bug in calculateBlockWindowHeap * Switch to trusted data when encountering virtual genesis in blockWithTrustedData * Move ConsensusReference to domain * Update ConsensusReference comment * Add comment * Rename shouldNotAddGenesis->skipAddingGenesis
693 lines
30 KiB
Go
693 lines
30 KiB
Go
package miningmanager_test
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domain/consensusreference"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/kaspanet/kaspad/domain/miningmanager/mempool"
|
|
|
|
"github.com/kaspanet/kaspad/domain/consensus"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/testapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/consensushashing"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/constants"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/subnetworks"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/testutils"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/transactionhelper"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/txscript"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/utxo"
|
|
"github.com/kaspanet/kaspad/domain/miningmanager"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// TestValidateAndInsertTransaction verifies that valid transactions were successfully inserted into the mempool.
|
|
func TestValidateAndInsertTransaction(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestValidateAndInsertTransaction")
|
|
if err != nil {
|
|
t.Fatalf("Error setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempool.DefaultConfig(&consensusConfig.Params))
|
|
transactionsToInsert := make([]*externalapi.DomainTransaction, 10)
|
|
for i := range transactionsToInsert {
|
|
transactionsToInsert[i] = createTransactionWithUTXOEntry(t, i)
|
|
_, err = miningManager.ValidateAndInsertTransaction(transactionsToInsert[i], false, true)
|
|
if err != nil {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
}
|
|
// The UTXOEntry was filled manually for those transactions, so the transactions won't be considered orphans.
|
|
// Therefore, all the transactions expected to be contained in the mempool.
|
|
transactionsFromMempool := miningManager.AllTransactions()
|
|
if len(transactionsToInsert) != len(transactionsFromMempool) {
|
|
t.Fatalf("Wrong number of transactions in mempool: expected: %d, got: %d", len(transactionsToInsert), len(transactionsFromMempool))
|
|
}
|
|
for _, transactionToInsert := range transactionsToInsert {
|
|
if !contains(transactionToInsert, transactionsFromMempool) {
|
|
t.Fatalf("Missing transaction %s in the mempool", consensushashing.TransactionID(transactionToInsert))
|
|
}
|
|
}
|
|
|
|
// The parent's transaction was inserted by consensus(AddBlock), and we want to verify that
|
|
// the transaction is not considered an orphan and inserted into the mempool.
|
|
transactionNotAnOrphan, err := createChildAndParentTxsAndAddParentToConsensus(tc)
|
|
if err != nil {
|
|
t.Fatalf("Error in createParentAndChildrenTransaction: %v", err)
|
|
}
|
|
_, err = miningManager.ValidateAndInsertTransaction(transactionNotAnOrphan, false, true)
|
|
if err != nil {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
transactionsFromMempool = miningManager.AllTransactions()
|
|
if !contains(transactionNotAnOrphan, transactionsFromMempool) {
|
|
t.Fatalf("Missing transaction %s in the mempool", consensushashing.TransactionID(transactionNotAnOrphan))
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestImmatureSpend(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestValidateAndInsertTransaction")
|
|
if err != nil {
|
|
t.Fatalf("Error setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempool.DefaultConfig(&consensusConfig.Params))
|
|
tx := createTransactionWithUTXOEntry(t, 0)
|
|
_, err = miningManager.ValidateAndInsertTransaction(tx, false, false)
|
|
txRuleError := &mempool.TxRuleError{}
|
|
if !errors.As(err, txRuleError) || txRuleError.RejectCode != mempool.RejectImmatureSpend {
|
|
t.Fatalf("Unexpected error %+v", err)
|
|
}
|
|
transactionsFromMempool := miningManager.AllTransactions()
|
|
if contains(tx, transactionsFromMempool) {
|
|
t.Fatalf("Mempool contains a transaction with immature coinbase")
|
|
}
|
|
})
|
|
}
|
|
|
|
// TestInsertDoubleTransactionsToMempool verifies that an attempt to insert a transaction
|
|
// more than once into the mempool will result in raising an appropriate error.
|
|
func TestInsertDoubleTransactionsToMempool(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestInsertDoubleTransactionsToMempool")
|
|
if err != nil {
|
|
t.Fatalf("Error setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempool.DefaultConfig(&consensusConfig.Params))
|
|
transaction := createTransactionWithUTXOEntry(t, 0)
|
|
_, err = miningManager.ValidateAndInsertTransaction(transaction, false, true)
|
|
if err != nil {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
_, err = miningManager.ValidateAndInsertTransaction(transaction, false, true)
|
|
if err == nil || !strings.Contains(err.Error(), "is already in the mempool") {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
})
|
|
}
|
|
|
|
// TestDoubleSpendInMempool verifies that an attempt to insert a transaction double-spending
|
|
// another transaction already in the mempool will result in raising an appropriate error.
|
|
func TestDoubleSpendInMempool(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestDoubleSpendInMempool")
|
|
if err != nil {
|
|
t.Fatalf("Error setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempool.DefaultConfig(&consensusConfig.Params))
|
|
transaction, err := createChildAndParentTxsAndAddParentToConsensus(tc)
|
|
if err != nil {
|
|
t.Fatalf("Error creating transaction: %+v", err)
|
|
}
|
|
_, err = miningManager.ValidateAndInsertTransaction(transaction, false, true)
|
|
if err != nil {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
|
|
doubleSpendingTransaction := transaction.Clone()
|
|
doubleSpendingTransaction.ID = nil
|
|
doubleSpendingTransaction.Outputs[0].Value-- // do some minor change so that txID is different
|
|
|
|
_, err = miningManager.ValidateAndInsertTransaction(doubleSpendingTransaction, false, true)
|
|
if err == nil || !strings.Contains(err.Error(), "already spent by transaction") {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
})
|
|
}
|
|
|
|
// TestHandleNewBlockTransactions verifies that all the transactions in the block were successfully removed from the mempool.
|
|
func TestHandleNewBlockTransactions(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestHandleNewBlockTransactions")
|
|
if err != nil {
|
|
t.Fatalf("Error setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempool.DefaultConfig(&consensusConfig.Params))
|
|
transactionsToInsert := make([]*externalapi.DomainTransaction, 10)
|
|
for i := range transactionsToInsert {
|
|
transaction := createTransactionWithUTXOEntry(t, i)
|
|
transactionsToInsert[i] = transaction
|
|
_, err = miningManager.ValidateAndInsertTransaction(transaction, false, true)
|
|
if err != nil {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
}
|
|
|
|
const partialLength = 3
|
|
blockWithFirstPartOfTheTransactions := append([]*externalapi.DomainTransaction{nil}, transactionsToInsert[0:partialLength]...)
|
|
blockWithRestOfTheTransactions := append([]*externalapi.DomainTransaction{nil}, transactionsToInsert[partialLength:]...)
|
|
_, err = miningManager.HandleNewBlockTransactions(blockWithFirstPartOfTheTransactions)
|
|
if err != nil {
|
|
t.Fatalf("HandleNewBlockTransactions: %v", err)
|
|
}
|
|
mempoolTransactions := miningManager.AllTransactions()
|
|
for _, removedTransaction := range blockWithFirstPartOfTheTransactions {
|
|
if contains(removedTransaction, mempoolTransactions) {
|
|
t.Fatalf("This transaction shouldnt be in mempool: %s", consensushashing.TransactionID(removedTransaction))
|
|
}
|
|
}
|
|
|
|
// There are no chained/double-spends transactions, and hence it is expected that all the other
|
|
// transactions, will still be included in the mempool.
|
|
mempoolTransactions = miningManager.AllTransactions()
|
|
for _, transaction := range blockWithRestOfTheTransactions[transactionhelper.CoinbaseTransactionIndex+1:] {
|
|
if !contains(transaction, mempoolTransactions) {
|
|
t.Fatalf("This transaction %s should be in mempool.", consensushashing.TransactionID(transaction))
|
|
}
|
|
}
|
|
// Handle all the other transactions.
|
|
_, err = miningManager.HandleNewBlockTransactions(blockWithRestOfTheTransactions)
|
|
if err != nil {
|
|
t.Fatalf("HandleNewBlockTransactions: %v", err)
|
|
}
|
|
if len(miningManager.AllTransactions()) != 0 {
|
|
blockIDs := domainBlocksToBlockIds(miningManager.AllTransactions())
|
|
t.Fatalf("The mempool contains unexpected transactions: %s", blockIDs)
|
|
}
|
|
})
|
|
}
|
|
|
|
func domainBlocksToBlockIds(blocks []*externalapi.DomainTransaction) []*externalapi.DomainTransactionID {
|
|
blockIDs := make([]*externalapi.DomainTransactionID, len(blocks))
|
|
for i := range blockIDs {
|
|
blockIDs[i] = consensushashing.TransactionID(blocks[i])
|
|
}
|
|
return blockIDs
|
|
}
|
|
|
|
// TestDoubleSpendWithBlock verifies that any transactions which are now double spends as a result of the block's new transactions
|
|
// will be removed from the mempool.
|
|
func TestDoubleSpendWithBlock(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestDoubleSpendWithBlock")
|
|
if err != nil {
|
|
t.Fatalf("Failed setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempool.DefaultConfig(&consensusConfig.Params))
|
|
transactionInTheMempool := createTransactionWithUTXOEntry(t, 0)
|
|
_, err = miningManager.ValidateAndInsertTransaction(transactionInTheMempool, false, true)
|
|
if err != nil {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
doubleSpendTransactionInTheBlock := createTransactionWithUTXOEntry(t, 0)
|
|
doubleSpendTransactionInTheBlock.Inputs[0].PreviousOutpoint = transactionInTheMempool.Inputs[0].PreviousOutpoint
|
|
blockTransactions := []*externalapi.DomainTransaction{nil, doubleSpendTransactionInTheBlock}
|
|
_, err = miningManager.HandleNewBlockTransactions(blockTransactions)
|
|
if err != nil {
|
|
t.Fatalf("HandleNewBlockTransactions: %v", err)
|
|
}
|
|
if contains(transactionInTheMempool, miningManager.AllTransactions()) {
|
|
t.Fatalf("The transaction %s, shouldn't be in the mempool, since at least one "+
|
|
"output was already spent.", consensushashing.TransactionID(transactionInTheMempool))
|
|
}
|
|
})
|
|
}
|
|
|
|
// TestOrphanTransactions verifies that a transaction could be a part of a new block template, only if it's not an orphan.
|
|
func TestOrphanTransactions(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestOrphanTransactions")
|
|
if err != nil {
|
|
t.Fatalf("Error setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempool.DefaultConfig(&consensusConfig.Params))
|
|
// Before each parent transaction, We will add two blocks by consensus in order to fund the parent transactions.
|
|
parentTransactions, childTransactions, err := createArraysOfParentAndChildrenTransactions(tc)
|
|
if err != nil {
|
|
t.Fatalf("Error in createArraysOfParentAndChildrenTransactions: %v", err)
|
|
}
|
|
for _, orphanTransaction := range childTransactions {
|
|
_, err = miningManager.ValidateAndInsertTransaction(orphanTransaction, false, true)
|
|
if err != nil {
|
|
t.Fatalf("ValidateAndInsertTransaction: %v", err)
|
|
}
|
|
}
|
|
transactionsMempool := miningManager.AllTransactions()
|
|
for _, transaction := range transactionsMempool {
|
|
if contains(transaction, childTransactions) {
|
|
t.Fatalf("Error: an orphan transaction is exist in the mempool")
|
|
}
|
|
}
|
|
|
|
block, err := miningManager.GetBlockTemplate(&externalapi.DomainCoinbaseData{
|
|
ScriptPublicKey: &externalapi.ScriptPublicKey{Script: nil, Version: 0},
|
|
ExtraData: nil})
|
|
if err != nil {
|
|
t.Fatalf("Failed get a block template: %v", err)
|
|
}
|
|
for _, transactionFromBlock := range block.Transactions[1:] {
|
|
for _, orphanTransaction := range childTransactions {
|
|
if consensushashing.TransactionID(transactionFromBlock) == consensushashing.TransactionID(orphanTransaction) {
|
|
t.Fatalf("Tranasaction with unknown parents is exist in a block that was built from GetTemplate option.")
|
|
}
|
|
}
|
|
}
|
|
tips, err := tc.Tips()
|
|
if err != nil {
|
|
t.Fatalf("Tips: %v.", err)
|
|
}
|
|
blockParentsTransactionsHash, _, err := tc.AddBlock(tips, nil, parentTransactions)
|
|
if err != nil {
|
|
t.Fatalf("AddBlock: %v", err)
|
|
}
|
|
|
|
_, _, err = tc.AddBlock([]*externalapi.DomainHash{blockParentsTransactionsHash}, nil, nil)
|
|
if err != nil {
|
|
t.Fatalf("AddBlock: %v", err)
|
|
}
|
|
|
|
blockParentsTransactions, err := tc.GetBlock(blockParentsTransactionsHash)
|
|
if err != nil {
|
|
t.Fatalf("GetBlock: %v", err)
|
|
}
|
|
_, err = miningManager.HandleNewBlockTransactions(blockParentsTransactions.Transactions)
|
|
if err != nil {
|
|
t.Fatalf("HandleNewBlockTransactions: %+v", err)
|
|
}
|
|
transactionsMempool = miningManager.AllTransactions()
|
|
if len(transactionsMempool) != len(childTransactions) {
|
|
t.Fatalf("Expected %d transactions in the mempool but got %d", len(childTransactions), len(transactionsMempool))
|
|
}
|
|
|
|
for _, transaction := range transactionsMempool {
|
|
if !contains(transaction, childTransactions) {
|
|
t.Fatalf("Error: the transaction %s, should be in the mempool since its not "+
|
|
"oprhan anymore.", consensushashing.TransactionID(transaction))
|
|
}
|
|
}
|
|
block, err = miningManager.GetBlockTemplate(&externalapi.DomainCoinbaseData{
|
|
ScriptPublicKey: &externalapi.ScriptPublicKey{Script: nil, Version: 0},
|
|
ExtraData: nil})
|
|
if err != nil {
|
|
t.Fatalf("GetBlockTemplate: %v", err)
|
|
}
|
|
for _, transactionFromBlock := range block.Transactions[1:] {
|
|
isContained := false
|
|
for _, childTransaction := range childTransactions {
|
|
if *consensushashing.TransactionID(transactionFromBlock) == *consensushashing.TransactionID(childTransaction) {
|
|
isContained = true
|
|
break
|
|
}
|
|
}
|
|
if !isContained {
|
|
t.Fatalf("Error: Unknown Transaction %s in a block.", consensushashing.TransactionID(transactionFromBlock))
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestHighPriorityTransactions(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestDoubleSpendWithBlock")
|
|
if err != nil {
|
|
t.Fatalf("Failed setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
mempoolConfig := mempool.DefaultConfig(&consensusConfig.Params)
|
|
mempoolConfig.MaximumTransactionCount = 1
|
|
mempoolConfig.MaximumOrphanTransactionCount = 1
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempoolConfig)
|
|
|
|
// Create 3 pairs of transaction parent-and-child pairs: 1 low priority and 2 high priority
|
|
lowPriorityParentTransaction, lowPriorityChildTransaction, err := createParentAndChildrenTransactions(tc)
|
|
if err != nil {
|
|
t.Fatalf("error creating low-priority transaction pair: %+v", err)
|
|
}
|
|
firstHighPriorityParentTransaction, firstHighPriorityChildTransaction, err := createParentAndChildrenTransactions(tc)
|
|
if err != nil {
|
|
t.Fatalf("error creating first high-priority transaction pair: %+v", err)
|
|
}
|
|
secondHighPriorityParentTransaction, secondHighPriorityChildTransaction, err := createParentAndChildrenTransactions(tc)
|
|
if err != nil {
|
|
t.Fatalf("error creating second high-priority transaction pair: %+v", err)
|
|
}
|
|
|
|
// Submit all the children, make sure the 2 highPriority ones remain in the orphan pool
|
|
_, err = miningManager.ValidateAndInsertTransaction(lowPriorityChildTransaction, false, true)
|
|
if err != nil {
|
|
t.Fatalf("error submitting low-priority transaction: %+v", err)
|
|
}
|
|
_, err = miningManager.ValidateAndInsertTransaction(firstHighPriorityChildTransaction, true, true)
|
|
if err != nil {
|
|
t.Fatalf("error submitting first high-priority transaction: %+v", err)
|
|
}
|
|
_, err = miningManager.ValidateAndInsertTransaction(secondHighPriorityChildTransaction, true, true)
|
|
if err != nil {
|
|
t.Fatalf("error submitting second high-priority transaction: %+v", err)
|
|
}
|
|
// There's no API to check what stayed in the orphan pool, but we'll find it out when we begin to unorphan
|
|
|
|
// Submit all the parents.
|
|
// Low priority transaction will only accept the parent, since the child was evicted from orphanPool
|
|
lowPriorityAcceptedTransactions, err :=
|
|
miningManager.ValidateAndInsertTransaction(lowPriorityParentTransaction, false, true)
|
|
if err != nil {
|
|
t.Fatalf("error submitting low-priority transaction: %+v", err)
|
|
}
|
|
expectedLowPriorityAcceptedTransactions := []*externalapi.DomainTransaction{lowPriorityParentTransaction}
|
|
if !reflect.DeepEqual(lowPriorityAcceptedTransactions, expectedLowPriorityAcceptedTransactions) {
|
|
t.Errorf("Expected only lowPriorityParent (%v) to be in lowPriorityAcceptedTransactions, but got %v",
|
|
consensushashing.TransactionIDs(expectedLowPriorityAcceptedTransactions),
|
|
consensushashing.TransactionIDs(lowPriorityAcceptedTransactions))
|
|
}
|
|
|
|
// Both high priority transactions should accept parent and child
|
|
|
|
// Insert firstHighPriorityParentTransaction
|
|
firstHighPriorityAcceptedTransactions, err :=
|
|
miningManager.ValidateAndInsertTransaction(firstHighPriorityParentTransaction, true, true)
|
|
if err != nil {
|
|
t.Fatalf("error submitting first high-priority transaction: %+v", err)
|
|
}
|
|
expectedFirstHighPriorityAcceptedTransactions :=
|
|
[]*externalapi.DomainTransaction{firstHighPriorityParentTransaction, firstHighPriorityChildTransaction}
|
|
if !reflect.DeepEqual(firstHighPriorityAcceptedTransactions, expectedFirstHighPriorityAcceptedTransactions) {
|
|
t.Errorf(
|
|
"Expected both firstHighPriority transaction (%v) to be in firstHighPriorityAcceptedTransactions, but got %v",
|
|
consensushashing.TransactionIDs(firstHighPriorityAcceptedTransactions),
|
|
consensushashing.TransactionIDs(expectedFirstHighPriorityAcceptedTransactions))
|
|
}
|
|
// Insert secondHighPriorityParentTransaction
|
|
secondHighPriorityAcceptedTransactions, err :=
|
|
miningManager.ValidateAndInsertTransaction(secondHighPriorityParentTransaction, true, true)
|
|
if err != nil {
|
|
t.Fatalf("error submitting second high-priority transaction: %+v", err)
|
|
}
|
|
expectedSecondHighPriorityAcceptedTransactions :=
|
|
[]*externalapi.DomainTransaction{secondHighPriorityParentTransaction, secondHighPriorityChildTransaction}
|
|
if !reflect.DeepEqual(secondHighPriorityAcceptedTransactions, expectedSecondHighPriorityAcceptedTransactions) {
|
|
t.Errorf(
|
|
"Expected both secondHighPriority transaction (%v) to be in secondHighPriorityAcceptedTransactions, but got %v",
|
|
consensushashing.TransactionIDs(secondHighPriorityAcceptedTransactions),
|
|
consensushashing.TransactionIDs(expectedSecondHighPriorityAcceptedTransactions))
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestRevalidateHighPriorityTransactions(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
consensusConfig.BlockCoinbaseMaturity = 0
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestRevalidateHighPriorityTransactions")
|
|
if err != nil {
|
|
t.Fatalf("Failed setting up TestConsensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
miningFactory := miningmanager.NewFactory()
|
|
mempoolConfig := mempool.DefaultConfig(&consensusConfig.Params)
|
|
tcAsConsensus := tc.(externalapi.Consensus)
|
|
tcAsConsensusPointer := &tcAsConsensus
|
|
consensusReference := consensusreference.NewConsensusReference(&tcAsConsensusPointer)
|
|
miningManager := miningFactory.NewMiningManager(consensusReference, &consensusConfig.Params, mempoolConfig)
|
|
|
|
// Create two valid transactions that double-spend each other (childTransaction1, childTransaction2)
|
|
parentTransaction, childTransaction1, err := createParentAndChildrenTransactions(tc)
|
|
if err != nil {
|
|
t.Fatalf("Error creating parentTransaction and childTransaction1: %+v", err)
|
|
}
|
|
tips, err := tc.Tips()
|
|
if err != nil {
|
|
t.Fatalf("Error getting tips: %+v", err)
|
|
}
|
|
|
|
fundingBlock, _, err := tc.AddBlock(tips, nil, []*externalapi.DomainTransaction{parentTransaction})
|
|
if err != nil {
|
|
t.Fatalf("Error getting function block: %+v", err)
|
|
}
|
|
|
|
childTransaction2 := childTransaction1.Clone()
|
|
childTransaction2.Outputs[0].Value-- // decrement value to change id
|
|
|
|
// Mine 1 block with confirming childTransaction1 and 2 blocks confirming childTransaction2, so that
|
|
// childTransaction2 is accepted
|
|
tip1, _, err := tc.AddBlock([]*externalapi.DomainHash{fundingBlock}, nil,
|
|
[]*externalapi.DomainTransaction{childTransaction1})
|
|
if err != nil {
|
|
t.Fatalf("Error adding tip1: %+v", err)
|
|
}
|
|
tip2, _, err := tc.AddBlock([]*externalapi.DomainHash{fundingBlock}, nil,
|
|
[]*externalapi.DomainTransaction{childTransaction2})
|
|
if err != nil {
|
|
t.Fatalf("Error adding tip2: %+v", err)
|
|
}
|
|
_, _, err = tc.AddBlock([]*externalapi.DomainHash{tip2}, nil, nil)
|
|
if err != nil {
|
|
t.Fatalf("Error mining on top of tip2: %+v", err)
|
|
}
|
|
|
|
// Add to mempool transaction that spends childTransaction2 (as high priority)
|
|
spendingTransaction, err := testutils.CreateTransaction(childTransaction2, 1000)
|
|
if err != nil {
|
|
t.Fatalf("Error creating spendingTransaction: %+v", err)
|
|
}
|
|
_, err = miningManager.ValidateAndInsertTransaction(spendingTransaction, true, false)
|
|
if err != nil {
|
|
t.Fatalf("Error inserting spendingTransaction: %+v", err)
|
|
}
|
|
|
|
// Revalidate, to make sure spendingTransaction is still valid
|
|
validTransactions, err := miningManager.RevalidateHighPriorityTransactions()
|
|
if err != nil {
|
|
t.Fatalf("Error from first RevalidateHighPriorityTransactions: %+v", err)
|
|
}
|
|
if len(validTransactions) != 1 || !validTransactions[0].Equal(spendingTransaction) {
|
|
t.Fatalf("Expected to have spendingTransaction as only validTransaction returned from "+
|
|
"RevalidateHighPriorityTransactions, but got %v instead", validTransactions)
|
|
}
|
|
|
|
// Mine 2 more blocks on top of tip1, to re-org out childTransaction1, thus making spendingTransaction invalid
|
|
for i := 0; i < 2; i++ {
|
|
tip1, _, err = tc.AddBlock([]*externalapi.DomainHash{tip1}, nil, nil)
|
|
if err != nil {
|
|
t.Fatalf("Error mining on top of tip1: %+v", err)
|
|
}
|
|
}
|
|
|
|
// Make sure spendingTransaction is still in mempool
|
|
allTransactions := miningManager.AllTransactions()
|
|
if len(allTransactions) != 1 || !allTransactions[0].Equal(spendingTransaction) {
|
|
t.Fatalf("Expected to have spendingTransaction as only validTransaction returned from "+
|
|
"RevalidateHighPriorityTransactions, but got %v instead", validTransactions)
|
|
}
|
|
|
|
// Revalidate again, this time validTransactions should be empty
|
|
validTransactions, err = miningManager.RevalidateHighPriorityTransactions()
|
|
if err != nil {
|
|
t.Fatalf("Error from first RevalidateHighPriorityTransactions: %+v", err)
|
|
}
|
|
if len(validTransactions) != 0 {
|
|
t.Fatalf("Expected to have empty validTransactions, but got %v instead", validTransactions)
|
|
}
|
|
// And also AllTransactions should be empty as well
|
|
allTransactions = miningManager.AllTransactions()
|
|
if len(allTransactions) != 0 {
|
|
t.Fatalf("Expected to have empty allTransactions, but got %v instead", allTransactions)
|
|
}
|
|
})
|
|
}
|
|
|
|
func createTransactionWithUTXOEntry(t *testing.T, i int) *externalapi.DomainTransaction {
|
|
prevOutTxID := externalapi.DomainTransactionID{}
|
|
prevOutPoint := externalapi.DomainOutpoint{TransactionID: prevOutTxID, Index: uint32(i)}
|
|
scriptPublicKey, redeemScript := testutils.OpTrueScript()
|
|
signatureScript, err := txscript.PayToScriptHashSignatureScript(redeemScript, nil)
|
|
if err != nil {
|
|
t.Fatalf("PayToScriptHashSignatureScript: %v", err)
|
|
}
|
|
txInput := externalapi.DomainTransactionInput{
|
|
PreviousOutpoint: prevOutPoint,
|
|
SignatureScript: signatureScript,
|
|
Sequence: constants.MaxTxInSequenceNum,
|
|
UTXOEntry: utxo.NewUTXOEntry(
|
|
100000000, // 1 KAS
|
|
scriptPublicKey,
|
|
true,
|
|
uint64(0)),
|
|
}
|
|
txOut := externalapi.DomainTransactionOutput{
|
|
Value: 10000,
|
|
ScriptPublicKey: scriptPublicKey,
|
|
}
|
|
tx := externalapi.DomainTransaction{
|
|
Version: constants.MaxTransactionVersion,
|
|
Inputs: []*externalapi.DomainTransactionInput{&txInput},
|
|
Outputs: []*externalapi.DomainTransactionOutput{&txOut},
|
|
SubnetworkID: subnetworks.SubnetworkIDNative,
|
|
Gas: 0,
|
|
Fee: 289,
|
|
Mass: 1,
|
|
LockTime: 0}
|
|
|
|
return &tx
|
|
}
|
|
|
|
func createArraysOfParentAndChildrenTransactions(tc testapi.TestConsensus) ([]*externalapi.DomainTransaction,
|
|
[]*externalapi.DomainTransaction, error) {
|
|
|
|
const numOfTransactions = 5
|
|
transactions := make([]*externalapi.DomainTransaction, numOfTransactions)
|
|
parentTransactions := make([]*externalapi.DomainTransaction, len(transactions))
|
|
var err error
|
|
for i := range transactions {
|
|
parentTransactions[i], transactions[i], err = createParentAndChildrenTransactions(tc)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
}
|
|
return parentTransactions, transactions, nil
|
|
}
|
|
|
|
func createParentAndChildrenTransactions(tc testapi.TestConsensus) (txParent *externalapi.DomainTransaction,
|
|
txChild *externalapi.DomainTransaction, err error) {
|
|
|
|
// We will add two blocks by consensus before the parent transactions, in order to fund the parent transactions.
|
|
tips, err := tc.Tips()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
_, _, err = tc.AddBlock(tips, nil, nil)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrapf(err, "AddBlock: %v", err)
|
|
}
|
|
|
|
tips, err = tc.Tips()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
fundingBlockHashForParent, _, err := tc.AddBlock(tips, nil, nil)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "AddBlock: ")
|
|
}
|
|
fundingBlockForParent, err := tc.GetBlock(fundingBlockHashForParent)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "GetBlock: ")
|
|
}
|
|
fundingTransactionForParent := fundingBlockForParent.Transactions[transactionhelper.CoinbaseTransactionIndex]
|
|
txParent, err = testutils.CreateTransaction(fundingTransactionForParent, 1000)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
txChild, err = testutils.CreateTransaction(txParent, 1000)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return txParent, txChild, nil
|
|
}
|
|
|
|
func createChildAndParentTxsAndAddParentToConsensus(tc testapi.TestConsensus) (*externalapi.DomainTransaction, error) {
|
|
firstBlockHash, _, err := tc.AddBlock([]*externalapi.DomainHash{tc.DAGParams().GenesisHash}, nil, nil)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "AddBlock: %v", err)
|
|
}
|
|
ParentBlockHash, _, err := tc.AddBlock([]*externalapi.DomainHash{firstBlockHash}, nil, nil)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "AddBlock: ")
|
|
}
|
|
ParentBlock, err := tc.GetBlock(ParentBlockHash)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "GetBlock: ")
|
|
}
|
|
parentTransaction := ParentBlock.Transactions[transactionhelper.CoinbaseTransactionIndex]
|
|
txChild, err := testutils.CreateTransaction(parentTransaction, 1000)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return txChild, nil
|
|
}
|
|
|
|
func contains(transaction *externalapi.DomainTransaction, transactions []*externalapi.DomainTransaction) bool {
|
|
for _, candidateTransaction := range transactions {
|
|
if candidateTransaction.Equal(transaction) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|