2018-06-21 14:33:36 -07:00
|
|
|
package lib
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"testing"
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
"github.com/stretchr/testify/require"
|
2018-06-21 14:33:36 -07:00
|
|
|
|
2018-07-02 13:34:06 -07:00
|
|
|
dbm "github.com/tendermint/tendermint/libs/db"
|
|
|
|
"github.com/tendermint/tendermint/libs/log"
|
2018-06-21 14:33:36 -07:00
|
|
|
|
2018-06-28 17:54:47 -07:00
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
2018-06-21 14:33:36 -07:00
|
|
|
|
2018-09-13 11:17:32 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
2018-06-21 14:33:36 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/store"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
)
|
|
|
|
|
|
|
|
type S struct {
|
|
|
|
I uint64
|
|
|
|
B bool
|
|
|
|
}
|
|
|
|
|
2018-09-13 11:17:32 -07:00
|
|
|
func defaultComponents(key sdk.StoreKey) (sdk.Context, *codec.Codec) {
|
2018-06-21 14:33:36 -07:00
|
|
|
db := dbm.NewMemDB()
|
|
|
|
cms := store.NewCommitMultiStore(db)
|
|
|
|
cms.MountStoreWithDB(key, sdk.StoreTypeIAVL, db)
|
|
|
|
cms.LoadLatestVersion()
|
2018-06-21 18:03:05 -07:00
|
|
|
ctx := sdk.NewContext(cms, abci.Header{}, false, log.NewNopLogger())
|
2018-09-13 11:17:32 -07:00
|
|
|
cdc := codec.New()
|
2018-06-21 14:33:36 -07:00
|
|
|
return ctx, cdc
|
|
|
|
}
|
|
|
|
|
2018-08-06 12:00:49 -07:00
|
|
|
func TestNewLinear(t *testing.T) {
|
2018-09-13 11:17:32 -07:00
|
|
|
cdc := codec.New()
|
2018-08-06 12:00:49 -07:00
|
|
|
require.NotPanics(t, func() { NewLinear(cdc, nil, nil) })
|
|
|
|
require.NotPanics(t, func() { NewLinear(cdc, nil, DefaultLinearKeys()) })
|
|
|
|
require.NotPanics(t, func() { NewLinear(cdc, nil, &LinearKeys{[]byte{0xAA}, []byte{0xBB}, []byte{0xCC}}) })
|
|
|
|
|
|
|
|
require.Panics(t, func() { NewLinear(cdc, nil, &LinearKeys{nil, nil, nil}) })
|
|
|
|
require.Panics(t, func() { NewLinear(cdc, nil, &LinearKeys{[]byte{0xAA}, nil, nil}) })
|
|
|
|
require.Panics(t, func() { NewLinear(cdc, nil, &LinearKeys{nil, []byte{0xBB}, nil}) })
|
|
|
|
require.Panics(t, func() { NewLinear(cdc, nil, &LinearKeys{nil, nil, []byte{0xCC}}) })
|
|
|
|
}
|
|
|
|
|
2018-06-21 14:33:36 -07:00
|
|
|
func TestList(t *testing.T) {
|
|
|
|
key := sdk.NewKVStoreKey("test")
|
|
|
|
ctx, cdc := defaultComponents(key)
|
|
|
|
store := ctx.KVStore(key)
|
|
|
|
lm := NewList(cdc, store, nil)
|
|
|
|
|
|
|
|
val := S{1, true}
|
|
|
|
var res S
|
|
|
|
|
|
|
|
lm.Push(val)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, uint64(1), lm.Len())
|
2018-06-21 14:33:36 -07:00
|
|
|
lm.Get(uint64(0), &res)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, val, res)
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
val = S{2, false}
|
|
|
|
lm.Set(uint64(0), val)
|
|
|
|
lm.Get(uint64(0), &res)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, val, res)
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
val = S{100, false}
|
|
|
|
lm.Push(val)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, uint64(2), lm.Len())
|
2018-06-21 14:33:36 -07:00
|
|
|
lm.Get(uint64(1), &res)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, val, res)
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
lm.Delete(uint64(1))
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, uint64(2), lm.Len())
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
lm.Iterate(&res, func(index uint64) (brk bool) {
|
|
|
|
var temp S
|
|
|
|
lm.Get(index, &temp)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, temp, res)
|
2018-06-21 14:33:36 -07:00
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
require.True(t, index != 1)
|
2018-06-21 14:33:36 -07:00
|
|
|
return
|
|
|
|
})
|
|
|
|
|
|
|
|
lm.Iterate(&res, func(index uint64) (brk bool) {
|
|
|
|
lm.Set(index, S{res.I + 1, !res.B})
|
|
|
|
return
|
|
|
|
})
|
|
|
|
|
|
|
|
lm.Get(uint64(0), &res)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, S{3, true}, res)
|
2018-06-21 14:33:36 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestQueue(t *testing.T) {
|
|
|
|
key := sdk.NewKVStoreKey("test")
|
|
|
|
ctx, cdc := defaultComponents(key)
|
|
|
|
store := ctx.KVStore(key)
|
|
|
|
|
|
|
|
qm := NewQueue(cdc, store, nil)
|
|
|
|
|
|
|
|
val := S{1, true}
|
|
|
|
var res S
|
|
|
|
|
|
|
|
qm.Push(val)
|
|
|
|
qm.Peek(&res)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, val, res)
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
qm.Pop()
|
|
|
|
empty := qm.IsEmpty()
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
require.True(t, empty)
|
|
|
|
require.NotNil(t, qm.Peek(&res))
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
qm.Push(S{1, true})
|
|
|
|
qm.Push(S{2, true})
|
|
|
|
qm.Push(S{3, true})
|
|
|
|
qm.Flush(&res, func() (brk bool) {
|
|
|
|
if res.I == 3 {
|
|
|
|
brk = true
|
|
|
|
}
|
|
|
|
return
|
|
|
|
})
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
require.False(t, qm.IsEmpty())
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
qm.Pop()
|
2018-06-29 18:10:15 -07:00
|
|
|
require.True(t, qm.IsEmpty())
|
2018-06-21 14:33:36 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestOptions(t *testing.T) {
|
|
|
|
key := sdk.NewKVStoreKey("test")
|
|
|
|
ctx, cdc := defaultComponents(key)
|
|
|
|
store := ctx.KVStore(key)
|
|
|
|
|
|
|
|
keys := &LinearKeys{
|
|
|
|
LengthKey: []byte{0xDE, 0xAD},
|
|
|
|
ElemKey: []byte{0xBE, 0xEF},
|
|
|
|
TopKey: []byte{0x12, 0x34},
|
|
|
|
}
|
|
|
|
linear := NewLinear(cdc, store, keys)
|
|
|
|
|
|
|
|
for i := 0; i < 10; i++ {
|
|
|
|
linear.Push(i)
|
|
|
|
}
|
|
|
|
|
|
|
|
var len uint64
|
|
|
|
var top uint64
|
|
|
|
var expected int
|
|
|
|
var actual int
|
|
|
|
|
|
|
|
// Checking keys.LengthKey
|
|
|
|
err := cdc.UnmarshalBinary(store.Get(keys.LengthKey), &len)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, len, linear.Len())
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
// Checking keys.ElemKey
|
|
|
|
for i := 0; i < 10; i++ {
|
|
|
|
linear.Get(uint64(i), &expected)
|
|
|
|
bz := store.Get(append(keys.ElemKey, []byte(fmt.Sprintf("%020d", i))...))
|
|
|
|
err = cdc.UnmarshalBinary(bz, &actual)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, expected, actual)
|
2018-06-21 14:33:36 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
linear.Pop()
|
|
|
|
|
|
|
|
err = cdc.UnmarshalBinary(store.Get(keys.TopKey), &top)
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err)
|
|
|
|
require.Equal(t, top, linear.getTop())
|
2018-06-21 14:33:36 -07:00
|
|
|
|
|
|
|
}
|