84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
package v2_test
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"cosmossdk.io/x/group"
|
|
v2 "cosmossdk.io/x/group/migrations/v2"
|
|
|
|
codectestutil "github.com/cosmos/cosmos-sdk/codec/testutil"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
)
|
|
|
|
func TestMigrateGenState(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
oldState *authtypes.GenesisState
|
|
newState *authtypes.GenesisState
|
|
}{
|
|
{
|
|
name: "group policy accounts are replaced by base accounts",
|
|
oldState: authtypes.NewGenesisState(authtypes.DefaultParams(), authtypes.GenesisAccounts{
|
|
&authtypes.ModuleAccount{
|
|
BaseAccount: &authtypes.BaseAccount{
|
|
Address: "cosmos1jv65s3grqf6v6jl3dp4t6c9t9rk99cd88lyufl",
|
|
AccountNumber: 3,
|
|
},
|
|
Name: "distribution",
|
|
Permissions: []string{},
|
|
},
|
|
&authtypes.ModuleAccount{
|
|
BaseAccount: &authtypes.BaseAccount{
|
|
Address: "cosmos1q32tjg5qm3n9fj8wjgpd7gl98prefntrckjkyvh8tntp7q33zj0s5tkjrk",
|
|
AccountNumber: 8,
|
|
},
|
|
Name: "cosmos1q32tjg5qm3n9fj8wjgpd7gl98prefntrckjkyvh8tntp7q33zj0s5tkjrk",
|
|
Permissions: []string{},
|
|
},
|
|
}),
|
|
newState: authtypes.NewGenesisState(authtypes.DefaultParams(), authtypes.GenesisAccounts{
|
|
&authtypes.ModuleAccount{
|
|
BaseAccount: &authtypes.BaseAccount{
|
|
Address: "cosmos1jv65s3grqf6v6jl3dp4t6c9t9rk99cd88lyufl",
|
|
AccountNumber: 3,
|
|
},
|
|
Name: "distribution",
|
|
Permissions: []string{},
|
|
},
|
|
func() *authtypes.BaseAccount {
|
|
baseAccount := &authtypes.BaseAccount{
|
|
Address: "cosmos1q32tjg5qm3n9fj8wjgpd7gl98prefntrckjkyvh8tntp7q33zj0s5tkjrk",
|
|
AccountNumber: 8,
|
|
}
|
|
|
|
k := make([]byte, 8)
|
|
binary.BigEndian.PutUint64(k, 0)
|
|
c, err := authtypes.NewModuleCredential(group.ModuleName, []byte{v2.GroupPolicyTablePrefix}, k)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = baseAccount.SetPubKey(c)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return baseAccount
|
|
}(),
|
|
},
|
|
),
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
require.Error(t, authtypes.ValidateGenesis(*tc.oldState))
|
|
actualState := v2.MigrateGenState(tc.oldState, codectestutil.CodecOptions{}.GetAddressCodec())
|
|
require.Equal(t, tc.newState, actualState)
|
|
require.NoError(t, authtypes.ValidateGenesis(*actualState))
|
|
})
|
|
}
|
|
}
|