58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package orm
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"pgregory.net/rapid"
|
|
|
|
storetypes "cosmossdk.io/store/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/runtime"
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
)
|
|
|
|
func TestSequence(t *testing.T) {
|
|
rapid.Check(t, func(rt *rapid.T) {
|
|
// Init sets up the real Sequence, including choosing a random initial value,
|
|
// and initializes the model state
|
|
key := storetypes.NewKVStoreKey("test")
|
|
testCtx := testutil.DefaultContextWithDB(t, key, storetypes.NewTransientStoreKey("transient_test"))
|
|
store := runtime.NewKVStoreService(key).OpenKVStore(testCtx.Ctx)
|
|
|
|
// Create primary key table
|
|
seq := NewSequence(0x1)
|
|
|
|
// Choose initial sequence value
|
|
initSeqVal := rapid.Uint64().Draw(rt, "initSeqVal")
|
|
err := seq.InitVal(store, initSeqVal)
|
|
require.NoError(t, err)
|
|
|
|
// Create model state
|
|
state := initSeqVal
|
|
|
|
rt.Repeat(map[string]func(*rapid.T){
|
|
// NextVal is one of the model commands. It checks that the next value of the
|
|
// sequence matches the model and increments the model state.
|
|
"NextVal": func(t *rapid.T) {
|
|
// Check that the next value in the sequence matches the model
|
|
require.Equal(t, state+1, seq.NextVal(store))
|
|
// Increment the model state
|
|
state++
|
|
},
|
|
// CurVal is one of the model commands. It checks that the current value of the
|
|
// sequence matches the model.
|
|
"CurVal": func(t *rapid.T) {
|
|
// Check the current value matches the model
|
|
require.Equal(t, state, seq.CurVal(store))
|
|
},
|
|
// PeekNextVal is one of the model commands. It checks that the next value of
|
|
// the sequence matches the model without modifying the state.
|
|
"PeekNextVal": func(t *rapid.T) {
|
|
// Check that the next value in the sequence matches the model
|
|
require.Equal(t, state+1, seq.PeekNextVal(store))
|
|
},
|
|
})
|
|
})
|
|
}
|