111 lines
2.2 KiB
Go
111 lines
2.2 KiB
Go
package lib
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
dbm "github.com/tendermint/tmlibs/db"
|
|
"github.com/tendermint/tmlibs/log"
|
|
|
|
abci "github.com/tendermint/abci/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/store"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
wire "github.com/cosmos/cosmos-sdk/wire"
|
|
)
|
|
|
|
type S struct {
|
|
I uint64
|
|
B bool
|
|
}
|
|
|
|
func defaultComponents(key sdk.StoreKey) (sdk.Context, *wire.Codec) {
|
|
db := dbm.NewMemDB()
|
|
cms := store.NewCommitMultiStore(db)
|
|
cms.MountStoreWithDB(key, sdk.StoreTypeIAVL, db)
|
|
cms.LoadLatestVersion()
|
|
ctx := sdk.NewContext(cms, abci.Header{}, false, nil, log.NewNopLogger())
|
|
cdc := wire.NewCodec()
|
|
return ctx, cdc
|
|
}
|
|
|
|
func TestListMapper(t *testing.T) {
|
|
key := sdk.NewKVStoreKey("list")
|
|
ctx, cdc := defaultComponents(key)
|
|
lm := NewListMapper(cdc, key, "data")
|
|
|
|
val := S{1, true}
|
|
var res S
|
|
|
|
lm.Push(ctx, val)
|
|
assert.Equal(t, uint64(1), lm.Len(ctx))
|
|
lm.Get(ctx, uint64(0), &res)
|
|
assert.Equal(t, val, res)
|
|
|
|
val = S{2, false}
|
|
lm.Set(ctx, uint64(0), val)
|
|
lm.Get(ctx, uint64(0), &res)
|
|
assert.Equal(t, val, res)
|
|
|
|
val = S{100, false}
|
|
lm.Push(ctx, val)
|
|
assert.Equal(t, uint64(2), lm.Len(ctx))
|
|
lm.Get(ctx, uint64(1), &res)
|
|
assert.Equal(t, val, res)
|
|
|
|
lm.Delete(ctx, uint64(1))
|
|
assert.Equal(t, uint64(2), lm.Len(ctx))
|
|
|
|
lm.IterateRead(ctx, &res, func(ctx sdk.Context, index uint64) (brk bool) {
|
|
var temp S
|
|
lm.Get(ctx, index, &temp)
|
|
assert.Equal(t, temp, res)
|
|
|
|
assert.True(t, index != 1)
|
|
return
|
|
})
|
|
|
|
lm.IterateWrite(ctx, &res, func(ctx sdk.Context, index uint64) (brk bool) {
|
|
lm.Set(ctx, index, S{res.I + 1, !res.B})
|
|
return
|
|
})
|
|
|
|
lm.Get(ctx, uint64(0), &res)
|
|
assert.Equal(t, S{3, true}, res)
|
|
}
|
|
|
|
func TestQueueMapper(t *testing.T) {
|
|
key := sdk.NewKVStoreKey("queue")
|
|
ctx, cdc := defaultComponents(key)
|
|
qm := NewQueueMapper(cdc, key, "data")
|
|
|
|
val := S{1, true}
|
|
var res S
|
|
|
|
qm.Push(ctx, val)
|
|
qm.Peek(ctx, &res)
|
|
assert.Equal(t, val, res)
|
|
|
|
qm.Pop(ctx)
|
|
empty := qm.IsEmpty(ctx)
|
|
|
|
assert.True(t, empty)
|
|
assert.NotNil(t, qm.Peek(ctx, &res))
|
|
|
|
qm.Push(ctx, S{1, true})
|
|
qm.Push(ctx, S{2, true})
|
|
qm.Push(ctx, S{3, true})
|
|
qm.Flush(ctx, &res, func(ctx sdk.Context) (brk bool) {
|
|
if res.I == 3 {
|
|
brk = true
|
|
}
|
|
return
|
|
})
|
|
|
|
assert.False(t, qm.IsEmpty(ctx))
|
|
|
|
qm.Pop(ctx)
|
|
assert.True(t, qm.IsEmpty(ctx))
|
|
}
|