package keeper_test import ( "errors" "strings" "testing" "time" sdk "github.com/cosmos/cosmos-sdk/types" authexported "github.com/cosmos/cosmos-sdk/x/auth/exported" "github.com/cosmos/cosmos-sdk/x/auth/vesting" supplyexported "github.com/cosmos/cosmos-sdk/x/supply/exported" "github.com/stretchr/testify/suite" abci "github.com/tendermint/tendermint/abci/types" "github.com/kava-labs/kava/app" cdpkeeper "github.com/kava-labs/kava/x/cdp/keeper" cdptypes "github.com/kava-labs/kava/x/cdp/types" hardkeeper "github.com/kava-labs/kava/x/hard/keeper" "github.com/kava-labs/kava/x/incentive/keeper" "github.com/kava-labs/kava/x/incentive/testutil" "github.com/kava-labs/kava/x/incentive/types" "github.com/kava-labs/kava/x/kavadist" ) // Test suite used for all keeper tests type PayoutTestSuite struct { suite.Suite keeper keeper.Keeper hardKeeper hardkeeper.Keeper cdpKeeper cdpkeeper.Keeper app app.TestApp ctx sdk.Context genesisTime time.Time addrs []sdk.AccAddress } // SetupTest is run automatically before each suite test func (suite *PayoutTestSuite) SetupTest() { config := sdk.GetConfig() app.SetBech32AddressPrefixes(config) _, suite.addrs = app.GeneratePrivKeyAddressPairs(5) suite.genesisTime = time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC) } func (suite *PayoutTestSuite) SetupApp() { suite.app = app.NewTestApp() suite.keeper = suite.app.GetIncentiveKeeper() suite.hardKeeper = suite.app.GetHardKeeper() suite.cdpKeeper = suite.app.GetCDPKeeper() suite.ctx = suite.app.NewContext(true, abci.Header{Height: 1, Time: suite.genesisTime}) } func (suite *PayoutTestSuite) SetupWithGenState(authBuilder app.AuthGenesisBuilder, incentBuilder testutil.IncentiveGenesisBuilder, hardBuilder testutil.HardGenesisBuilder) { suite.SetupApp() suite.app.InitializeFromGenesisStatesWithTime( suite.genesisTime, authBuilder.BuildMarshalled(), NewPricefeedGenStateMultiFromTime(suite.genesisTime), NewCDPGenStateMulti(), hardBuilder.BuildMarshalled(), incentBuilder.BuildMarshalled(), ) } func (suite *PayoutTestSuite) getAccount(addr sdk.AccAddress) authexported.Account { ak := suite.app.GetAccountKeeper() return ak.GetAccount(suite.ctx, addr) } func (suite *PayoutTestSuite) getModuleAccount(name string) supplyexported.ModuleAccountI { sk := suite.app.GetSupplyKeeper() return sk.GetModuleAccount(suite.ctx, name) } func (suite *PayoutTestSuite) TestSendCoinsToPeriodicVestingAccount() { type accountArgs struct { periods vesting.Periods origVestingCoins sdk.Coins startTime int64 endTime int64 } type args struct { accArgs accountArgs period vesting.Period ctxTime time.Time mintModAccountCoins bool expectedPeriods vesting.Periods expectedStartTime int64 expectedEndTime int64 } type errArgs struct { expectErr bool contains string } type testCase struct { name string args args errArgs errArgs } type testCases []testCase tests := testCases{ { name: "insert period at beginning schedule", args: args{ accArgs: accountArgs{ periods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, origVestingCoins: cs(c("ukava", 20)), startTime: 100, endTime: 120, }, period: vesting.Period{Length: 2, Amount: cs(c("ukava", 6))}, ctxTime: time.Unix(101, 0), mintModAccountCoins: true, expectedPeriods: vesting.Periods{ vesting.Period{Length: 3, Amount: cs(c("ukava", 6))}, vesting.Period{Length: 2, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, expectedStartTime: 100, expectedEndTime: 120, }, errArgs: errArgs{ expectErr: false, contains: "", }, }, { name: "insert period at beginning with new start time", args: args{ accArgs: accountArgs{ periods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, origVestingCoins: cs(c("ukava", 20)), startTime: 100, endTime: 120, }, period: vesting.Period{Length: 7, Amount: cs(c("ukava", 6))}, ctxTime: time.Unix(80, 0), mintModAccountCoins: true, expectedPeriods: vesting.Periods{ vesting.Period{Length: 7, Amount: cs(c("ukava", 6))}, vesting.Period{Length: 18, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, expectedStartTime: 80, expectedEndTime: 120, }, errArgs: errArgs{ expectErr: false, contains: "", }, }, { name: "insert period in middle of schedule", args: args{ accArgs: accountArgs{ periods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, origVestingCoins: cs(c("ukava", 20)), startTime: 100, endTime: 120, }, period: vesting.Period{Length: 7, Amount: cs(c("ukava", 6))}, ctxTime: time.Unix(101, 0), mintModAccountCoins: true, expectedPeriods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 3, Amount: cs(c("ukava", 6))}, vesting.Period{Length: 2, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, expectedStartTime: 100, expectedEndTime: 120, }, errArgs: errArgs{ expectErr: false, contains: "", }, }, { name: "append to end of schedule", args: args{ accArgs: accountArgs{ periods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, origVestingCoins: cs(c("ukava", 20)), startTime: 100, endTime: 120, }, period: vesting.Period{Length: 7, Amount: cs(c("ukava", 6))}, ctxTime: time.Unix(125, 0), mintModAccountCoins: true, expectedPeriods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 12, Amount: cs(c("ukava", 6))}}, expectedStartTime: 100, expectedEndTime: 132, }, errArgs: errArgs{ expectErr: false, contains: "", }, }, { name: "add coins to existing period", args: args{ accArgs: accountArgs{ periods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, origVestingCoins: cs(c("ukava", 20)), startTime: 100, endTime: 120, }, period: vesting.Period{Length: 5, Amount: cs(c("ukava", 6))}, ctxTime: time.Unix(110, 0), mintModAccountCoins: true, expectedPeriods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 11))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, expectedStartTime: 100, expectedEndTime: 120, }, errArgs: errArgs{ expectErr: false, contains: "", }, }, { name: "insufficient mod account balance", args: args{ accArgs: accountArgs{ periods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, origVestingCoins: cs(c("ukava", 20)), startTime: 100, endTime: 120, }, period: vesting.Period{Length: 7, Amount: cs(c("ukava", 6))}, ctxTime: time.Unix(125, 0), mintModAccountCoins: false, expectedPeriods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 12, Amount: cs(c("ukava", 6))}}, expectedStartTime: 100, expectedEndTime: 132, }, errArgs: errArgs{ expectErr: true, contains: "insufficient funds", }, }, { name: "add large period mid schedule", args: args{ accArgs: accountArgs{ periods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}}, origVestingCoins: cs(c("ukava", 20)), startTime: 100, endTime: 120, }, period: vesting.Period{Length: 50, Amount: cs(c("ukava", 6))}, ctxTime: time.Unix(110, 0), mintModAccountCoins: true, expectedPeriods: vesting.Periods{ vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 5, Amount: cs(c("ukava", 5))}, vesting.Period{Length: 40, Amount: cs(c("ukava", 6))}}, expectedStartTime: 100, expectedEndTime: 160, }, errArgs: errArgs{ expectErr: false, contains: "", }, }, } for _, tc := range tests { suite.Run(tc.name, func() { authBuilder := app.NewAuthGenesisBuilder().WithSimplePeriodicVestingAccount( suite.addrs[0], tc.args.accArgs.origVestingCoins, tc.args.accArgs.periods, tc.args.accArgs.startTime, ) if tc.args.mintModAccountCoins { authBuilder = authBuilder.WithSimpleModuleAccount(kavadist.ModuleName, tc.args.period.Amount) } suite.genesisTime = tc.args.ctxTime suite.SetupApp() suite.app.InitializeFromGenesisStates( authBuilder.BuildMarshalled(), ) err := suite.keeper.SendTimeLockedCoinsToPeriodicVestingAccount(suite.ctx, kavadist.ModuleName, suite.addrs[0], tc.args.period.Amount, tc.args.period.Length) if tc.errArgs.expectErr { suite.Require().Error(err) suite.Require().True(strings.Contains(err.Error(), tc.errArgs.contains)) } else { suite.Require().NoError(err) acc := suite.getAccount(suite.addrs[0]) vacc, ok := acc.(*vesting.PeriodicVestingAccount) suite.Require().True(ok) suite.Require().Equal(tc.args.expectedPeriods, vacc.VestingPeriods) suite.Require().Equal(tc.args.expectedStartTime, vacc.StartTime) suite.Require().Equal(tc.args.expectedEndTime, vacc.EndTime) } }) } } func (suite *PayoutTestSuite) TestSendCoinsToBaseAccount() { authBuilder := app.NewAuthGenesisBuilder(). WithSimpleAccount(suite.addrs[1], cs(c("ukava", 400))). WithSimpleModuleAccount(kavadist.ModuleName, cs(c("ukava", 600))) suite.genesisTime = time.Unix(100, 0) suite.SetupApp() suite.app.InitializeFromGenesisStates( authBuilder.BuildMarshalled(), ) // send coins to base account err := suite.keeper.SendTimeLockedCoinsToAccount(suite.ctx, kavadist.ModuleName, suite.addrs[1], cs(c("ukava", 100)), 5) suite.Require().NoError(err) acc := suite.getAccount(suite.addrs[1]) vacc, ok := acc.(*vesting.PeriodicVestingAccount) suite.True(ok) expectedPeriods := vesting.Periods{ vesting.Period{Length: int64(5), Amount: cs(c("ukava", 100))}, } suite.Equal(expectedPeriods, vacc.VestingPeriods) suite.Equal(cs(c("ukava", 100)), vacc.OriginalVesting) suite.Equal(cs(c("ukava", 500)), vacc.Coins) suite.Equal(int64(105), vacc.EndTime) suite.Equal(int64(100), vacc.StartTime) } func (suite *PayoutTestSuite) TestSendCoinsToInvalidAccount() { authBuilder := app.NewAuthGenesisBuilder(). WithSimpleModuleAccount(kavadist.ModuleName, cs(c("ukava", 600))). WithEmptyValidatorVestingAccount(suite.addrs[2]) suite.SetupApp() suite.app.InitializeFromGenesisStates( authBuilder.BuildMarshalled(), ) err := suite.keeper.SendTimeLockedCoinsToAccount(suite.ctx, kavadist.ModuleName, suite.addrs[2], cs(c("ukava", 100)), 5) suite.Require().True(errors.Is(err, types.ErrInvalidAccountType)) macc := suite.getModuleAccount(cdptypes.ModuleName) err = suite.keeper.SendTimeLockedCoinsToAccount(suite.ctx, kavadist.ModuleName, macc.GetAddress(), cs(c("ukava", 100)), 5) suite.Require().True(errors.Is(err, types.ErrInvalidAccountType)) } func (suite *PayoutTestSuite) TestGetPeriodLength() { type args struct { blockTime time.Time multiplier types.Multiplier expectedLength int64 } type errArgs struct { expectPass bool contains string } type periodTest struct { name string args args errArgs errArgs } testCases := []periodTest{ { name: "first half of month", args: args{ blockTime: time.Date(2020, 11, 2, 15, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2021, 5, 15, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 11, 2, 15, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "first half of month long lockup", args: args{ blockTime: time.Date(2020, 11, 2, 15, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Medium, 24, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2022, 11, 15, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 11, 2, 15, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "second half of month", args: args{ blockTime: time.Date(2020, 12, 31, 15, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2021, 7, 1, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 12, 31, 15, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "second half of month long lockup", args: args{ blockTime: time.Date(2020, 12, 31, 15, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Large, 24, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2023, 1, 1, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 12, 31, 15, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "end of feb", args: args{ blockTime: time.Date(2021, 2, 28, 15, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2021, 9, 1, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2021, 2, 28, 15, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "leap year", args: args{ blockTime: time.Date(2020, 2, 29, 15, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2020, 9, 1, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 2, 29, 15, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "leap year long lockup", args: args{ blockTime: time.Date(2020, 2, 29, 15, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Large, 24, sdk.MustNewDecFromStr("1")), expectedLength: time.Date(2022, 3, 1, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 2, 29, 15, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "exactly half of month", args: args{ blockTime: time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC), multiplier: types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2021, 7, 1, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, { name: "just before half of month", args: args{ blockTime: time.Date(2020, 12, 15, 13, 59, 59, 0, time.UTC), multiplier: types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.333333")), expectedLength: time.Date(2021, 6, 15, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 12, 15, 13, 59, 59, 0, time.UTC).Unix(), }, errArgs: errArgs{ expectPass: true, contains: "", }, }, } for _, tc := range testCases { suite.Run(tc.name, func() { suite.genesisTime = tc.args.blockTime suite.SetupApp() length, err := suite.keeper.GetPeriodLength(suite.ctx, tc.args.multiplier) if tc.errArgs.expectPass { suite.Require().NoError(err) suite.Require().Equal(tc.args.expectedLength, length) } else { suite.Require().Error(err) } }) } } func TestPayoutTestSuite(t *testing.T) { suite.Run(t, new(PayoutTestSuite)) }