55054282d2
* fumpt using main not master... * be more descriptive * fumpt * fix nits Co-authored-by: Julien Robert <julien@rbrt.fr>
66 lines
2.0 KiB
Go
66 lines
2.0 KiB
Go
package authz_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/authz"
|
|
authzmodule "github.com/cosmos/cosmos-sdk/x/authz/module"
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tendermint/tendermint/proto/tendermint/types"
|
|
)
|
|
|
|
func TestExpiredGrantsQueue(t *testing.T) {
|
|
app := simapp.Setup(t, false)
|
|
ctx := app.BaseApp.NewContext(false, types.Header{})
|
|
addrs := simapp.AddTestAddrsIncremental(app, ctx, 5, sdk.NewInt(30000000))
|
|
granter := addrs[0]
|
|
grantee1 := addrs[1]
|
|
grantee2 := addrs[2]
|
|
grantee3 := addrs[3]
|
|
grantee4 := addrs[4]
|
|
expiration := ctx.BlockTime().AddDate(0, 1, 0)
|
|
expiration2 := expiration.AddDate(1, 0, 0)
|
|
smallCoins := sdk.NewCoins(sdk.NewInt64Coin("stake", 10))
|
|
|
|
save := func(grantee sdk.AccAddress, exp *time.Time) {
|
|
err := app.AuthzKeeper.SaveGrant(ctx, grantee, granter, banktypes.NewSendAuthorization(smallCoins), exp)
|
|
require.NoError(t, err, "Grant from %s", grantee.String())
|
|
}
|
|
save(grantee1, &expiration)
|
|
save(grantee2, &expiration)
|
|
save(grantee3, &expiration2)
|
|
save(grantee4, nil)
|
|
|
|
queryHelper := baseapp.NewQueryServerTestHelper(ctx, app.InterfaceRegistry())
|
|
authz.RegisterQueryServer(queryHelper, app.AuthzKeeper)
|
|
queryClient := authz.NewQueryClient(queryHelper)
|
|
|
|
checkGrants := func(ctx sdk.Context, expectedNum int) {
|
|
authzmodule.BeginBlocker(ctx, app.AuthzKeeper)
|
|
|
|
res, err := queryClient.GranterGrants(ctx.Context(), &authz.QueryGranterGrantsRequest{
|
|
Granter: granter.String(),
|
|
})
|
|
require.NoError(t, err)
|
|
require.NotNil(t, res)
|
|
require.Equal(t, expectedNum, len(res.Grants))
|
|
}
|
|
|
|
checkGrants(ctx, 4)
|
|
|
|
// expiration is exclusive!
|
|
ctx = ctx.WithBlockTime(expiration)
|
|
checkGrants(ctx, 4)
|
|
|
|
ctx = ctx.WithBlockTime(expiration.AddDate(0, 0, 1))
|
|
checkGrants(ctx, 2)
|
|
|
|
ctx = ctx.WithBlockTime(expiration2.AddDate(0, 0, 1))
|
|
checkGrants(ctx, 1)
|
|
}
|