2015-02-25 04:34:12 -08:00
|
|
|
package blockpool
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/ethereum/go-ethereum/blockpool/test"
|
2015-03-19 15:46:54 -07:00
|
|
|
"github.com/ethereum/go-ethereum/common"
|
2015-02-25 04:34:12 -08:00
|
|
|
"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) {
|
2015-04-07 10:29:35 -07:00
|
|
|
t.Skip() // :FIXME:
|
|
|
|
|
2015-02-25 04:34:12 -08:00
|
|
|
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)
|
2015-03-19 15:46:54 -07:00
|
|
|
indexes := blockPoolTester.hashPool.HashesToIndexes([]common.Hash{bb.Hash()})
|
2015-02-25 04:34:12 -08:00
|
|
|
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) {
|
2015-04-07 10:29:35 -07:00
|
|
|
t.Skip() // :FIXME:
|
|
|
|
|
2015-02-25 04:34:12 -08:00
|
|
|
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)
|
|
|
|
}
|
|
|
|
}
|
2015-03-19 15:46:54 -07:00
|
|
|
|
2015-03-19 16:00:19 -07:00
|
|
|
func TestIncorrectTD(t *testing.T) {
|
2015-04-08 12:33:54 -07:00
|
|
|
t.Skip() // @zelig this one requires fixing for the TD
|
2015-04-04 12:41:42 -07:00
|
|
|
|
2015-03-19 16:00:19 -07:00
|
|
|
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)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-03-19 15:46:54 -07:00
|
|
|
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()
|
|
|
|
|
|
|
|
}
|