132 lines
3.8 KiB
Go
132 lines
3.8 KiB
Go
package v2_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
govtypes "cosmossdk.io/api/cosmos/gov/v1beta1"
|
|
"cosmossdk.io/core/header"
|
|
coretesting "cosmossdk.io/core/testing"
|
|
storetypes "cosmossdk.io/store/types"
|
|
"cosmossdk.io/x/authz"
|
|
v2 "cosmossdk.io/x/authz/migrations/v2"
|
|
authzmodule "cosmossdk.io/x/authz/module"
|
|
"cosmossdk.io/x/bank"
|
|
banktypes "cosmossdk.io/x/bank/types"
|
|
|
|
codectestutil "github.com/cosmos/cosmos-sdk/codec/testutil"
|
|
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
|
|
"github.com/cosmos/cosmos-sdk/crypto/keys/ed25519"
|
|
"github.com/cosmos/cosmos-sdk/runtime"
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
moduletestutil "github.com/cosmos/cosmos-sdk/types/module/testutil"
|
|
)
|
|
|
|
func TestMigration(t *testing.T) {
|
|
encodingConfig := moduletestutil.MakeTestEncodingConfig(codectestutil.CodecOptions{}, authzmodule.AppModule{}, bank.AppModule{})
|
|
cdc := encodingConfig.Codec
|
|
|
|
authzKey := storetypes.NewKVStoreKey("authz")
|
|
ctx := testutil.DefaultContext(authzKey, storetypes.NewTransientStoreKey("transient_test"))
|
|
granter1 := sdk.AccAddress(ed25519.GenPrivKey().PubKey().Address())
|
|
grantee1 := sdk.AccAddress(ed25519.GenPrivKey().PubKey().Address())
|
|
granter2 := sdk.AccAddress(ed25519.GenPrivKey().PubKey().Address())
|
|
grantee2 := sdk.AccAddress(ed25519.GenPrivKey().PubKey().Address())
|
|
|
|
sendMsgType := banktypes.SendAuthorization{}.MsgTypeURL()
|
|
genericMsgType := sdk.MsgTypeURL(&govtypes.MsgVote{})
|
|
coins100 := sdk.NewCoins(sdk.NewInt64Coin("atom", 100))
|
|
blockTime := ctx.HeaderInfo().Time
|
|
oneDay := blockTime.AddDate(0, 0, 1)
|
|
oneYear := blockTime.AddDate(1, 0, 0)
|
|
sendAuthz := banktypes.NewSendAuthorization(coins100, nil, codectestutil.CodecOptions{}.GetAddressCodec())
|
|
|
|
grants := []struct {
|
|
granter sdk.AccAddress
|
|
grantee sdk.AccAddress
|
|
msgType string
|
|
authorization func() authz.Grant
|
|
}{
|
|
{
|
|
granter1,
|
|
grantee1,
|
|
sendMsgType,
|
|
func() authz.Grant {
|
|
any, err := codectypes.NewAnyWithValue(sendAuthz)
|
|
require.NoError(t, err)
|
|
return authz.Grant{
|
|
Authorization: any,
|
|
Expiration: &oneDay,
|
|
}
|
|
},
|
|
},
|
|
{
|
|
granter1,
|
|
grantee2,
|
|
sendMsgType,
|
|
func() authz.Grant {
|
|
any, err := codectypes.NewAnyWithValue(sendAuthz)
|
|
require.NoError(t, err)
|
|
return authz.Grant{
|
|
Authorization: any,
|
|
Expiration: &oneDay,
|
|
}
|
|
},
|
|
},
|
|
{
|
|
granter2,
|
|
grantee1,
|
|
genericMsgType,
|
|
func() authz.Grant {
|
|
any, err := codectypes.NewAnyWithValue(authz.NewGenericAuthorization(genericMsgType))
|
|
require.NoError(t, err)
|
|
return authz.Grant{
|
|
Authorization: any,
|
|
Expiration: &oneYear,
|
|
}
|
|
},
|
|
},
|
|
{
|
|
granter2,
|
|
grantee2,
|
|
genericMsgType,
|
|
func() authz.Grant {
|
|
any, err := codectypes.NewAnyWithValue(authz.NewGenericAuthorization(genericMsgType))
|
|
require.NoError(t, err)
|
|
return authz.Grant{
|
|
Authorization: any,
|
|
Expiration: &blockTime,
|
|
}
|
|
},
|
|
},
|
|
}
|
|
|
|
storeService := runtime.NewKVStoreService(authzKey)
|
|
store := storeService.OpenKVStore(ctx)
|
|
env := runtime.NewEnvironment(storeService, coretesting.NewNopLogger())
|
|
|
|
for _, g := range grants {
|
|
grant := g.authorization()
|
|
err := store.Set(v2.GrantStoreKey(g.grantee, g.granter, g.msgType), cdc.MustMarshal(&grant))
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
ctx = ctx.WithHeaderInfo(header.Info{Time: ctx.HeaderInfo().Time.Add(1 * time.Hour)})
|
|
require.NoError(t, v2.MigrateStore(ctx, env, cdc))
|
|
|
|
bz, err := store.Get(v2.GrantStoreKey(grantee1, granter2, genericMsgType))
|
|
require.NoError(t, err)
|
|
require.NotNil(t, bz)
|
|
|
|
bz, err = store.Get(v2.GrantStoreKey(grantee1, granter1, sendMsgType))
|
|
require.NoError(t, err)
|
|
require.NotNil(t, bz)
|
|
|
|
bz, err = store.Get(v2.GrantStoreKey(grantee2, granter2, genericMsgType))
|
|
require.NoError(t, err)
|
|
require.Nil(t, bz)
|
|
}
|