path: root/eth/downloader/downloader_test.go

package downloader

import (
    "encoding/binary"
    "math/big"
    "testing"
    "time"

    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/core/types"
    "github.com/ethereum/go-ethereum/logger"
    "github.com/ethereum/go-ethereum/logger/glog"
)

var knownHash = common.Hash{1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}

func createHashes(amount int) (hashes []common.Hash) {
    hashes = make([]common.Hash, amount+1)
    hashes[len(hashes)-1] = knownHash

    for i := range hashes[:len(hashes)-1] {
        binary.BigEndian.PutUint64(hashes[i][:8], uint64(i+2))
    }

    return
}

func createBlocksFromHashes(hashes []common.Hash) map[common.Hash]*types.Block {
    blocks := make(map[common.Hash]*types.Block)
    for i, hash := range hashes {
        header := &types.Header{Number: big.NewInt(int64(i))}
        blocks[hash] = types.NewBlockWithHeader(header)
        blocks[hash].HeaderHash = hash
    }

    return blocks
}

type downloadTester struct {
    downloader *Downloader
    hashes     []common.Hash
    blocks     map[common.Hash]*types.Block
    t          *testing.T
    pcount     int
    done       chan bool

    insertedBlocks int
}

func newTester(t *testing.T, hashes []common.Hash, blocks map[common.Hash]*types.Block) *downloadTester {
    tester := &downloadTester{t: t, hashes: hashes, blocks: blocks, done: make(chan bool)}
    downloader := New(tester.hasBlock, tester.insertChain)
    tester.downloader = downloader

    return tester
}

func (dl *downloadTester) hasBlock(hash common.Hash) bool {
    if knownHash == hash {
        return true
    }
    return false
}

func (dl *downloadTester) insertChain(blocks types.Blocks) error {
    dl.insertedBlocks += len(blocks)

    if len(dl.blocks)-1 <= dl.insertedBlocks {
        dl.done <- true
    }

    return nil
}

func (dl *downloadTester) getHashes(hash common.Hash) error {
    dl.downloader.HashCh <- dl.hashes
    return nil
}

func (dl *downloadTester) getBlocks(id string) func([]common.Hash) error {
    return func(hashes []common.Hash) error {
        blocks := make([]*types.Block, len(hashes))
        for i, hash := range hashes {
            blocks[i] = dl.blocks[hash]
        }

        go dl.downloader.DeliverBlocks(id, blocks)

        return nil
    }
}

func (dl *downloadTester) newPeer(id string, td *big.Int, hash common.Hash) {
    dl.pcount++

    dl.downloader.RegisterPeer(id, td, hash, dl.getHashes, dl.getBlocks(id))
}

func (dl *downloadTester) badBlocksPeer(id string, td *big.Int, hash common.Hash) {
    dl.pcount++

    // This bad peer never returns any blocks
    dl.downloader.RegisterPeer(id, td, hash, dl.getHashes, func([]common.Hash) error {
        return nil
    })
}

func TestDownload(t *testing.T) {
    glog.SetV(logger.Detail)
    glog.SetToStderr(true)

    hashes := createHashes(1000)
    blocks := createBlocksFromHashes(hashes)
    tester := newTester(t, hashes, blocks)

    tester.newPeer("peer1", big.NewInt(10000), hashes[len(hashes)-1])
    tester.newPeer("peer2", big.NewInt(0), common.Hash{})
    tester.badBlocksPeer("peer3", big.NewInt(0), common.Hash{})
    tester.badBlocksPeer("peer4", big.NewInt(0), common.Hash{})

success:
    select {
    case <-tester.done:
        break success
    case <-time.After(10 * time.Second): // XXX this could actually fail on a slow computer
        t.Error("timout")
    }
}