97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package list
|
|
|
|
import (
|
|
"math/rand"
|
|
"testing"
|
|
|
|
dbm "github.com/tendermint/tendermint/libs/db"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/store/rootmulti"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type TestStruct struct {
|
|
I uint64
|
|
B bool
|
|
}
|
|
|
|
func defaultComponents(key sdk.StoreKey) (sdk.Context, *codec.Codec) {
|
|
db := dbm.NewMemDB()
|
|
cms := rootmulti.NewStore(db)
|
|
cms.MountStoreWithDB(key, sdk.StoreTypeIAVL, db)
|
|
cms.LoadLatestVersion()
|
|
ctx := sdk.NewContext(cms, abci.Header{}, false, log.NewNopLogger())
|
|
cdc := codec.New()
|
|
return ctx, cdc
|
|
}
|
|
func TestList(t *testing.T) {
|
|
key := sdk.NewKVStoreKey("test")
|
|
ctx, cdc := defaultComponents(key)
|
|
store := ctx.KVStore(key)
|
|
lm := NewList(cdc, store)
|
|
|
|
val := TestStruct{1, true}
|
|
var res TestStruct
|
|
|
|
lm.Push(val)
|
|
require.Equal(t, uint64(1), lm.Len())
|
|
lm.Get(uint64(0), &res)
|
|
require.Equal(t, val, res)
|
|
|
|
val = TestStruct{2, false}
|
|
lm.Set(uint64(0), val)
|
|
lm.Get(uint64(0), &res)
|
|
require.Equal(t, val, res)
|
|
|
|
val = TestStruct{100, false}
|
|
lm.Push(val)
|
|
require.Equal(t, uint64(2), lm.Len())
|
|
lm.Get(uint64(1), &res)
|
|
require.Equal(t, val, res)
|
|
|
|
lm.Delete(uint64(1))
|
|
require.Equal(t, uint64(2), lm.Len())
|
|
|
|
lm.Iterate(&res, func(index uint64) (brk bool) {
|
|
var temp TestStruct
|
|
lm.Get(index, &temp)
|
|
require.Equal(t, temp, res)
|
|
|
|
require.True(t, index != 1)
|
|
return
|
|
})
|
|
|
|
lm.Iterate(&res, func(index uint64) (brk bool) {
|
|
lm.Set(index, TestStruct{res.I + 1, !res.B})
|
|
return
|
|
})
|
|
|
|
lm.Get(uint64(0), &res)
|
|
require.Equal(t, TestStruct{3, true}, res)
|
|
}
|
|
|
|
func TestListRandom(t *testing.T) {
|
|
key := sdk.NewKVStoreKey("test")
|
|
ctx, cdc := defaultComponents(key)
|
|
store := ctx.KVStore(key)
|
|
list := NewList(cdc, store)
|
|
mocklist := []uint32{}
|
|
|
|
for i := 0; i < 100; i++ {
|
|
item := rand.Uint32()
|
|
list.Push(item)
|
|
mocklist = append(mocklist, item)
|
|
}
|
|
|
|
for k, v := range mocklist {
|
|
var i uint32
|
|
require.NotPanics(t, func() { list.Get(uint64(k), &i) })
|
|
require.Equal(t, v, i)
|
|
}
|
|
}
|