mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 10:05:18 +00:00
75 lines
2.9 KiB
Go
75 lines
2.9 KiB
Go
|
package community_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
sdkmath "cosmossdk.io/math"
|
||
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
||
|
|
"github.com/kava-labs/kava/app"
|
||
|
|
"github.com/kava-labs/kava/x/community"
|
||
|
|
"github.com/kava-labs/kava/x/community/types"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestABCIStakingRewardsArePaidOutOnDisableInflationBlock(t *testing.T) {
|
||
|
|
app.SetSDKConfig()
|
||
|
|
tApp := app.NewTestApp()
|
||
|
|
tApp.InitializeFromGenesisStates()
|
||
|
|
keeper := tApp.GetCommunityKeeper()
|
||
|
|
accountKeeper := tApp.GetAccountKeeper()
|
||
|
|
bankKeeper := tApp.GetBankKeeper()
|
||
|
|
|
||
|
|
// a block that runs after addition of the disable inflation code on chain
|
||
|
|
// but before the disable inflation time
|
||
|
|
initialBlockTime := time.Now()
|
||
|
|
ctx := tApp.NewContext(true, tmproto.Header{Height: 1, Time: initialBlockTime})
|
||
|
|
|
||
|
|
poolAcc := accountKeeper.GetModuleAccount(ctx, types.ModuleName)
|
||
|
|
feeCollectorAcc := accountKeeper.GetModuleAccount(ctx, authtypes.FeeCollectorName)
|
||
|
|
|
||
|
|
disableTime := initialBlockTime.Add(9 * time.Second)
|
||
|
|
|
||
|
|
// set state
|
||
|
|
params, _ := keeper.GetParams(ctx)
|
||
|
|
params.UpgradeTimeDisableInflation = disableTime
|
||
|
|
params.UpgradeTimeSetStakingRewardsPerSecond = sdkmath.LegacyNewDec(1000000) // 1 KAVA
|
||
|
|
params.StakingRewardsPerSecond = sdkmath.LegacyZeroDec()
|
||
|
|
keeper.SetParams(ctx, params)
|
||
|
|
|
||
|
|
// fund community pool account
|
||
|
|
tApp.FundAccount(ctx, poolAcc.GetAddress(), sdk.NewCoins(sdk.NewCoin("ukava", sdkmath.NewInt(10000000)))) // 10 KAVA
|
||
|
|
initialFeeCollectorBalance := bankKeeper.GetBalance(ctx, feeCollectorAcc.GetAddress(), "ukava").Amount
|
||
|
|
|
||
|
|
// run one block
|
||
|
|
community.BeginBlocker(ctx, keeper)
|
||
|
|
|
||
|
|
// assert that staking rewards in parameters are still set to zero
|
||
|
|
params, found := keeper.GetParams(ctx)
|
||
|
|
require.True(t, found)
|
||
|
|
require.Equal(t, sdkmath.LegacyZeroDec(), params.StakingRewardsPerSecond)
|
||
|
|
|
||
|
|
// assert no rewards are given yet
|
||
|
|
rewards := bankKeeper.GetBalance(ctx, feeCollectorAcc.GetAddress(), "ukava").Amount.Sub(initialFeeCollectorBalance)
|
||
|
|
require.Equal(t, sdkmath.ZeroInt(), rewards)
|
||
|
|
|
||
|
|
// new block when disable inflation runs, 10 seconds from initial block for easy math
|
||
|
|
blockTime := disableTime.Add(1 * time.Second)
|
||
|
|
ctx = tApp.NewContext(true, tmproto.Header{Height: ctx.BlockHeight() + 1, Time: blockTime})
|
||
|
|
|
||
|
|
// run the next block
|
||
|
|
community.BeginBlocker(ctx, keeper)
|
||
|
|
|
||
|
|
// assert that staking rewards have been set and disable inflation time is zero
|
||
|
|
params, found = keeper.GetParams(ctx)
|
||
|
|
require.True(t, found)
|
||
|
|
require.True(t, params.UpgradeTimeDisableInflation.IsZero())
|
||
|
|
require.Equal(t, sdkmath.LegacyNewDec(1000000), params.StakingRewardsPerSecond)
|
||
|
|
|
||
|
|
// assert that 10 KAVA has been distributed in rewards
|
||
|
|
rewards = bankKeeper.GetBalance(ctx, feeCollectorAcc.GetAddress(), "ukava").Amount.Sub(initialFeeCollectorBalance)
|
||
|
|
require.Equal(t, sdkmath.NewInt(10000000).String(), rewards.String())
|
||
|
|
}
|