48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package collections
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMap(t *testing.T) {
|
|
sk, ctx := deps()
|
|
schemaBuilder := NewSchemaBuilder(sk)
|
|
m := NewMap(schemaBuilder, NewPrefix("hi"), "m", Uint64Key, Uint64Value)
|
|
_, err := schemaBuilder.Build()
|
|
require.NoError(t, err)
|
|
|
|
// 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 Test_encodeKey(t *testing.T) {
|
|
prefix := "prefix"
|
|
number := []byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0}
|
|
expectedKey := append([]byte(prefix), number...)
|
|
|
|
gotKey, err := encodeKeyWithPrefix(NewPrefix(prefix).Bytes(), Uint64Key, 0)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expectedKey, gotKey)
|
|
}
|