mukan-sdk/x/bank/migrations/v4/gen_state_test.go

package v4_test

import (
	"testing"

	"github.com/stretchr/testify/assert"

	sdkmath "cosmossdk.io/math"

	sdk "github.com/cosmos/cosmos-sdk/types"
	v4 "github.com/cosmos/cosmos-sdk/x/bank/migrations/v4"
	"github.com/cosmos/cosmos-sdk/x/bank/types"
)

func TestMigrateGenState(t *testing.T) {
	tests := []struct {
		name     string
		oldState *types.GenesisState
		newState *types.GenesisState
	}{
		{
			name: "Balances supply metadata all unchanged",
			oldState: &types.GenesisState{
				Params: types.Params{},
				Balances: []types.Balance{{
					Address: "balance1",
					Coins:   sdk.Coins{sdk.NewCoin("balance1coin", sdkmath.NewInt(8))},
				}},
				Supply: sdk.Coins{sdk.NewCoin("supplycoin", sdkmath.NewInt(800))},
				DenomMetadata: []types.Metadata{{
					Description: "metadesk",
					DenomUnits:  nil,
					Base:        "meta",
					Display:     "meta",
					Name:        "foo",
					Symbol:      "META",
					URI:         "",
					URIHash:     "",
				}},
				SendEnabled: []types.SendEnabled{},
			},
			newState: &types.GenesisState{
				Params: types.Params{},
				Balances: []types.Balance{{
					Address: "balance1",
					Coins:   sdk.Coins{sdk.NewCoin("balance1coin", sdkmath.NewInt(8))},
				}},
				Supply: sdk.Coins{sdk.NewCoin("supplycoin", sdkmath.NewInt(800))},
				DenomMetadata: []types.Metadata{{
					Description: "metadesk",
					DenomUnits:  nil,
					Base:        "meta",
					Display:     "meta",
					Name:        "foo",
					Symbol:      "META",
					URI:         "",
					URIHash:     "",
				}},
				SendEnabled: []types.SendEnabled{},
			},
		},

		{
			name: "default send enabled true not changed",
			oldState: &types.GenesisState{
				Params: types.Params{DefaultSendEnabled: true},
			},
			newState: &types.GenesisState{
				Params: types.Params{DefaultSendEnabled: true},
			},
		},
		{
			name: "default send enabled false not changed",
			oldState: &types.GenesisState{
				Params: types.Params{DefaultSendEnabled: false, SendEnabled: []*types.SendEnabled{}},
			},
			newState: &types.GenesisState{
				Params: types.Params{DefaultSendEnabled: false},
			},
		},
		{
			name: "send enabled entries moved",
			oldState: &types.GenesisState{
				Params: types.Params{
					SendEnabled: []*types.SendEnabled{
						{Denom: "movecointrue", Enabled: true},
						{Denom: "movecoinfalse", Enabled: false},
					},
				},
			},
			newState: &types.GenesisState{
				Params: types.Params{},
				SendEnabled: []types.SendEnabled{
					{Denom: "movecointrue", Enabled: true},
					{Denom: "movecoinfalse", Enabled: false},
				},
			},
		},
		{
			name: "params entries added to existing",
			oldState: &types.GenesisState{
				Params: types.Params{
					SendEnabled: []*types.SendEnabled{
						{Denom: "movecointrue", Enabled: true},
						{Denom: "movecoinfalse", Enabled: false},
					},
				},
				SendEnabled: []types.SendEnabled{
					{Denom: "staycoin", Enabled: true},
				},
			},
			newState: &types.GenesisState{
				Params: types.Params{},
				SendEnabled: []types.SendEnabled{
					{Denom: "staycoin", Enabled: true},
					{Denom: "movecointrue", Enabled: true},
					{Denom: "movecoinfalse", Enabled: false},
				},
			},
		},
		{
			name: "conflicting params ignored",
			oldState: &types.GenesisState{
				Params: types.Params{
					SendEnabled: []*types.SendEnabled{
						{Denom: "staycoin", Enabled: false},
					},
				},
				SendEnabled: []types.SendEnabled{
					{Denom: "staycoin", Enabled: true},
				},
			},
			newState: &types.GenesisState{
				Params: types.Params{},
				SendEnabled: []types.SendEnabled{
					{Denom: "staycoin", Enabled: true},
				},
			},
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			actual := v4.MigrateGenState(tc.oldState)
			assert.Equal(t, tc.newState, actual)
		})
	}

	t.Run("ensure original not changed", func(t *testing.T) {
		origState := types.GenesisState{
			Params: types.Params{
				SendEnabled: []*types.SendEnabled{
					{Denom: "movecointrue", Enabled: true},
					{Denom: "movecoinfalse", Enabled: false},
				},
			},
		}
		_ = v4.MigrateGenState(&origState)
		assert.Len(t, origState.Params.SendEnabled, 2)
	})
}