eth/downloader: added additional tests

This commit is contained in:
obscuren
2015-05-03 14:09:33 +02:00
parent c6ad3aec05
commit 25bc88113f
2 changed files with 73 additions and 4 deletions

View File

@ -23,12 +23,19 @@ func createHashes(start, amount int) (hashes []common.Hash) {
return
}
func createBlock(i int, prevHash, hash common.Hash) *types.Block {
header := &types.Header{Number: big.NewInt(int64(i))}
block := types.NewBlockWithHeader(header)
block.HeaderHash = hash
block.ParentHeaderHash = knownHash
return block
}
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(len(hashes) - i))}
blocks[hash] = types.NewBlockWithHeader(header)
blocks[hash].HeaderHash = hash
blocks[hash] = createBlock(len(hashes)-i, knownHash, hash)
}
return blocks
@ -162,7 +169,7 @@ func TestTaking(t *testing.T) {
t.Error("download error", err)
}
bs1 := tester.downloader.TakeBlocks(1000)
bs1 := tester.downloader.TakeBlocks()
if len(bs1) != 1000 {
t.Error("expected to take 1000, got", len(bs1))
}