migrate_test.go

package v5

import (
	"testing"

	"github.com/cosmos/gogoproto/types"
	"github.com/stretchr/testify/require"

	"cosmossdk.io/collections"
	"cosmossdk.io/collections/colltest"

	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
)

func TestMigrate(t *testing.T) {
	kv, ctx := colltest.MockStore()
	sb := collections.NewSchemaBuilder(kv)
	seq := collections.NewSequence(sb, collections.NewPrefix(0), "seq")

	wantValue := uint64(100)

	// set old sequence to wanted value
	legacySeqBytes, err := (&types.UInt64Value{Value: wantValue}).Marshal()
	require.NoError(t, err)

	err = kv.OpenKVStore(ctx).Set(authtypes.LegacyGlobalAccountNumberKey, legacySeqBytes)
	require.NoError(t, err)

	err = Migrate(ctx, kv, seq)
	require.NoError(t, err)

	// check that after migration the sequence is what we want it to be
	gotValue, err := seq.Peek(ctx)
	require.NoError(t, err)
	require.Equal(t, wantValue, gotValue)

	// case the global account number was not set
	ctx = kv.NewStoreContext() // this resets the store to zero
	wantValue = collections.DefaultSequenceStart

	err = Migrate(ctx, kv, seq)
	require.NoError(t, err)

	gotValue, err = seq.Next(ctx)
	require.NoError(t, err)
	require.Equal(t, wantValue, gotValue)
}