Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-06-05 05:36:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add TestBlockStatus (#1361)

Browse Source 
Co-authored-by: stasatdaglabs <39559713+stasatdaglabs@users.noreply.github.com>
This commit is contained in:
Ori Newman 2021-01-06 10:37:10 +02:00 committed by GitHub
parent 741e0962be
commit 3ec1cbe236
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 132 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
domain/consensus/model/testapi/test_consensus.go
View File

@ -23,6 +23,9 @@ type TestConsensus interface {
AddBlock(parentHashes []*externalapi.DomainHash, coinbaseData *externalapi.DomainCoinbaseData,
transactions []*externalapi.DomainTransaction) (*externalapi.DomainHash, *externalapi.BlockInsertionResult, error)
AddHeader(parentHashes []*externalapi.DomainHash, coinbaseData *externalapi.DomainCoinbaseData,
transactions []*externalapi.DomainTransaction) (*externalapi.DomainHash, *externalapi.BlockInsertionResult, error)
DiscardAllStores()
AcceptanceDataStore() model.AcceptanceDataStore

106
domain/consensus/processes/blockprocessor/validateandinsertblock_test.go Normal file
View File

@ -0,0 +1,106 @@
package blockprocessor_test
import (
"github.com/kaspanet/kaspad/domain/consensus"
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
"github.com/kaspanet/kaspad/domain/consensus/ruleerrors"
"github.com/kaspanet/kaspad/domain/consensus/utils/blockheader"
"github.com/kaspanet/kaspad/domain/consensus/utils/consensushashing"
"github.com/kaspanet/kaspad/domain/consensus/utils/testutils"
"github.com/kaspanet/kaspad/domain/dagconfig"
"github.com/pkg/errors"
"testing"
)
func TestBlockStatus(t *testing.T) {
testutils.ForAllNets(t, true, func(t *testing.T, params *dagconfig.Params) {
factory := consensus.NewFactory()
tc, teardown, err := factory.NewTestConsensus(params, "TestBlockStatus")
if err != nil {
t.Fatalf("Error setting up consensus: %+v", err)
}
defer teardown(false)
checkStatus := func(hash *externalapi.DomainHash, expectedStatus externalapi.BlockStatus) {
blockStatus, err := tc.BlockStatusStore().Get(tc.DatabaseContext(), hash)
if err != nil {
t.Fatalf("BlockStatusStore().Get: %+v", err)
}
if blockStatus != expectedStatus {
t.Fatalf("Expected to have status %s but got %s", expectedStatus, blockStatus)
}
}
tipHash := params.GenesisHash
for i := 0; i < 2; i++ {
tipHash, _, err = tc.AddBlock([]*externalapi.DomainHash{tipHash}, nil, nil)
if err != nil {
t.Fatalf("AddBlock: %+v", err)
}
checkStatus(tipHash, externalapi.StatusUTXOValid)
}
headerHash, _, err := tc.AddHeader([]*externalapi.DomainHash{tipHash}, nil, nil)
if err != nil {
t.Fatalf("AddBlock: %+v", err)
}
checkStatus(headerHash, externalapi.StatusHeaderOnly)
nonChainBlockHash, _, err := tc.AddBlock([]*externalapi.DomainHash{params.GenesisHash}, nil, nil)
if err != nil {
t.Fatalf("AddBlock: %+v", err)
}
checkStatus(nonChainBlockHash, externalapi.StatusUTXOPendingVerification)
disqualifiedBlock, _, err := tc.BuildBlockWithParents([]*externalapi.DomainHash{tipHash}, nil, nil)
if err != nil {
t.Fatalf("AddBlock: %+v", err)
}
disqualifiedBlock.Header = blockheader.NewImmutableBlockHeader(
disqualifiedBlock.Header.Version(),
disqualifiedBlock.Header.ParentHashes(),
disqualifiedBlock.Header.HashMerkleRoot(),
externalapi.NewDomainHashFromByteArray(&[externalapi.DomainHashSize]byte{}), // This should disqualify the block
disqualifiedBlock.Header.UTXOCommitment(),
disqualifiedBlock.Header.TimeInMilliseconds(),
disqualifiedBlock.Header.Bits(),
disqualifiedBlock.Header.Nonce(),
)
_, err = tc.ValidateAndInsertBlock(disqualifiedBlock)
if err != nil {
t.Fatalf("ValidateAndInsertBlock: %+v", err)
}
checkStatus(consensushashing.BlockHash(disqualifiedBlock), externalapi.StatusDisqualifiedFromChain)
invalidBlock, _, err := tc.BuildBlockWithParents([]*externalapi.DomainHash{tipHash}, nil, nil)
if err != nil {
t.Fatalf("AddBlock: %+v", err)
}
invalidBlock.Header = blockheader.NewImmutableBlockHeader(
disqualifiedBlock.Header.Version(),
disqualifiedBlock.Header.ParentHashes(),
externalapi.NewDomainHashFromByteArray(&[externalapi.DomainHashSize]byte{}), // This should invalidate the block
disqualifiedBlock.Header.AcceptedIDMerkleRoot(),
disqualifiedBlock.Header.UTXOCommitment(),
disqualifiedBlock.Header.TimeInMilliseconds(),
disqualifiedBlock.Header.Bits(),
disqualifiedBlock.Header.Nonce(),
)
_, err = tc.ValidateAndInsertBlock(invalidBlock)
if err == nil {
t.Fatalf("block is expected to be invalid")
}
if !errors.As(err, &ruleerrors.RuleError{}) {
t.Fatalf("ValidateAndInsertBlock: %+v", err)
}
checkStatus(consensushashing.BlockHash(invalidBlock), externalapi.StatusInvalid)
})
}

2
domain/consensus/processes/blockvalidator/block_body_in_isolation_test.go
View File

@ -21,7 +21,7 @@ func TestChainedTransactions(t *testing.T) {
factory := consensus.NewFactory()
tc, teardown, err := factory.NewTestConsensus(params, "TestUTXOCommitment")
tc, teardown, err := factory.NewTestConsensus(params, "TestChainedTransactions")
if err != nil {
t.Fatalf("Error setting up consensus: %+v", err)
}

22
domain/consensus/test_consensus.go
View File

@ -60,6 +60,28 @@ func (tc *testConsensus) AddBlock(parentHashes []*externalapi.DomainHash, coinba
return consensushashing.BlockHash(block), blockInsertionResult, nil
}
func (tc *testConsensus) AddHeader(parentHashes []*externalapi.DomainHash, coinbaseData *externalapi.DomainCoinbaseData,
transactions []*externalapi.DomainTransaction) (*externalapi.DomainHash, *externalapi.BlockInsertionResult, 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, nil, err
}
block.Transactions = nil
blockInsertionResult, err := tc.blockProcessor.ValidateAndInsertBlock(block)
if err != nil {
return nil, nil, err
}
return consensushashing.BlockHash(block), blockInsertionResult, nil
}
func (tc *testConsensus) DiscardAllStores() {
tc.AcceptanceDataStore().Discard()
tc.BlockHeaderStore().Discard()