[NOD-1532] Check UTXOCommitment for all blocks

This commit is contained in:
Mike Zak 2020-11-16 12:42:26 +02:00 committed by Svarog
parent fca8ed57bd
commit 151910c27a

View File

@ -3,6 +3,8 @@ package consensusstatemanager_test
import ( import (
"testing" "testing"
"github.com/kaspanet/kaspad/domain/consensus/model/testapi"
"github.com/kaspanet/kaspad/domain/consensus/utils/transactionhelper" "github.com/kaspanet/kaspad/domain/consensus/utils/transactionhelper"
"github.com/kaspanet/kaspad/domain/consensus" "github.com/kaspanet/kaspad/domain/consensus"
@ -34,6 +36,7 @@ func TestUTXOCommitment(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Error creating block A: %+v", err) t.Fatalf("Error creating block A: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockAHash, "A")
// Block B: // Block B:
blockBHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockAHash}, nil, nil) blockBHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockAHash}, nil, nil)
if err != nil { if err != nil {
@ -43,11 +46,13 @@ func TestUTXOCommitment(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Error getting block B: %+v", err) t.Fatalf("Error getting block B: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockBHash, "B")
// Block C: // Block C:
blockCHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockBHash}, nil, nil) blockCHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockBHash}, nil, nil)
if err != nil { if err != nil {
t.Fatalf("Error creating block C: %+v", err) t.Fatalf("Error creating block C: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockCHash, "C")
// Block D: // Block D:
blockDTransaction, err := testutils.CreateTransaction( blockDTransaction, err := testutils.CreateTransaction(
blockB.Transactions[transactionhelper.CoinbaseTransactionIndex]) blockB.Transactions[transactionhelper.CoinbaseTransactionIndex])
@ -59,24 +64,30 @@ func TestUTXOCommitment(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Error creating block D: %+v", err) t.Fatalf("Error creating block D: %+v", err)
} }
checkBlockUTXOCommitment(t, consensus, blockDHash, "D")
// Block E: // Block E:
blockEHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockCHash, blockDHash}, nil, nil) blockEHash, err := consensus.AddBlock([]*externalapi.DomainHash{blockCHash, blockDHash}, nil, nil)
if err != nil { if err != nil {
t.Fatalf("Error creating block E: %+v", err) t.Fatalf("Error creating block E: %+v", err)
} }
blockE, err := consensus.GetBlock(blockEHash) checkBlockUTXOCommitment(t, consensus, blockEHash, "E")
})
}
func checkBlockUTXOCommitment(t *testing.T, consensus testapi.TestConsensus, blockHash *externalapi.DomainHash, blockName string) {
block, err := consensus.GetBlock(blockHash)
if err != nil { if err != nil {
t.Fatalf("Error getting block E: %+v", err) t.Fatalf("Error getting block %s: %+v", blockName, err)
} }
// Get the past UTXO set of block E // Get the past UTXO set of block
csm := consensus.ConsensusStateManager() csm := consensus.ConsensusStateManager()
utxoSetIterator, err := csm.RestorePastUTXOSetIterator(blockEHash) utxoSetIterator, err := csm.RestorePastUTXOSetIterator(blockHash)
if err != nil { if err != nil {
t.Fatalf("Error restoring past UTXO of block E: %+v", err) t.Fatalf("Error restoring past UTXO of block %s: %+v", blockName, err)
} }
// Build a Multiset for block E // Build a Multiset
ms := multiset.New() ms := multiset.New()
for utxoSetIterator.Next() { for utxoSetIterator.Next() {
outpoint, entry, err := utxoSetIterator.Get() outpoint, entry, err := utxoSetIterator.Get()
@ -93,12 +104,11 @@ func TestUTXOCommitment(t *testing.T) {
utxoCommitment := ms.Hash() utxoCommitment := ms.Hash()
// Make sure that the two commitments are equal // Make sure that the two commitments are equal
if *utxoCommitment != blockE.Header.UTXOCommitment { if *utxoCommitment != block.Header.UTXOCommitment {
t.Fatalf("TestUTXOCommitment: calculated UTXO commitment and "+ t.Fatalf("TestUTXOCommitment: calculated UTXO commitment for block %s and "+
"actual UTXO commitment don't match. Want: %s, got: %s", "actual UTXO commitment don't match. Want: %s, got: %s", blockName,
utxoCommitment, blockE.Header.UTXOCommitment) utxoCommitment, block.Header.UTXOCommitment)
} }
})
} }
func TestPastUTXOMultiset(t *testing.T) { func TestPastUTXOMultiset(t *testing.T) {