265 lines
12 KiB
Go
265 lines
12 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"cosmossdk.io/math"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
moduletestutil "github.com/cosmos/cosmos-sdk/types/module/testutil"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
"github.com/cosmos/cosmos-sdk/x/distribution"
|
|
"github.com/cosmos/cosmos-sdk/x/distribution/keeper"
|
|
distrtestutil "github.com/cosmos/cosmos-sdk/x/distribution/testutil"
|
|
disttypes "github.com/cosmos/cosmos-sdk/x/distribution/types"
|
|
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
|
|
)
|
|
|
|
func TestAllocateTokensToValidatorWithCommission(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
key := sdk.NewKVStoreKey(disttypes.StoreKey)
|
|
testCtx := testutil.DefaultContextWithDB(t, key, sdk.NewTransientStoreKey("transient_test"))
|
|
encCfg := moduletestutil.MakeTestEncodingConfig(distribution.AppModuleBasic{})
|
|
ctx := testCtx.Ctx.WithBlockHeader(tmproto.Header{Time: time.Now()})
|
|
|
|
bankKeeper := distrtestutil.NewMockBankKeeper(ctrl)
|
|
stakingKeeper := distrtestutil.NewMockStakingKeeper(ctrl)
|
|
accountKeeper := distrtestutil.NewMockAccountKeeper(ctrl)
|
|
|
|
accountKeeper.EXPECT().GetModuleAddress("distribution").Return(distrAcc.GetAddress())
|
|
|
|
distrKeeper := keeper.NewKeeper(
|
|
encCfg.Codec,
|
|
key,
|
|
accountKeeper,
|
|
bankKeeper,
|
|
stakingKeeper,
|
|
"fee_collector",
|
|
authtypes.NewModuleAddress("gov").String(),
|
|
)
|
|
|
|
// create validator with 50% commission
|
|
val, err := distrtestutil.CreateValidator(valConsPk0, math.NewInt(100))
|
|
require.NoError(t, err)
|
|
val.Commission = stakingtypes.NewCommission(sdk.NewDecWithPrec(5, 1), sdk.NewDecWithPrec(5, 1), math.LegacyNewDec(0))
|
|
stakingKeeper.EXPECT().ValidatorByConsAddr(gomock.Any(), sdk.GetConsAddress(valConsPk0)).Return(val).AnyTimes()
|
|
|
|
// allocate tokens
|
|
tokens := sdk.DecCoins{
|
|
{Denom: sdk.DefaultBondDenom, Amount: math.LegacyNewDec(10)},
|
|
}
|
|
distrKeeper.AllocateTokensToValidator(ctx, val, tokens)
|
|
|
|
// check commission
|
|
expected := sdk.DecCoins{
|
|
{Denom: sdk.DefaultBondDenom, Amount: math.LegacyNewDec(5)},
|
|
}
|
|
require.Equal(t, expected, distrKeeper.GetValidatorAccumulatedCommission(ctx, val.GetOperator()).Commission)
|
|
|
|
// check current rewards
|
|
require.Equal(t, expected, distrKeeper.GetValidatorCurrentRewards(ctx, val.GetOperator()).Rewards)
|
|
}
|
|
|
|
func TestAllocateTokensToManyValidators(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
key := sdk.NewKVStoreKey(disttypes.StoreKey)
|
|
testCtx := testutil.DefaultContextWithDB(t, key, sdk.NewTransientStoreKey("transient_test"))
|
|
encCfg := moduletestutil.MakeTestEncodingConfig(distribution.AppModuleBasic{})
|
|
ctx := testCtx.Ctx.WithBlockHeader(tmproto.Header{Time: time.Now()})
|
|
|
|
bankKeeper := distrtestutil.NewMockBankKeeper(ctrl)
|
|
stakingKeeper := distrtestutil.NewMockStakingKeeper(ctrl)
|
|
accountKeeper := distrtestutil.NewMockAccountKeeper(ctrl)
|
|
|
|
feeCollectorAcc := authtypes.NewEmptyModuleAccount("fee_collector")
|
|
accountKeeper.EXPECT().GetModuleAddress("distribution").Return(distrAcc.GetAddress())
|
|
accountKeeper.EXPECT().GetModuleAccount(gomock.Any(), "fee_collector").Return(feeCollectorAcc)
|
|
|
|
distrKeeper := keeper.NewKeeper(
|
|
encCfg.Codec,
|
|
key,
|
|
accountKeeper,
|
|
bankKeeper,
|
|
stakingKeeper,
|
|
"fee_collector",
|
|
authtypes.NewModuleAddress("gov").String(),
|
|
)
|
|
|
|
// reset fee pool & set params
|
|
distrKeeper.SetParams(ctx, disttypes.DefaultParams())
|
|
distrKeeper.SetFeePool(ctx, disttypes.InitialFeePool())
|
|
|
|
// create validator with 50% commission
|
|
valAddr0 := sdk.ValAddress(valConsAddr0)
|
|
val0, err := distrtestutil.CreateValidator(valConsPk0, math.NewInt(100))
|
|
require.NoError(t, err)
|
|
val0.Commission = stakingtypes.NewCommission(sdk.NewDecWithPrec(5, 1), sdk.NewDecWithPrec(5, 1), math.LegacyNewDec(0))
|
|
stakingKeeper.EXPECT().ValidatorByConsAddr(gomock.Any(), sdk.GetConsAddress(valConsPk0)).Return(val0).AnyTimes()
|
|
|
|
// create second validator with 0% commission
|
|
valAddr1 := sdk.ValAddress(valConsAddr1)
|
|
val1, err := distrtestutil.CreateValidator(valConsPk1, math.NewInt(100))
|
|
require.NoError(t, err)
|
|
val1.Commission = stakingtypes.NewCommission(math.LegacyNewDec(0), math.LegacyNewDec(0), math.LegacyNewDec(0))
|
|
stakingKeeper.EXPECT().ValidatorByConsAddr(gomock.Any(), sdk.GetConsAddress(valConsPk1)).Return(val1).AnyTimes()
|
|
|
|
abciValA := abci.Validator{
|
|
Address: valConsPk0.Address(),
|
|
Power: 100,
|
|
}
|
|
abciValB := abci.Validator{
|
|
Address: valConsPk1.Address(),
|
|
Power: 100,
|
|
}
|
|
|
|
// assert initial state: zero outstanding rewards, zero community pool, zero commission, zero current rewards
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr0).Rewards.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr1).Rewards.IsZero())
|
|
require.True(t, distrKeeper.GetFeePool(ctx).CommunityPool.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorAccumulatedCommission(ctx, valAddr0).Commission.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorAccumulatedCommission(ctx, valAddr1).Commission.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorCurrentRewards(ctx, valAddr0).Rewards.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorCurrentRewards(ctx, valAddr1).Rewards.IsZero())
|
|
|
|
// allocate tokens as if both had voted and second was proposer
|
|
fees := sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(100)))
|
|
bankKeeper.EXPECT().GetAllBalances(gomock.Any(), feeCollectorAcc.GetAddress()).Return(fees)
|
|
bankKeeper.EXPECT().SendCoinsFromModuleToModule(gomock.Any(), "fee_collector", disttypes.ModuleName, fees)
|
|
|
|
votes := []abci.VoteInfo{
|
|
{
|
|
Validator: abciValA,
|
|
SignedLastBlock: true,
|
|
},
|
|
{
|
|
Validator: abciValB,
|
|
SignedLastBlock: true,
|
|
},
|
|
}
|
|
distrKeeper.AllocateTokens(ctx, 200, votes)
|
|
|
|
// 98 outstanding rewards (100 less 2 to community pool)
|
|
require.Equal(t, sdk.DecCoins{{Denom: sdk.DefaultBondDenom, Amount: sdk.NewDecWithPrec(490, 1)}}, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr0).Rewards)
|
|
require.Equal(t, sdk.DecCoins{{Denom: sdk.DefaultBondDenom, Amount: sdk.NewDecWithPrec(490, 1)}}, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr1).Rewards)
|
|
|
|
// 2 community pool coins
|
|
require.Equal(t, sdk.DecCoins{{Denom: sdk.DefaultBondDenom, Amount: math.LegacyNewDec(2)}}, distrKeeper.GetFeePool(ctx).CommunityPool)
|
|
|
|
// 50% commission for first proposer, (0.5 * 98%) * 100 / 2 = 23.25
|
|
require.Equal(t, sdk.DecCoins{{Denom: sdk.DefaultBondDenom, Amount: sdk.NewDecWithPrec(2450, 2)}}, distrKeeper.GetValidatorAccumulatedCommission(ctx, valAddr0).Commission)
|
|
|
|
// zero commission for second proposer
|
|
require.True(t, distrKeeper.GetValidatorAccumulatedCommission(ctx, valAddr1).Commission.IsZero())
|
|
|
|
// just staking.proportional for first proposer less commission = (0.5 * 98%) * 100 / 2 = 24.50
|
|
require.Equal(t, sdk.DecCoins{{Denom: sdk.DefaultBondDenom, Amount: sdk.NewDecWithPrec(2450, 2)}}, distrKeeper.GetValidatorCurrentRewards(ctx, valAddr0).Rewards)
|
|
|
|
// proposer reward + staking.proportional for second proposer = (0.5 * (98%)) * 100 = 49
|
|
require.Equal(t, sdk.DecCoins{{Denom: sdk.DefaultBondDenom, Amount: sdk.NewDecWithPrec(490, 1)}}, distrKeeper.GetValidatorCurrentRewards(ctx, valAddr1).Rewards)
|
|
}
|
|
|
|
func TestAllocateTokensTruncation(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
key := sdk.NewKVStoreKey(disttypes.StoreKey)
|
|
testCtx := testutil.DefaultContextWithDB(t, key, sdk.NewTransientStoreKey("transient_test"))
|
|
encCfg := moduletestutil.MakeTestEncodingConfig(distribution.AppModuleBasic{})
|
|
ctx := testCtx.Ctx.WithBlockHeader(tmproto.Header{Time: time.Now()})
|
|
|
|
bankKeeper := distrtestutil.NewMockBankKeeper(ctrl)
|
|
stakingKeeper := distrtestutil.NewMockStakingKeeper(ctrl)
|
|
accountKeeper := distrtestutil.NewMockAccountKeeper(ctrl)
|
|
|
|
feeCollectorAcc := authtypes.NewEmptyModuleAccount("fee_collector")
|
|
accountKeeper.EXPECT().GetModuleAddress("distribution").Return(distrAcc.GetAddress())
|
|
accountKeeper.EXPECT().GetModuleAccount(gomock.Any(), "fee_collector").Return(feeCollectorAcc)
|
|
|
|
distrKeeper := keeper.NewKeeper(
|
|
encCfg.Codec,
|
|
key,
|
|
accountKeeper,
|
|
bankKeeper,
|
|
stakingKeeper,
|
|
"fee_collector",
|
|
authtypes.NewModuleAddress("gov").String(),
|
|
)
|
|
|
|
// reset fee pool
|
|
distrKeeper.SetFeePool(ctx, disttypes.InitialFeePool())
|
|
distrKeeper.SetParams(ctx, disttypes.DefaultParams())
|
|
|
|
// create validator with 10% commission
|
|
valAddr0 := sdk.ValAddress(valConsAddr0)
|
|
val0, err := distrtestutil.CreateValidator(valConsPk0, math.NewInt(100))
|
|
require.NoError(t, err)
|
|
val0.Commission = stakingtypes.NewCommission(sdk.NewDecWithPrec(1, 1), sdk.NewDecWithPrec(1, 1), math.LegacyNewDec(0))
|
|
stakingKeeper.EXPECT().ValidatorByConsAddr(gomock.Any(), sdk.GetConsAddress(valConsPk0)).Return(val0).AnyTimes()
|
|
|
|
// create second validator with 10% commission
|
|
valAddr1 := sdk.ValAddress(valConsAddr1)
|
|
val1, err := distrtestutil.CreateValidator(valConsPk1, math.NewInt(100))
|
|
require.NoError(t, err)
|
|
val1.Commission = stakingtypes.NewCommission(sdk.NewDecWithPrec(1, 1), sdk.NewDecWithPrec(1, 1), math.LegacyNewDec(0))
|
|
stakingKeeper.EXPECT().ValidatorByConsAddr(gomock.Any(), sdk.GetConsAddress(valConsPk1)).Return(val1).AnyTimes()
|
|
|
|
// create third validator with 10% commission
|
|
valAddr2 := sdk.ValAddress(valConsAddr2)
|
|
val2, err := stakingtypes.NewValidator(sdk.ValAddress(valConsAddr2), valConsPk1, stakingtypes.Description{})
|
|
require.NoError(t, err)
|
|
val2.Commission = stakingtypes.NewCommission(sdk.NewDecWithPrec(1, 1), sdk.NewDecWithPrec(1, 1), math.LegacyNewDec(0))
|
|
stakingKeeper.EXPECT().ValidatorByConsAddr(gomock.Any(), sdk.GetConsAddress(valConsPk2)).Return(val2).AnyTimes()
|
|
|
|
abciValA := abci.Validator{
|
|
Address: valConsPk0.Address(),
|
|
Power: 11,
|
|
}
|
|
abciValB := abci.Validator{
|
|
Address: valConsPk1.Address(),
|
|
Power: 10,
|
|
}
|
|
abciValC := abci.Validator{
|
|
Address: valConsPk2.Address(),
|
|
Power: 10,
|
|
}
|
|
|
|
// assert initial state: zero outstanding rewards, zero community pool, zero commission, zero current rewards
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr0).Rewards.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr1).Rewards.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr1).Rewards.IsZero())
|
|
require.True(t, distrKeeper.GetFeePool(ctx).CommunityPool.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorAccumulatedCommission(ctx, valAddr0).Commission.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorAccumulatedCommission(ctx, valAddr1).Commission.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorCurrentRewards(ctx, valAddr0).Rewards.IsZero())
|
|
require.True(t, distrKeeper.GetValidatorCurrentRewards(ctx, valAddr1).Rewards.IsZero())
|
|
|
|
// allocate tokens as if both had voted and second was proposer
|
|
fees := sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(634195840)))
|
|
bankKeeper.EXPECT().GetAllBalances(gomock.Any(), feeCollectorAcc.GetAddress()).Return(fees)
|
|
bankKeeper.EXPECT().SendCoinsFromModuleToModule(gomock.Any(), "fee_collector", disttypes.ModuleName, fees)
|
|
|
|
votes := []abci.VoteInfo{
|
|
{
|
|
Validator: abciValA,
|
|
SignedLastBlock: true,
|
|
},
|
|
{
|
|
Validator: abciValB,
|
|
SignedLastBlock: true,
|
|
},
|
|
{
|
|
Validator: abciValC,
|
|
SignedLastBlock: true,
|
|
},
|
|
}
|
|
distrKeeper.AllocateTokens(ctx, 31, votes)
|
|
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr0).Rewards.IsValid())
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr1).Rewards.IsValid())
|
|
require.True(t, distrKeeper.GetValidatorOutstandingRewards(ctx, valAddr2).Rewards.IsValid())
|
|
}
|