package blockpool import ( "testing" "time" "github.com/ethereum/go-ethereum/blockpool/test" "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/core/types" "github.com/ethereum/go-ethereum/pow" ) func TestInvalidBlock(t *testing.T) { test.LogInit() _, blockPool, blockPoolTester := newTestBlockPool(t) blockPoolTester.blockChain[0] = nil blockPoolTester.initRefBlockChain(2) blockPoolTester.refBlockChain[2] = []int{} blockPool.Start() peer1 := blockPoolTester.newPeer("peer1", 1, 3) peer1.AddPeer() go peer1.serveBlocks(2, 3) go peer1.serveBlockHashes(3, 2, 1, 0) peer1.serveBlocks(0, 1, 2) blockPool.Wait(waitTimeout) blockPool.Stop() blockPoolTester.refBlockChain[2] = []int{} blockPoolTester.checkBlockChain(blockPoolTester.refBlockChain) if len(peer1.peerErrors) == 1 { if peer1.peerErrors[0] != ErrInvalidBlock { t.Errorf("wrong error, got %v, expected %v", peer1.peerErrors[0], ErrInvalidBlock) } } else { t.Errorf("expected %v error, got %v", ErrInvalidBlock, peer1.peerErrors) } } func TestVerifyPoW(t *testing.T) { test.LogInit() _, blockPool, blockPoolTester := newTestBlockPool(t) blockPoolTester.blockChain[0] = nil blockPoolTester.initRefBlockChain(3) first := false blockPoolTester.blockPool.verifyPoW = func(b pow.Block) bool { bb, _ := b.(*types.Block) indexes := blockPoolTester.hashPool.HashesToIndexes([]common.Hash{bb.Hash()}) if indexes[0] == 2 && !first { first = true return false } else { return true } } blockPool.Start() peer1 := blockPoolTester.newPeer("peer1", 1, 3) peer2 := blockPoolTester.newPeer("peer2", 1, 3) peer1.AddPeer() peer2.AddPeer() go peer1.serveBlocks(2, 3) go peer1.serveBlockHashes(3, 2, 1, 0) peer1.serveBlocks(0, 1, 2, 3) blockPoolTester.blockPool.verifyPoW = func(b pow.Block) bool { return true } peer2.serveBlocks(1, 2) blockPool.Wait(waitTimeout) blockPool.Stop() blockPoolTester.refBlockChain[3] = []int{} blockPoolTester.checkBlockChain(blockPoolTester.refBlockChain) if len(peer1.peerErrors) == 1 { if peer1.peerErrors[0] != ErrInvalidPoW { t.Errorf("wrong error, expected %v, got %v", ErrInvalidPoW, peer1.peerErrors[0]) } } else { t.Errorf("expected %v error, got %v", ErrInvalidPoW, peer1.peerErrors) } } func TestUnrequestedBlock(t *testing.T) { test.LogInit() _, blockPool, blockPoolTester := newTestBlockPool(t) blockPoolTester.blockChain[0] = nil blockPool.Start() peer1 := blockPoolTester.newPeer("peer1", 1, 3) peer1.AddPeer() peer1.sendBlocks(1, 2) blockPool.Stop() if len(peer1.peerErrors) == 1 { if peer1.peerErrors[0] != ErrUnrequestedBlock { t.Errorf("wrong error, got %v, expected %v", peer1.peerErrors[0], ErrUnrequestedBlock) } } else { t.Errorf("expected %v error, got %v", ErrUnrequestedBlock, peer1.peerErrors) } } func TestErrInsufficientChainInfo(t *testing.T) { test.LogInit() _, blockPool, blockPoolTester := newTestBlockPool(t) blockPool.Config.BlockHashesTimeout = 100 * time.Millisecond blockPool.Start() peer1 := blockPoolTester.newPeer("peer1", 1, 3) peer1.AddPeer() blockPool.Wait(waitTimeout) blockPool.Stop() if len(peer1.peerErrors) == 1 { if peer1.peerErrors[0] != ErrInsufficientChainInfo { t.Errorf("wrong error, got %v, expected %v", peer1.peerErrors[0], ErrInsufficientChainInfo) } } else { t.Errorf("expected %v error, got %v", ErrInsufficientChainInfo, peer1.peerErrors) } } func TestIncorrectTD(t *testing.T) { test.LogInit() _, blockPool, blockPoolTester := newTestBlockPool(t) blockPoolTester.blockChain[0] = nil blockPoolTester.initRefBlockChain(3) blockPool.Start() peer1 := blockPoolTester.newPeer("peer1", 1, 3) peer1.AddPeer() go peer1.serveBlocks(2, 3) go peer1.serveBlockHashes(3, 2, 1, 0) peer1.serveBlocks(0, 1, 2) blockPool.Wait(waitTimeout) blockPool.Stop() blockPoolTester.refBlockChain[3] = []int{} blockPoolTester.checkBlockChain(blockPoolTester.refBlockChain) if len(peer1.peerErrors) == 1 { if peer1.peerErrors[0] != ErrIncorrectTD { t.Errorf("wrong error, got %v, expected %v", peer1.peerErrors[0], ErrIncorrectTD) } } else { t.Errorf("expected %v error, got %v", ErrIncorrectTD, peer1.peerErrors) } } func TestPeerSuspension(t *testing.T) { test.LogInit() _, blockPool, blockPoolTester := newTestBlockPool(t) blockPool.Config.PeerSuspensionInterval = 100 * time.Millisecond blockPool.Start() peer1 := blockPoolTester.newPeer("peer1", 1, 3) peer1.AddPeer() blockPool.peers.peerError("peer1", 0, "") bestpeer, _ := blockPool.peers.getPeer("peer1") if bestpeer != nil { t.Errorf("peer1 not removed on error") } peer1.AddPeer() bestpeer, _ = blockPool.peers.getPeer("peer1") if bestpeer != nil { t.Errorf("peer1 not removed on reconnect") } time.Sleep(100 * time.Millisecond) peer1.AddPeer() bestpeer, _ = blockPool.peers.getPeer("peer1") if bestpeer == nil { t.Errorf("peer1 not connected after PeerSuspensionInterval") } // blockPool.Wait(waitTimeout) blockPool.Stop() }