fca24b6687
Co-authored-by: testinginprod <testinginprod@somewhere.idk> Co-authored-by: Aaron Craelius <aaron@regen.network>
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package collections
|
|
|
|
import (
|
|
"testing"
|
|
|
|
storetypes "github.com/cosmos/cosmos-sdk/store/types"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMap(t *testing.T) {
|
|
sk, ctx := deps()
|
|
m := NewMap(sk, NewPrefix("hi"), Uint64Key, Uint64Value)
|
|
|
|
// test not has
|
|
has, err := m.Has(ctx, 1)
|
|
require.NoError(t, err)
|
|
require.False(t, has)
|
|
// test get error
|
|
_, err = m.Get(ctx, 1)
|
|
require.ErrorIs(t, err, ErrNotFound)
|
|
|
|
// test set/get
|
|
err = m.Set(ctx, 1, 100)
|
|
require.NoError(t, err)
|
|
v, err := m.Get(ctx, 1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(100), v)
|
|
|
|
// test remove
|
|
err = m.Remove(ctx, 1)
|
|
require.NoError(t, err)
|
|
has, err = m.Has(ctx, 1)
|
|
require.NoError(t, err)
|
|
require.False(t, has)
|
|
}
|
|
|
|
func TestMap_encodeKey(t *testing.T) {
|
|
prefix := "prefix"
|
|
number := []byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0}
|
|
expectedKey := append([]byte(prefix), number...)
|
|
|
|
m := NewMap(storetypes.NewKVStoreKey("test"), NewPrefix(prefix), Uint64Key, Uint64Value)
|
|
|
|
gotKey, err := m.encodeKey(0)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expectedKey, gotKey)
|
|
}
|