smeb_y/0g-chain
1
0
Fork 0
mirror of https://github.com/0glabs/0g-chain.git synced 2024-12-26 00:05:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
04946493ae
0g-chain/x/harvest/keeper/timelock_test.go
Denali Marsh 5ce045f46f
rename hvt (#661)
2020-09-25 11:07:20 -04:00

335 lines
14 KiB
Go
Raw Blame History

package keeper_test
import (
"time"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/x/auth"
"github.com/cosmos/cosmos-sdk/x/auth/vesting"
"github.com/kava-labs/kava/app"
"github.com/kava-labs/kava/x/harvest/types"
abci "github.com/tendermint/tendermint/abci/types"
"github.com/tendermint/tendermint/crypto"
)
func (suite *KeeperTestSuite) TestSendTimeLockedCoinsToAccount() {
type accountArgs struct {
addr sdk.AccAddress
vestingAccountBefore bool
vestingAccountAfter bool
coins sdk.Coins
periods vesting.Periods
origVestingCoins sdk.Coins
startTime int64
endTime int64
}
type args struct {
accArgs accountArgs
period vesting.Period
blockTime time.Time
expectedAccountBalance sdk.Coins
expectedModAccountBalance sdk.Coins
expectedPeriods vesting.Periods
expectedStartTime int64
expectedEndTime int64
}
type errArgs struct {
expectPass bool
contains string
}
type testCase struct {
name string
args args
errArgs errArgs
}
testCases := []testCase{
{
name: "send liquid coins to base account",
args: args{
accArgs: accountArgs{
addr: sdk.AccAddress(crypto.AddressHash([]byte("test"))),
vestingAccountBefore: false,
vestingAccountAfter: false,
coins: sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(1000)), sdk.NewCoin("btcb", sdk.NewInt(1000))),
},
period: vesting.Period{Length: 0, Amount: sdk.NewCoins(sdk.NewCoin("hard", sdk.NewInt(100)))},
blockTime: time.Date(2020, 11, 1, 14, 0, 0, 0, time.UTC),
expectedAccountBalance: sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(1000)), sdk.NewCoin("btcb", sdk.NewInt(1000)), sdk.NewCoin("hard", sdk.NewInt(100))),
expectedModAccountBalance: sdk.NewCoins(sdk.NewCoin("hard", sdk.NewInt(900))),
expectedPeriods: vesting.Periods{},
expectedStartTime: 0,
expectedEndTime: 0,
},
errArgs: errArgs{
expectPass: true,
contains: "",
},
},
{
name: "send liquid coins to vesting account",
args: args{
accArgs: accountArgs{
addr: sdk.AccAddress(crypto.AddressHash([]byte("test"))),
vestingAccountBefore: true,
vestingAccountAfter: true,
coins: sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(1000)), sdk.NewCoin("btcb", sdk.NewInt(1000))),
periods: vesting.Periods{
vesting.Period{Amount: sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(100))), Length: 100},
},
origVestingCoins: sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(100))),
startTime: time.Date(2020, 11, 1, 14, 0, 0, 0, time.UTC).Unix(),
endTime: time.Date(2020, 11, 1, 14, 0, 0, 0, time.UTC).Unix() + 100,
},
period: vesting.Period{Length: 0, Amount: sdk.NewCoins(sdk.NewCoin("hard", sdk.NewInt(100)))},
blockTime: time.Date(2020, 11, 1, 14, 0, 0, 0, time.UTC),
expectedAccountBalance: sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(1000)), sdk.NewCoin("btcb", sdk.NewInt(1000)), sdk.NewCoin("hard", sdk.NewInt(100))),
expectedModAccountBalance: sdk.NewCoins(sdk.NewCoin("hard", sdk.NewInt(900))),
expectedPeriods: vesting.Periods{
vesting.Period{Amount: sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(100))), Length: 100},
},
expectedStartTime: time.Date(2020, 11, 1, 14, 0, 0, 0, time.UTC).Unix(),
expectedEndTime: time.Date(2020, 11, 1, 14, 0, 0, 0, time.UTC).Unix() + 100,
},
errArgs: errArgs{
expectPass: true,
contains: "",
},
},
{
name: "insert period at beginning of schedule",
args: args{
accArgs: accountArgs{
addr: sdk.AccAddress(crypto.AddressHash([]byte("test"))),
vestingAccountBefore: true,
vestingAccountAfter: true,
coins: cs(c("bnb", 20)),
periods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
origVestingCoins: cs(c("bnb", 20)),
startTime: 100,
endTime: 120,
},
period: vesting.Period{Length: 2, Amount: cs(c("hard", 6))},
blockTime: time.Unix(101, 0),
expectedAccountBalance: cs(c("bnb", 20), c("hard", 6)),
expectedModAccountBalance: cs(c("hard", 994)),
expectedPeriods: vesting.Periods{
vesting.Period{Length: 3, Amount: cs(c("hard", 6))},
vesting.Period{Length: 2, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
expectedStartTime: 100,
expectedEndTime: 120,
},
errArgs: errArgs{
expectPass: true,
contains: "",
},
},
{
name: "insert period at beginning with new start time",
args: args{
accArgs: accountArgs{
addr: sdk.AccAddress(crypto.AddressHash([]byte("test"))),
vestingAccountBefore: true,
vestingAccountAfter: true,
coins: cs(c("bnb", 20)),
periods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
origVestingCoins: cs(c("bnb", 20)),
startTime: 100,
endTime: 120,
},
period: vesting.Period{Length: 7, Amount: cs(c("hard", 6))},
blockTime: time.Unix(80, 0),
expectedAccountBalance: cs(c("bnb", 20), c("hard", 6)),
expectedModAccountBalance: cs(c("hard", 994)),
expectedPeriods: vesting.Periods{
vesting.Period{Length: 7, Amount: cs(c("hard", 6))},
vesting.Period{Length: 18, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
expectedStartTime: 80,
expectedEndTime: 120,
},
errArgs: errArgs{
expectPass: true,
contains: "",
},
},
{
name: "insert period in middle of schedule",
args: args{
accArgs: accountArgs{
addr: sdk.AccAddress(crypto.AddressHash([]byte("test"))),
vestingAccountBefore: true,
vestingAccountAfter: true,
coins: cs(c("bnb", 20)),
periods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
origVestingCoins: cs(c("bnb", 20)),
startTime: 100,
endTime: 120,
},
period: vesting.Period{Length: 7, Amount: cs(c("hard", 6))},
blockTime: time.Unix(101, 0),
expectedAccountBalance: cs(c("bnb", 20), c("hard", 6)),
expectedModAccountBalance: cs(c("hard", 994)),
expectedPeriods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 3, Amount: cs(c("hard", 6))},
vesting.Period{Length: 2, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
expectedStartTime: 100,
expectedEndTime: 120,
},
errArgs: errArgs{
expectPass: true,
contains: "",
},
},
{
name: "append to end of schedule",
args: args{
accArgs: accountArgs{
addr: sdk.AccAddress(crypto.AddressHash([]byte("test"))),
vestingAccountBefore: true,
vestingAccountAfter: true,
coins: cs(c("bnb", 20)),
periods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
origVestingCoins: cs(c("bnb", 20)),
startTime: 100,
endTime: 120,
},
period: vesting.Period{Length: 7, Amount: cs(c("hard", 6))},
blockTime: time.Unix(125, 0),
expectedAccountBalance: cs(c("bnb", 20), c("hard", 6)),
expectedModAccountBalance: cs(c("hard", 994)),
expectedPeriods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 12, Amount: cs(c("hard", 6))}},
expectedStartTime: 100,
expectedEndTime: 132,
},
errArgs: errArgs{
expectPass: true,
contains: "",
},
},
{
name: "add coins to existing period",
args: args{
accArgs: accountArgs{
addr: sdk.AccAddress(crypto.AddressHash([]byte("test"))),
vestingAccountBefore: true,
vestingAccountAfter: true,
coins: cs(c("bnb", 20)),
periods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
origVestingCoins: cs(c("bnb", 20)),
startTime: 100,
endTime: 120,
},
period: vesting.Period{Length: 5, Amount: cs(c("hard", 6))},
blockTime: time.Unix(110, 0),
expectedAccountBalance: cs(c("bnb", 20), c("hard", 6)),
expectedModAccountBalance: cs(c("hard", 994)),
expectedPeriods: vesting.Periods{
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5), c("hard", 6))},
vesting.Period{Length: 5, Amount: cs(c("bnb", 5))}},
expectedStartTime: 100,
expectedEndTime: 120,
},
errArgs: errArgs{
expectPass: true,
contains: "",
},
},
}
for _, tc := range testCases {
suite.Run(tc.name, func() {
// create new app with one funded account
config := sdk.GetConfig()
app.SetBech32AddressPrefixes(config)
// Initialize test app and set context
tApp := app.NewTestApp()
ctx := tApp.NewContext(true, abci.Header{Height: 1, Time: tc.args.blockTime})
authGS := app.NewAuthGenState([]sdk.AccAddress{tc.args.accArgs.addr}, []sdk.Coins{tc.args.accArgs.coins})
harvestGS := types.NewGenesisState(types.NewParams(
true,
types.DistributionSchedules{
types.NewDistributionSchedule(true, "bnb", time.Date(2020, 10, 8, 14, 0, 0, 0, time.UTC), time.Date(2020, 11, 22, 14, 0, 0, 0, time.UTC), sdk.NewCoin("hard", sdk.NewInt(5000)), time.Date(2021, 11, 22, 14, 0, 0, 0, time.UTC), types.Multipliers{types.NewMultiplier(types.Small, 0, sdk.MustNewDecFromStr("0.33")), types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.5")), types.NewMultiplier(types.Large, 24, sdk.OneDec())}),
},
types.DelegatorDistributionSchedules{types.NewDelegatorDistributionSchedule(
types.NewDistributionSchedule(true, "bnb", time.Date(2020, 10, 8, 14, 0, 0, 0, time.UTC), time.Date(2025, 10, 8, 14, 0, 0, 0, time.UTC), sdk.NewCoin("hard", sdk.NewInt(500)), time.Date(2026, 10, 8, 14, 0, 0, 0, time.UTC), types.Multipliers{types.NewMultiplier(types.Small, 0, sdk.MustNewDecFromStr("0.33")), types.NewMultiplier(types.Medium, 6, sdk.MustNewDecFromStr("0.5")), types.NewMultiplier(types.Large, 24, sdk.OneDec())}),
time.Hour*24,
),
},
), types.DefaultPreviousBlockTime, types.DefaultDistributionTimes)
tApp.InitializeFromGenesisStates(authGS, app.GenesisState{types.ModuleName: types.ModuleCdc.MustMarshalJSON(harvestGS)})
if tc.args.accArgs.vestingAccountBefore {
ak := tApp.GetAccountKeeper()
acc := ak.GetAccount(ctx, tc.args.accArgs.addr)
bacc := auth.NewBaseAccount(acc.GetAddress(), acc.GetCoins(), acc.GetPubKey(), acc.GetAccountNumber(), acc.GetSequence())
bva, err := vesting.NewBaseVestingAccount(bacc, tc.args.accArgs.origVestingCoins, tc.args.accArgs.endTime)
suite.Require().NoError(err)
pva := vesting.NewPeriodicVestingAccountRaw(bva, tc.args.accArgs.startTime, tc.args.accArgs.periods)
ak.SetAccount(ctx, pva)
}
supplyKeeper := tApp.GetSupplyKeeper()
supplyKeeper.MintCoins(ctx, types.LPAccount, sdk.NewCoins(sdk.NewCoin("hard", sdk.NewInt(1000))))
keeper := tApp.GetHarvestKeeper()
suite.app = tApp
suite.ctx = ctx
suite.keeper = keeper
err := suite.keeper.SendTimeLockedCoinsToAccount(suite.ctx, types.LPAccount, tc.args.accArgs.addr, tc.args.period.Amount, tc.args.period.Length)
if tc.errArgs.expectPass {
suite.Require().NoError(err)
acc := suite.getAccount(tc.args.accArgs.addr)
suite.Require().Equal(tc.args.expectedAccountBalance, acc.GetCoins())
mAcc := suite.getModuleAccount(types.LPAccount)
suite.Require().Equal(tc.args.expectedModAccountBalance, mAcc.GetCoins())
vacc, ok := acc.(*vesting.PeriodicVestingAccount)
if tc.args.accArgs.vestingAccountAfter {
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)
} else {
suite.Require().False(ok)
}
}
})
}
}
func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) }
func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) }
Reference in New Issue View Git Blame Copy Permalink