go-ethereum/core/block_processor_test.go

package core

import (
	"math/big"
	"testing"

	"github.com/ethereum/go-ethereum/ethdb"
	"github.com/ethereum/go-ethereum/event"
)

func proc() (*BlockProcessor, *ChainManager) {
	db, _ := ethdb.NewMemDatabase()
	var mux event.TypeMux

	chainMan := NewChainManager(db, &mux)
	return NewBlockProcessor(db, nil, chainMan, &mux), chainMan
}

func TestNumber(t *testing.T) {
	bp, chain := proc()
	block1 := chain.NewBlock(nil)
	block1.Header().Number = big.NewInt(3)

	err := bp.ValidateBlock(block1, chain.Genesis())
	if err != BlockNumberErr {
		t.Errorf("expected block number error")
	}

	block1 = chain.NewBlock(nil)
	err = bp.ValidateBlock(block1, chain.Genesis())
	if err == BlockNumberErr {
		t.Errorf("didn't expect block number error")
	}
}