mirror of
				https://github.com/0glabs/0g-chain.git
				synced 2025-11-04 03:07:28 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			523 lines
		
	
	
		
			16 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			523 lines
		
	
	
		
			16 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package keeper_test
 | 
						|
 | 
						|
import (
 | 
						|
	"strings"
 | 
						|
	"testing"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/suite"
 | 
						|
 | 
						|
	sdk "github.com/cosmos/cosmos-sdk/types"
 | 
						|
	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
 | 
						|
	vestingtypes "github.com/cosmos/cosmos-sdk/x/auth/vesting/types"
 | 
						|
	tmprototypes "github.com/tendermint/tendermint/proto/tendermint/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"
 | 
						|
	kavadisttypes "github.com/kava-labs/kava/x/kavadist/types"
 | 
						|
)
 | 
						|
 | 
						|
// 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, tmprototypes.Header{Time: suite.genesisTime})
 | 
						|
}
 | 
						|
 | 
						|
func (suite *PayoutTestSuite) SetupWithGenState(authBuilder app.AuthBankGenesisBuilder, incentBuilder testutil.IncentiveGenesisBuilder, hardBuilder testutil.HardGenesisBuilder) {
 | 
						|
	suite.SetupApp()
 | 
						|
 | 
						|
	suite.app.InitializeFromGenesisStatesWithTime(
 | 
						|
		suite.genesisTime,
 | 
						|
		authBuilder.BuildMarshalled(suite.app.AppCodec()),
 | 
						|
		NewPricefeedGenStateMultiFromTime(suite.app.AppCodec(), suite.genesisTime),
 | 
						|
		NewCDPGenStateMulti(suite.app.AppCodec()),
 | 
						|
		hardBuilder.BuildMarshalled(suite.app.AppCodec()),
 | 
						|
		incentBuilder.BuildMarshalled(suite.app.AppCodec()),
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *PayoutTestSuite) getAccount(addr sdk.AccAddress) authtypes.AccountI {
 | 
						|
	ak := suite.app.GetAccountKeeper()
 | 
						|
	return ak.GetAccount(suite.ctx, addr)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *PayoutTestSuite) getModuleAccount(name string) authtypes.ModuleAccountI {
 | 
						|
	ak := suite.app.GetAccountKeeper()
 | 
						|
	return ak.GetModuleAccount(suite.ctx, name)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *PayoutTestSuite) TestSendCoinsToPeriodicVestingAccount() {
 | 
						|
	type accountArgs struct {
 | 
						|
		periods          []vestingtypes.Period
 | 
						|
		origVestingCoins sdk.Coins
 | 
						|
		startTime        int64
 | 
						|
		endTime          int64
 | 
						|
	}
 | 
						|
	type args struct {
 | 
						|
		accArgs             accountArgs
 | 
						|
		period              vestingtypes.Period
 | 
						|
		ctxTime             time.Time
 | 
						|
		mintModAccountCoins bool
 | 
						|
		expectedPeriods     []vestingtypes.Period
 | 
						|
		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: []vestingtypes.Period{
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					},
 | 
						|
					origVestingCoins: cs(c("ukava", 20)),
 | 
						|
					startTime:        100,
 | 
						|
					endTime:          120,
 | 
						|
				},
 | 
						|
				period:              vestingtypes.Period{Length: 2, Amount: cs(c("ukava", 6))},
 | 
						|
				ctxTime:             time.Unix(101, 0),
 | 
						|
				mintModAccountCoins: true,
 | 
						|
				expectedPeriods: []vestingtypes.Period{
 | 
						|
					{Length: 3, Amount: cs(c("ukava", 6))},
 | 
						|
					{Length: 2, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{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: []vestingtypes.Period{
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					},
 | 
						|
					origVestingCoins: cs(c("ukava", 20)),
 | 
						|
					startTime:        100,
 | 
						|
					endTime:          120,
 | 
						|
				},
 | 
						|
				period:              vestingtypes.Period{Length: 7, Amount: cs(c("ukava", 6))},
 | 
						|
				ctxTime:             time.Unix(80, 0),
 | 
						|
				mintModAccountCoins: true,
 | 
						|
				expectedPeriods: []vestingtypes.Period{
 | 
						|
					{Length: 7, Amount: cs(c("ukava", 6))},
 | 
						|
					{Length: 18, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{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: []vestingtypes.Period{
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					},
 | 
						|
					origVestingCoins: cs(c("ukava", 20)),
 | 
						|
					startTime:        100,
 | 
						|
					endTime:          120,
 | 
						|
				},
 | 
						|
				period:              vestingtypes.Period{Length: 7, Amount: cs(c("ukava", 6))},
 | 
						|
				ctxTime:             time.Unix(101, 0),
 | 
						|
				mintModAccountCoins: true,
 | 
						|
				expectedPeriods: []vestingtypes.Period{
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 3, Amount: cs(c("ukava", 6))},
 | 
						|
					{Length: 2, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{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: []vestingtypes.Period{
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					},
 | 
						|
					origVestingCoins: cs(c("ukava", 20)),
 | 
						|
					startTime:        100,
 | 
						|
					endTime:          120,
 | 
						|
				},
 | 
						|
				period:              vestingtypes.Period{Length: 7, Amount: cs(c("ukava", 6))},
 | 
						|
				ctxTime:             time.Unix(125, 0),
 | 
						|
				mintModAccountCoins: true,
 | 
						|
				expectedPeriods: []vestingtypes.Period{
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{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: []vestingtypes.Period{
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					},
 | 
						|
					origVestingCoins: cs(c("ukava", 20)),
 | 
						|
					startTime:        100,
 | 
						|
					endTime:          120,
 | 
						|
				},
 | 
						|
				period:              vestingtypes.Period{Length: 5, Amount: cs(c("ukava", 6))},
 | 
						|
				ctxTime:             time.Unix(110, 0),
 | 
						|
				mintModAccountCoins: true,
 | 
						|
				expectedPeriods: []vestingtypes.Period{
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 11))},
 | 
						|
					{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: []vestingtypes.Period{
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					},
 | 
						|
					origVestingCoins: cs(c("ukava", 20)),
 | 
						|
					startTime:        100,
 | 
						|
					endTime:          120,
 | 
						|
				},
 | 
						|
				period:              vestingtypes.Period{Length: 7, Amount: cs(c("ukava", 6))},
 | 
						|
				ctxTime:             time.Unix(125, 0),
 | 
						|
				mintModAccountCoins: false,
 | 
						|
				expectedPeriods: []vestingtypes.Period{
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{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: []vestingtypes.Period{
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
						{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					},
 | 
						|
					origVestingCoins: cs(c("ukava", 20)),
 | 
						|
					startTime:        100,
 | 
						|
					endTime:          120,
 | 
						|
				},
 | 
						|
				period:              vestingtypes.Period{Length: 50, Amount: cs(c("ukava", 6))},
 | 
						|
				ctxTime:             time.Unix(110, 0),
 | 
						|
				mintModAccountCoins: true,
 | 
						|
				expectedPeriods: []vestingtypes.Period{
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{Length: 5, Amount: cs(c("ukava", 5))},
 | 
						|
					{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.NewAuthBankGenesisBuilder().WithSimplePeriodicVestingAccount(
 | 
						|
				suite.addrs[0],
 | 
						|
				tc.args.accArgs.origVestingCoins,
 | 
						|
				tc.args.accArgs.periods,
 | 
						|
				tc.args.accArgs.startTime,
 | 
						|
			)
 | 
						|
			if tc.args.mintModAccountCoins {
 | 
						|
				authBuilder = authBuilder.WithSimpleModuleAccount(kavadisttypes.ModuleName, tc.args.period.Amount)
 | 
						|
			}
 | 
						|
 | 
						|
			suite.genesisTime = tc.args.ctxTime
 | 
						|
			suite.SetupApp()
 | 
						|
			suite.app.InitializeFromGenesisStates(
 | 
						|
				authBuilder.BuildMarshalled(suite.app.AppCodec()),
 | 
						|
			)
 | 
						|
 | 
						|
			err := suite.keeper.SendTimeLockedCoinsToPeriodicVestingAccount(suite.ctx, kavadisttypes.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.(*vestingtypes.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.NewAuthBankGenesisBuilder().
 | 
						|
		WithSimpleAccount(suite.addrs[1], cs(c("ukava", 400))).
 | 
						|
		WithSimpleModuleAccount(kavadisttypes.ModuleName, cs(c("ukava", 600)))
 | 
						|
 | 
						|
	suite.genesisTime = time.Unix(100, 0)
 | 
						|
	suite.SetupApp()
 | 
						|
	suite.app.InitializeFromGenesisStates(
 | 
						|
		authBuilder.BuildMarshalled(suite.app.AppCodec()),
 | 
						|
	)
 | 
						|
 | 
						|
	// send coins to base account
 | 
						|
	err := suite.keeper.SendTimeLockedCoinsToAccount(suite.ctx, kavadisttypes.ModuleName, suite.addrs[1], cs(c("ukava", 100)), 5)
 | 
						|
	suite.Require().NoError(err)
 | 
						|
	acc := suite.getAccount(suite.addrs[1])
 | 
						|
	vacc, ok := acc.(*vestingtypes.PeriodicVestingAccount)
 | 
						|
	suite.True(ok)
 | 
						|
	expectedPeriods := []vestingtypes.Period{
 | 
						|
		{Length: int64(5), Amount: cs(c("ukava", 100))},
 | 
						|
	}
 | 
						|
 | 
						|
	bk := suite.app.GetBankKeeper()
 | 
						|
 | 
						|
	suite.Equal(expectedPeriods, vacc.VestingPeriods)
 | 
						|
	suite.Equal(cs(c("ukava", 100)), vacc.OriginalVesting)
 | 
						|
	suite.Equal(cs(c("ukava", 500)), bk.GetAllBalances(suite.ctx, vacc.GetAddress()))
 | 
						|
	suite.Equal(int64(105), vacc.EndTime)
 | 
						|
	suite.Equal(int64(100), vacc.StartTime)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *PayoutTestSuite) TestSendCoinsToInvalidAccount() {
 | 
						|
	authBuilder := app.NewAuthBankGenesisBuilder().
 | 
						|
		WithSimpleModuleAccount(kavadisttypes.ModuleName, cs(c("ukava", 600)))
 | 
						|
 | 
						|
	suite.SetupApp()
 | 
						|
	suite.app.InitializeFromGenesisStates(
 | 
						|
		authBuilder.BuildMarshalled(suite.app.AppCodec()),
 | 
						|
	)
 | 
						|
 | 
						|
	// No longer an empty validator vesting account, just a regular addr
 | 
						|
	err := suite.keeper.SendTimeLockedCoinsToAccount(suite.ctx, kavadisttypes.ModuleName, suite.addrs[2], cs(c("ukava", 100)), 5)
 | 
						|
	suite.Require().ErrorIs(err, types.ErrAccountNotFound)
 | 
						|
 | 
						|
	macc := suite.getModuleAccount(cdptypes.ModuleName)
 | 
						|
	err = suite.keeper.SendTimeLockedCoinsToAccount(suite.ctx, kavadisttypes.ModuleName, macc.GetAddress(), cs(c("ukava", 100)), 5)
 | 
						|
	suite.Require().ErrorIs(err, types.ErrInvalidAccountType)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *PayoutTestSuite) TestGetPeriodLength() {
 | 
						|
	type args struct {
 | 
						|
		blockTime time.Time
 | 
						|
		lockup    int64
 | 
						|
	}
 | 
						|
	type periodTest struct {
 | 
						|
		name           string
 | 
						|
		args           args
 | 
						|
		expectedLength int64
 | 
						|
	}
 | 
						|
	testCases := []periodTest{
 | 
						|
		{
 | 
						|
			name: "first half of month",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 11, 2, 15, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    6,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "first half of month long lockup",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 11, 2, 15, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    24,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "second half of month",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 12, 31, 15, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    6,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "second half of month long lockup",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 12, 31, 15, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    24,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "end of feb",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2021, 2, 28, 15, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    6,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "leap year",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 2, 29, 15, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    6,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "leap year long lockup",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 2, 29, 15, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    24,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "exactly half of month, is pushed to start of month + lockup",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC),
 | 
						|
				lockup:    6,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "just before half of month",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 12, 15, 13, 59, 59, 0, time.UTC),
 | 
						|
				lockup:    6,
 | 
						|
			},
 | 
						|
			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(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name: "just after start of month payout time, is pushed to mid month + lockup",
 | 
						|
			args: args{
 | 
						|
				blockTime: time.Date(2020, 12, 1, 14, 0, 1, 0, time.UTC),
 | 
						|
				lockup:    1,
 | 
						|
			},
 | 
						|
			expectedLength: time.Date(2021, 1, 15, 14, 0, 0, 0, time.UTC).Unix() - time.Date(2020, 12, 1, 14, 0, 1, 0, time.UTC).Unix(),
 | 
						|
		},
 | 
						|
	}
 | 
						|
	for _, tc := range testCases {
 | 
						|
		suite.Run(tc.name, func() {
 | 
						|
			length := suite.keeper.GetPeriodLength(tc.args.blockTime, tc.args.lockup)
 | 
						|
			suite.Require().Equal(tc.expectedLength, length)
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestPayoutTestSuite(t *testing.T) {
 | 
						|
	suite.Run(t, new(PayoutTestSuite))
 | 
						|
}
 |