mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-03-30 15:08:33 +00:00
c88fa1492e
* [NOD-375] Move to pkg/errors * [NOD-375] Fix tests * [NOD-375] Make AreErrorsEqual a shared function
97 lines
2.9 KiB
Go
97 lines
2.9 KiB
Go
package testtools
|
|
|
|
import (
|
|
"github.com/daglabs/btcd/dagconfig"
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/daglabs/btcd/mining"
|
|
"github.com/daglabs/btcd/util/daghash"
|
|
|
|
"github.com/daglabs/btcd/blockdag"
|
|
|
|
"github.com/daglabs/btcd/txscript"
|
|
"github.com/daglabs/btcd/util"
|
|
"github.com/daglabs/btcd/util/subnetworkid"
|
|
"github.com/daglabs/btcd/wire"
|
|
)
|
|
|
|
// RegisterSubnetworkForTest is used to register network on DAG with specified gas limit
|
|
func RegisterSubnetworkForTest(dag *blockdag.BlockDAG, params *dagconfig.Params, gasLimit uint64) (*subnetworkid.SubnetworkID, error) {
|
|
buildNextBlock := func(parentHashes []*daghash.Hash, txs []*wire.MsgTx) (*util.Block, error) {
|
|
msgBlock, err := mining.PrepareBlockForTest(dag, params, parentHashes, txs, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return util.NewBlock(msgBlock), nil
|
|
}
|
|
|
|
addBlockToDAG := func(block *util.Block) error {
|
|
isOrphan, delay, err := dag.ProcessBlock(block, blockdag.BFNoPoWCheck)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if delay != 0 {
|
|
return errors.Errorf("ProcessBlock: block is is too far in the future")
|
|
}
|
|
|
|
if isOrphan {
|
|
return errors.Errorf("ProcessBlock: unexpected returned orphan block")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Create a block in order to fund later transactions
|
|
fundsBlock, err := buildNextBlock(dag.TipHashes(), []*wire.MsgTx{})
|
|
if err != nil {
|
|
return nil, errors.Errorf("could not build funds block: %s", err)
|
|
}
|
|
|
|
err = addBlockToDAG(fundsBlock)
|
|
if err != nil {
|
|
return nil, errors.Errorf("could not add funds block to DAG: %s", err)
|
|
}
|
|
|
|
fundsBlockCbTx := fundsBlock.Transactions()[0].MsgTx()
|
|
|
|
// Create a block with a valid subnetwork registry transaction
|
|
signatureScript, err := txscript.PayToScriptHashSignatureScript(blockdag.OpTrueScript, nil)
|
|
if err != nil {
|
|
return nil, errors.Errorf("Failed to build signature script: %s", err)
|
|
}
|
|
txIn := &wire.TxIn{
|
|
PreviousOutpoint: *wire.NewOutpoint(fundsBlockCbTx.TxID(), 0),
|
|
Sequence: wire.MaxTxInSequenceNum,
|
|
SignatureScript: signatureScript,
|
|
}
|
|
|
|
scriptPubKey, err := txscript.PayToScriptHashScript(blockdag.OpTrueScript)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
txOut := &wire.TxOut{
|
|
ScriptPubKey: scriptPubKey,
|
|
Value: fundsBlockCbTx.TxOut[0].Value,
|
|
}
|
|
registryTx := wire.NewRegistryMsgTx(1, []*wire.TxIn{txIn}, []*wire.TxOut{txOut}, gasLimit)
|
|
|
|
// Add it to the DAG
|
|
registryBlock, err := buildNextBlock([]*daghash.Hash{fundsBlock.Hash()}, []*wire.MsgTx{registryTx})
|
|
if err != nil {
|
|
return nil, errors.Errorf("could not build registry block: %s", err)
|
|
}
|
|
err = addBlockToDAG(registryBlock)
|
|
if err != nil {
|
|
return nil, errors.Errorf("could not add registry block to DAG: %s", err)
|
|
}
|
|
|
|
// Build a subnetwork ID from the registry transaction
|
|
subnetworkID, err := blockdag.TxToSubnetworkID(registryTx)
|
|
if err != nil {
|
|
return nil, errors.Errorf("could not build subnetwork ID: %s", err)
|
|
}
|
|
return subnetworkID, nil
|
|
}
|