You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
2.9 KiB
114 lines
2.9 KiB
// Copyright (c) 2013-2014 The btcsuite developers |
|
// Use of this source code is governed by an ISC |
|
// license that can be found in the LICENSE file. |
|
|
|
package blockchain_test |
|
|
|
import ( |
|
"testing" |
|
|
|
"github.com/btcsuite/btcd/blockchain" |
|
"github.com/btcsuite/btcd/chaincfg" |
|
"github.com/btcsuite/btcd/wire" |
|
"github.com/btcsuite/btcutil" |
|
) |
|
|
|
// TestHaveBlock tests the HaveBlock API to ensure proper functionality. |
|
func TestHaveBlock(t *testing.T) { |
|
// Load up blocks such that there is a side chain. |
|
// (genesis block) -> 1 -> 2 -> 3 -> 4 |
|
// \-> 3a |
|
testFiles := []string{ |
|
"blk_0_to_4.dat.bz2", |
|
"blk_3A.dat.bz2", |
|
} |
|
|
|
var blocks []*btcutil.Block |
|
for _, file := range testFiles { |
|
blockTmp, err := loadBlocks(file) |
|
if err != nil { |
|
t.Errorf("Error loading file: %v\n", err) |
|
return |
|
} |
|
for _, block := range blockTmp { |
|
blocks = append(blocks, block) |
|
} |
|
} |
|
|
|
// Create a new database and chain instance to run tests against. |
|
chain, teardownFunc, err := chainSetup("haveblock") |
|
if err != nil { |
|
t.Errorf("Failed to setup chain instance: %v", err) |
|
return |
|
} |
|
defer teardownFunc() |
|
|
|
// Since we're not dealing with the real block chain, disable |
|
// checkpoints and set the coinbase maturity to 1. |
|
chain.DisableCheckpoints(true) |
|
blockchain.TstSetCoinbaseMaturity(1) |
|
|
|
timeSource := blockchain.NewMedianTime() |
|
for i := 1; i < len(blocks); i++ { |
|
isOrphan, err := chain.ProcessBlock(blocks[i], timeSource, |
|
blockchain.BFNone) |
|
if err != nil { |
|
t.Errorf("ProcessBlock fail on block %v: %v\n", i, err) |
|
return |
|
} |
|
if isOrphan { |
|
t.Errorf("ProcessBlock incorrectly returned block %v "+ |
|
"is an orphan\n", i) |
|
return |
|
} |
|
} |
|
|
|
// Insert an orphan block. |
|
isOrphan, err := chain.ProcessBlock(btcutil.NewBlock(&Block100000), |
|
timeSource, blockchain.BFNone) |
|
if err != nil { |
|
t.Errorf("Unable to process block: %v", err) |
|
return |
|
} |
|
if !isOrphan { |
|
t.Errorf("ProcessBlock indicated block is an not orphan when " + |
|
"it should be\n") |
|
return |
|
} |
|
|
|
tests := []struct { |
|
hash string |
|
want bool |
|
}{ |
|
// Genesis block should be present (in the main chain). |
|
{hash: chaincfg.MainNetParams.GenesisHash.String(), want: true}, |
|
|
|
// Block 3a should be present (on a side chain). |
|
{hash: "00000000474284d20067a4d33f6a02284e6ef70764a3a26d6a5b9df52ef663dd", want: true}, |
|
|
|
// Block 100000 should be present (as an orphan). |
|
{hash: "000000000003ba27aa200b1cecaad478d2b00432346c3f1f3986da1afd33e506", want: true}, |
|
|
|
// Random hashes should not be availble. |
|
{hash: "123", want: false}, |
|
} |
|
|
|
for i, test := range tests { |
|
hash, err := wire.NewShaHashFromStr(test.hash) |
|
if err != nil { |
|
t.Errorf("NewShaHashFromStr: %v", err) |
|
continue |
|
} |
|
|
|
result, err := chain.HaveBlock(hash) |
|
if err != nil { |
|
t.Errorf("HaveBlock #%d unexpected error: %v", i, err) |
|
return |
|
} |
|
if result != test.want { |
|
t.Errorf("HaveBlock #%d got %v want %v", i, result, |
|
test.want) |
|
continue |
|
} |
|
} |
|
}
|
|
|