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.
96 lines
2.4 KiB
96 lines
2.4 KiB
package files_test
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/tendermint/tendermint/lite"
|
|
liteErr "github.com/tendermint/tendermint/lite/errors"
|
|
"github.com/tendermint/tendermint/lite/files"
|
|
)
|
|
|
|
func checkEqual(stored, loaded lite.FullCommit, chainID string) error {
|
|
err := loaded.ValidateBasic(chainID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !bytes.Equal(stored.ValidatorsHash(), loaded.ValidatorsHash()) {
|
|
return errors.New("Different block hashes")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func TestFileProvider(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
|
|
dir, err := ioutil.TempDir("", "fileprovider-test")
|
|
assert.Nil(err)
|
|
defer os.RemoveAll(dir)
|
|
p := files.NewProvider(dir)
|
|
|
|
chainID := "test-files"
|
|
appHash := []byte("some-data")
|
|
keys := lite.GenValKeys(5)
|
|
count := 10
|
|
|
|
// make a bunch of seeds...
|
|
seeds := make([]lite.FullCommit, count)
|
|
for i := 0; i < count; i++ {
|
|
// two seeds for each validator, to check how we handle dups
|
|
// (10, 0), (10, 1), (10, 1), (10, 2), (10, 2), ...
|
|
vals := keys.ToValidators(10, int64(count/2))
|
|
h := 20 + 10*i
|
|
check := keys.GenCommit(chainID, h, nil, vals, appHash, 0, 5)
|
|
seeds[i] = lite.NewFullCommit(check, vals)
|
|
}
|
|
|
|
// check provider is empty
|
|
seed, err := p.GetByHeight(20)
|
|
require.NotNil(err)
|
|
assert.True(liteErr.IsCommitNotFoundErr(err))
|
|
|
|
seed, err = p.GetByHash(seeds[3].ValidatorsHash())
|
|
require.NotNil(err)
|
|
assert.True(liteErr.IsCommitNotFoundErr(err))
|
|
|
|
// now add them all to the provider
|
|
for _, s := range seeds {
|
|
err = p.StoreCommit(s)
|
|
require.Nil(err)
|
|
// and make sure we can get it back
|
|
s2, err := p.GetByHash(s.ValidatorsHash())
|
|
assert.Nil(err)
|
|
err = checkEqual(s, s2, chainID)
|
|
assert.Nil(err)
|
|
// by height as well
|
|
s2, err = p.GetByHeight(s.Height())
|
|
err = checkEqual(s, s2, chainID)
|
|
assert.Nil(err)
|
|
}
|
|
|
|
// make sure we get the last hash if we overstep
|
|
seed, err = p.GetByHeight(5000)
|
|
if assert.Nil(err, "%+v", err) {
|
|
assert.Equal(seeds[count-1].Height(), seed.Height())
|
|
err = checkEqual(seeds[count-1], seed, chainID)
|
|
assert.Nil(err)
|
|
}
|
|
|
|
// and middle ones as well
|
|
seed, err = p.GetByHeight(47)
|
|
if assert.Nil(err, "%+v", err) {
|
|
// we only step by 10, so 40 must be the one below this
|
|
assert.Equal(40, seed.Height())
|
|
}
|
|
|
|
// and proper error for too low
|
|
_, err = p.GetByHeight(5)
|
|
assert.NotNil(err)
|
|
assert.True(liteErr.IsCommitNotFoundErr(err))
|
|
}
|