prysm-pulse/beacon-chain/powchain/block_cache_test.go
Nishant Das 5bfc457904
Request Headers From ETH1 Node Instead of Blocks (#7121)
* request headers instead of blocks
* Merge branch 'master' into changeToHeader
* Apply suggestions from code review

Co-authored-by: Victor Farazdagi <simple.square@gmail.com>
* change references
* one more change
* last one
* Merge refs/heads/master into changeToHeader
2020-08-27 16:27:08 +00:00

105 lines
2.7 KiB
Go

package powchain
import (
"math/big"
"testing"
"github.com/ethereum/go-ethereum/common"
gethTypes "github.com/ethereum/go-ethereum/core/types"
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
"github.com/prysmaticlabs/prysm/shared/testutil/require"
)
func TestHashKeyFn_OK(t *testing.T) {
hInfo := &headerInfo{
Hash: common.HexToHash("0x0123456"),
}
key, err := hashKeyFn(hInfo)
require.NoError(t, err)
assert.Equal(t, hInfo.Hash.Hex(), key)
}
func TestHashKeyFn_InvalidObj(t *testing.T) {
_, err := hashKeyFn("bad")
assert.Equal(t, ErrNotAHeaderInfo, err)
}
func TestHeightKeyFn_OK(t *testing.T) {
hInfo := &headerInfo{
Number: big.NewInt(555),
}
key, err := heightKeyFn(hInfo)
require.NoError(t, err)
assert.Equal(t, hInfo.Number.String(), key)
}
func TestHeightKeyFn_InvalidObj(t *testing.T) {
_, err := heightKeyFn("bad")
assert.Equal(t, ErrNotAHeaderInfo, err)
}
func TestBlockCache_byHash(t *testing.T) {
cache := newHeaderCache()
header := &gethTypes.Header{
ParentHash: common.HexToHash("0x12345"),
Number: big.NewInt(55),
}
exists, _, err := cache.HeaderInfoByHash(header.Hash())
require.NoError(t, err)
assert.Equal(t, false, exists, "Expected block info not to exist in empty cache")
err = cache.AddHeader(header)
require.NoError(t, err)
exists, fetchedInfo, err := cache.HeaderInfoByHash(header.Hash())
require.NoError(t, err)
assert.Equal(t, true, exists, "Expected headerInfo to exist")
assert.Equal(t, 0, fetchedInfo.Number.Cmp(header.Number), "Expected fetched info number to be equal")
assert.Equal(t, header.Hash(), fetchedInfo.Hash, "Expected hash to be equal")
}
func TestBlockCache_byHeight(t *testing.T) {
cache := newHeaderCache()
header := &gethTypes.Header{
ParentHash: common.HexToHash("0x12345"),
Number: big.NewInt(55),
}
exists, _, err := cache.HeaderInfoByHeight(header.Number)
require.NoError(t, err)
assert.Equal(t, false, exists, "Expected block info not to exist in empty cache")
err = cache.AddHeader(header)
require.NoError(t, err)
exists, fetchedInfo, err := cache.HeaderInfoByHeight(header.Number)
require.NoError(t, err)
assert.Equal(t, true, exists, "Expected headerInfo to exist")
assert.Equal(t, 0, fetchedInfo.Number.Cmp(header.Number), "Expected fetched info number to be equal")
assert.Equal(t, header.Hash(), fetchedInfo.Hash, "Expected hash to be equal")
}
func TestBlockCache_maxSize(t *testing.T) {
cache := newHeaderCache()
for i := int64(0); i < int64(maxCacheSize+10); i++ {
header := &gethTypes.Header{
Number: big.NewInt(i),
}
err := cache.AddHeader(header)
require.NoError(t, err)
}
assert.Equal(t, int(maxCacheSize), len(cache.hashCache.ListKeys()))
assert.Equal(t, int(maxCacheSize), len(cache.heightCache.ListKeys()))
}