You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
140 lines
4.2 KiB
140 lines
4.2 KiB
package lite
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
dbm "github.com/tendermint/tendermint/libs/db"
|
|
log "github.com/tendermint/tendermint/libs/log"
|
|
lerr "github.com/tendermint/tendermint/lite/errors"
|
|
"github.com/tendermint/tendermint/types"
|
|
)
|
|
|
|
// missingProvider doesn't store anything, always a miss.
|
|
// Designed as a mock for testing.
|
|
type missingProvider struct{}
|
|
|
|
// NewMissingProvider returns a provider which does not store anything and always misses.
|
|
func NewMissingProvider() PersistentProvider {
|
|
return missingProvider{}
|
|
}
|
|
|
|
func (missingProvider) SaveFullCommit(FullCommit) error { return nil }
|
|
func (missingProvider) LatestFullCommit(chainID string, minHeight, maxHeight int64) (FullCommit, error) {
|
|
return FullCommit{}, lerr.ErrCommitNotFound()
|
|
}
|
|
func (missingProvider) ValidatorSet(chainID string, height int64) (*types.ValidatorSet, error) {
|
|
return nil, errors.New("missing validator set")
|
|
}
|
|
func (missingProvider) SetLogger(_ log.Logger) {}
|
|
|
|
func TestMemProvider(t *testing.T) {
|
|
p := NewDBProvider("mem", dbm.NewMemDB())
|
|
checkProvider(t, p, "test-mem", "empty")
|
|
}
|
|
|
|
func TestMultiProvider(t *testing.T) {
|
|
p := NewMultiProvider(
|
|
NewMissingProvider(),
|
|
NewDBProvider("mem", dbm.NewMemDB()),
|
|
NewMissingProvider(),
|
|
)
|
|
checkProvider(t, p, "test-cache", "kjfhekfhkewhgit")
|
|
}
|
|
|
|
func checkProvider(t *testing.T, p PersistentProvider, chainID, app string) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
appHash := []byte(app)
|
|
keys := genPrivKeys(5)
|
|
count := 10
|
|
|
|
// Make a bunch of full commits.
|
|
fcz := make([]FullCommit, count)
|
|
for i := 0; i < count; i++ {
|
|
vals := keys.ToValidators(10, int64(count/2))
|
|
h := int64(20 + 10*i)
|
|
fcz[i] = keys.GenFullCommit(chainID, h, nil, vals, vals, appHash, []byte("params"), []byte("results"), 0, 5)
|
|
}
|
|
|
|
// Check that provider is initially empty.
|
|
fc, err := p.LatestFullCommit(chainID, 1, 1<<63-1)
|
|
require.NotNil(err)
|
|
assert.True(lerr.IsErrCommitNotFound(err))
|
|
|
|
// Save all full commits to the provider.
|
|
for _, fc := range fcz {
|
|
err = p.SaveFullCommit(fc)
|
|
require.Nil(err)
|
|
// Make sure we can get it back.
|
|
fc2, err := p.LatestFullCommit(chainID, fc.Height(), fc.Height())
|
|
assert.Nil(err)
|
|
assert.Equal(fc.SignedHeader, fc2.SignedHeader)
|
|
assert.Equal(fc.Validators, fc2.Validators)
|
|
assert.Equal(fc.NextValidators, fc2.NextValidators)
|
|
}
|
|
|
|
// Make sure we get the last hash if we overstep.
|
|
fc, err = p.LatestFullCommit(chainID, 1, 5000)
|
|
if assert.Nil(err) {
|
|
assert.Equal(fcz[count-1].Height(), fc.Height())
|
|
assert.Equal(fcz[count-1], fc)
|
|
}
|
|
|
|
// ... and middle ones as well.
|
|
fc, err = p.LatestFullCommit(chainID, 1, 47)
|
|
if assert.Nil(err) {
|
|
// we only step by 10, so 40 must be the one below this
|
|
assert.EqualValues(40, fc.Height())
|
|
}
|
|
|
|
}
|
|
|
|
// This will make a get height, and if it is good, set the data as well.
|
|
func checkLatestFullCommit(t *testing.T, p PersistentProvider, chainID string, ask, expect int64) {
|
|
fc, err := p.LatestFullCommit(chainID, 1, ask)
|
|
require.Nil(t, err)
|
|
if assert.Equal(t, expect, fc.Height()) {
|
|
err = p.SaveFullCommit(fc)
|
|
require.Nil(t, err)
|
|
}
|
|
}
|
|
|
|
func TestMultiLatestFullCommit(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
// We will write data to the second level of the cache (p2), and see what
|
|
// gets cached/stored in.
|
|
p := NewDBProvider("mem1", dbm.NewMemDB())
|
|
p2 := NewDBProvider("mem2", dbm.NewMemDB())
|
|
cp := NewMultiProvider(p, p2)
|
|
|
|
chainID := "cache-best-height"
|
|
appHash := []byte("01234567")
|
|
keys := genPrivKeys(5)
|
|
count := 10
|
|
|
|
// Set a bunch of full commits.
|
|
for i := 0; i < count; i++ {
|
|
vals := keys.ToValidators(10, int64(count/2))
|
|
h := int64(10 * (i + 1))
|
|
fc := keys.GenFullCommit(chainID, h, nil, vals, vals, appHash, []byte("params"), []byte("results"), 0, 5)
|
|
err := p2.SaveFullCommit(fc)
|
|
require.NoError(err)
|
|
}
|
|
|
|
// Get a few heights from the cache and set them proper.
|
|
checkLatestFullCommit(t, cp, chainID, 57, 50)
|
|
checkLatestFullCommit(t, cp, chainID, 33, 30)
|
|
|
|
// make sure they are set in p as well (but nothing else)
|
|
checkLatestFullCommit(t, p, chainID, 44, 30)
|
|
checkLatestFullCommit(t, p, chainID, 50, 50)
|
|
checkLatestFullCommit(t, p, chainID, 99, 50)
|
|
|
|
// now, query the cache for a higher value
|
|
checkLatestFullCommit(t, p2, chainID, 99, 90)
|
|
checkLatestFullCommit(t, cp, chainID, 99, 90)
|
|
}
|