aboutsummaryrefslogtreecommitdiffstats
path: root/core/chain_manager_test.go
diff options
context:
space:
mode:
authorobscuren <geffobscura@gmail.com>2015-04-29 18:43:24 +0800
committerobscuren <geffobscura@gmail.com>2015-04-29 18:51:05 +0800
commit4e0796771190b6f8a976d931540d5f21789a882f (patch)
treec8ec0ede8d8ba364eecb693dfeb2fb0a8666d064 /core/chain_manager_test.go
parentaf73d1d6823565c5bf8cc02d6f110bcc1edb2d99 (diff)
downloadgo-tangerine-4e0796771190b6f8a976d931540d5f21789a882f.tar.gz
go-tangerine-4e0796771190b6f8a976d931540d5f21789a882f.tar.zst
go-tangerine-4e0796771190b6f8a976d931540d5f21789a882f.zip
core: fixed chain reorg during splits
Chain reorgs weren't properly handled when a chain was further ahead. Previously we'd end up with mixed chains in our canonical numbering sequence. Added test for this type of forking. ``` /-o-o-o A o-C-+ \-o-o-o-o B ``` Ends up with with C A1, A2, A3, B4
Diffstat (limited to 'core/chain_manager_test.go')
-rw-r--r--core/chain_manager_test.go52
1 files changed, 51 insertions, 1 deletions
diff --git a/core/chain_manager_test.go b/core/chain_manager_test.go
index c2911150a..a88afd7c8 100644
--- a/core/chain_manager_test.go
+++ b/core/chain_manager_test.go
@@ -9,6 +9,7 @@ import (
"strconv"
"testing"
+ "github.com/ethereum/go-ethereum/core/state"
"github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/ethdb"
"github.com/ethereum/go-ethereum/event"
@@ -56,12 +57,14 @@ func testFork(t *testing.T, bman *BlockProcessor, i, N int, f func(td1, td2 *big
}
// Compare difficulties
f(tdpre, td)
+
+ // Loop over parents making sure reconstruction is done properly
}
func printChain(bc *ChainManager) {
for i := bc.CurrentBlock().Number().Uint64(); i > 0; i-- {
b := bc.GetBlockByNumber(uint64(i))
- fmt.Printf("\t%x\n", b.Hash())
+ fmt.Printf("\t%x %v\n", b.Hash(), b.Difficulty())
}
}
@@ -344,3 +347,50 @@ func TestGetAncestors(t *testing.T) {
ancestors := chainMan.GetAncestors(chain[len(chain)-1], 4)
fmt.Println(ancestors)
}
+
+type bproc struct{}
+
+func (bproc) Process(*types.Block) (state.Logs, error) { return nil, nil }
+
+func makeChainWithDiff(genesis *types.Block, d []int, seed byte) []*types.Block {
+ var chain []*types.Block
+ for i, difficulty := range d {
+ header := &types.Header{Number: big.NewInt(int64(i + 1)), Difficulty: big.NewInt(int64(difficulty))}
+ block := types.NewBlockWithHeader(header)
+ copy(block.HeaderHash[:2], []byte{byte(i + 1), seed})
+ if i == 0 {
+ block.ParentHeaderHash = genesis.Hash()
+ } else {
+ copy(block.ParentHeaderHash[:2], []byte{byte(i), seed})
+ }
+
+ chain = append(chain, block)
+ }
+ return chain
+}
+
+func TestReorg(t *testing.T) {
+ db, _ := ethdb.NewMemDatabase()
+ var eventMux event.TypeMux
+
+ genesis := GenesisBlock(db)
+ bc := &ChainManager{blockDb: db, stateDb: db, genesisBlock: genesis, eventMux: &eventMux}
+ bc.cache = NewBlockCache(100)
+ bc.futureBlocks = NewBlockCache(100)
+ bc.processor = bproc{}
+ bc.ResetWithGenesisBlock(genesis)
+ bc.txState = state.ManageState(bc.State())
+
+ chain1 := makeChainWithDiff(genesis, []int{1, 2, 4}, 10)
+ chain2 := makeChainWithDiff(genesis, []int{1, 2, 3, 4}, 11)
+
+ bc.InsertChain(chain1)
+ bc.InsertChain(chain2)
+
+ prev := bc.CurrentBlock()
+ for block := bc.GetBlockByNumber(bc.CurrentBlock().NumberU64() - 1); block.NumberU64() != 0; prev, block = block, bc.GetBlockByNumber(block.NumberU64()-1) {
+ if prev.ParentHash() != block.Hash() {
+ t.Errorf("parent hash mismatch %x - %x", prev.ParentHash(), block.Hash())
+ }
+ }
+}