refactor TestIncrementsMsgUnbond
This commit is contained in:
parent
a07ed83dd2
commit
46d62ebfa9
|
@ -456,115 +456,116 @@ func TestEditValidatorIncreaseMinSelfDelegationBeyondCurrentBond(t *testing.T) {
|
||||||
require.Nil(t, res)
|
require.Nil(t, res)
|
||||||
}
|
}
|
||||||
|
|
||||||
//func TestIncrementsMsgUnbond(t *testing.T) {
|
func TestIncrementsMsgUnbond(t *testing.T) {
|
||||||
// initPower := int64(1000)
|
initPower := int64(1000)
|
||||||
// initBond := sdk.TokensFromConsensusPower(initPower)
|
initBond := sdk.TokensFromConsensusPower(initPower)
|
||||||
// ctx, _, bk, keeper, _ := CreateTestInput(t, false, initPower)
|
|
||||||
// handler := NewHandler(keeper)
|
app, ctx, delAddrs, valAddrs := bootstrapHandlerGenesisTest(t, initPower, 2, 1000000000)
|
||||||
//
|
handler := staking.NewHandler(app.StakingKeeper)
|
||||||
// params := keeper.GetParams(ctx)
|
|
||||||
// denom := params.BondDenom
|
params := app.StakingKeeper.GetParams(ctx)
|
||||||
//
|
denom := params.BondDenom
|
||||||
// // create validator, delegate
|
|
||||||
// validatorAddr, delegatorAddr := sdk.ValAddress(Addrs[0]), Addrs[1]
|
// create validator, delegate
|
||||||
//
|
validatorAddr, delegatorAddr := valAddrs[0], delAddrs[1]
|
||||||
// msgCreateValidator := NewTestMsgCreateValidator(validatorAddr, PKs[0], initBond)
|
|
||||||
// res, err := handler(ctx, msgCreateValidator)
|
msgCreateValidator := NewTestMsgCreateValidator(validatorAddr, PKs[0], initBond)
|
||||||
// require.NoError(t, err)
|
res, err := handler(ctx, msgCreateValidator)
|
||||||
// require.NotNil(t, res)
|
require.NoError(t, err)
|
||||||
//
|
require.NotNil(t, res)
|
||||||
// // initial balance
|
|
||||||
// amt1 := bk.GetBalance(ctx, delegatorAddr, denom).Amount
|
// initial balance
|
||||||
//
|
amt1 := app.BankKeeper.GetBalance(ctx, delegatorAddr, denom).Amount
|
||||||
// msgDelegate := NewTestMsgDelegate(delegatorAddr, validatorAddr, initBond)
|
|
||||||
// res, err = handler(ctx, msgDelegate)
|
msgDelegate := NewTestMsgDelegate(delegatorAddr, validatorAddr, initBond)
|
||||||
// require.NoError(t, err)
|
res, err = handler(ctx, msgDelegate)
|
||||||
// require.NotNil(t, res)
|
require.NoError(t, err)
|
||||||
//
|
require.NotNil(t, res)
|
||||||
// // balance should have been subtracted after delegation
|
|
||||||
// amt2 := bk.GetBalance(ctx, delegatorAddr, denom).Amount
|
// balance should have been subtracted after delegation
|
||||||
// require.True(sdk.IntEq(t, amt1.Sub(initBond), amt2))
|
amt2 := app.BankKeeper.GetBalance(ctx, delegatorAddr, denom).Amount
|
||||||
//
|
require.True(sdk.IntEq(t, amt1.Sub(initBond), amt2))
|
||||||
// // apply TM updates
|
|
||||||
// keeper.ApplyAndReturnValidatorSetUpdates(ctx)
|
// apply TM updates
|
||||||
//
|
app.StakingKeeper.ApplyAndReturnValidatorSetUpdates(ctx)
|
||||||
// validator, found := keeper.GetValidator(ctx, validatorAddr)
|
|
||||||
// require.True(t, found)
|
validator, found := app.StakingKeeper.GetValidator(ctx, validatorAddr)
|
||||||
// require.Equal(t, initBond.MulRaw(2), validator.DelegatorShares.RoundInt())
|
require.True(t, found)
|
||||||
// require.Equal(t, initBond.MulRaw(2), validator.BondedTokens())
|
require.Equal(t, initBond.MulRaw(2), validator.DelegatorShares.RoundInt())
|
||||||
//
|
require.Equal(t, initBond.MulRaw(2), validator.BondedTokens())
|
||||||
// // just send the same msgUnbond multiple times
|
|
||||||
// // TODO use decimals here
|
// just send the same msgUnbond multiple times
|
||||||
// unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(10))
|
// TODO use decimals here
|
||||||
// msgUndelegate := NewMsgUndelegate(delegatorAddr, validatorAddr, unbondAmt)
|
unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(10))
|
||||||
// numUnbonds := int64(5)
|
msgUndelegate := types.NewMsgUndelegate(delegatorAddr, validatorAddr, unbondAmt)
|
||||||
//
|
numUnbonds := int64(5)
|
||||||
// for i := int64(0); i < numUnbonds; i++ {
|
|
||||||
// res, err := handler(ctx, msgUndelegate)
|
for i := int64(0); i < numUnbonds; i++ {
|
||||||
// require.NoError(t, err)
|
res, err := handler(ctx, msgUndelegate)
|
||||||
// require.NotNil(t, res)
|
require.NoError(t, err)
|
||||||
//
|
require.NotNil(t, res)
|
||||||
// ts := &gogotypes.Timestamp{}
|
|
||||||
// types.ModuleCdc.MustUnmarshalBinaryLengthPrefixed(res.Data, ts)
|
ts := &gogotypes.Timestamp{}
|
||||||
//
|
types.ModuleCdc.MustUnmarshalBinaryLengthPrefixed(res.Data, ts)
|
||||||
// finishTime, err := gogotypes.TimestampFromProto(ts)
|
|
||||||
// require.NoError(t, err)
|
finishTime, err := gogotypes.TimestampFromProto(ts)
|
||||||
//
|
require.NoError(t, err)
|
||||||
// ctx = ctx.WithBlockTime(finishTime)
|
|
||||||
// EndBlocker(ctx, keeper)
|
ctx = ctx.WithBlockTime(finishTime)
|
||||||
//
|
staking.EndBlocker(ctx, app.StakingKeeper)
|
||||||
// // check that the accounts and the bond account have the appropriate values
|
|
||||||
// validator, found = keeper.GetValidator(ctx, validatorAddr)
|
// check that the accounts and the bond account have the appropriate values
|
||||||
// require.True(t, found)
|
validator, found = app.StakingKeeper.GetValidator(ctx, validatorAddr)
|
||||||
// bond, found := keeper.GetDelegation(ctx, delegatorAddr, validatorAddr)
|
require.True(t, found)
|
||||||
// require.True(t, found)
|
bond, found := app.StakingKeeper.GetDelegation(ctx, delegatorAddr, validatorAddr)
|
||||||
//
|
require.True(t, found)
|
||||||
// expBond := initBond.Sub(unbondAmt.Amount.Mul(sdk.NewInt(i + 1)))
|
|
||||||
// expDelegatorShares := initBond.MulRaw(2).Sub(unbondAmt.Amount.Mul(sdk.NewInt(i + 1)))
|
expBond := initBond.Sub(unbondAmt.Amount.Mul(sdk.NewInt(i + 1)))
|
||||||
// expDelegatorAcc := initBond.Sub(expBond)
|
expDelegatorShares := initBond.MulRaw(2).Sub(unbondAmt.Amount.Mul(sdk.NewInt(i + 1)))
|
||||||
//
|
expDelegatorAcc := initBond.Sub(expBond)
|
||||||
// gotBond := bond.Shares.RoundInt()
|
|
||||||
// gotDelegatorShares := validator.DelegatorShares.RoundInt()
|
gotBond := bond.Shares.RoundInt()
|
||||||
// gotDelegatorAcc := bk.GetBalance(ctx, delegatorAddr, params.BondDenom).Amount
|
gotDelegatorShares := validator.DelegatorShares.RoundInt()
|
||||||
//
|
gotDelegatorAcc := app.BankKeeper.GetBalance(ctx, delegatorAddr, params.BondDenom).Amount
|
||||||
// require.Equal(t, expBond.Int64(), gotBond.Int64(),
|
|
||||||
// "i: %v\nexpBond: %v\ngotBond: %v\nvalidator: %v\nbond: %v\n",
|
require.Equal(t, expBond.Int64(), gotBond.Int64(),
|
||||||
// i, expBond, gotBond, validator, bond)
|
"i: %v\nexpBond: %v\ngotBond: %v\nvalidator: %v\nbond: %v\n",
|
||||||
// require.Equal(t, expDelegatorShares.Int64(), gotDelegatorShares.Int64(),
|
i, expBond, gotBond, validator, bond)
|
||||||
// "i: %v\nexpDelegatorShares: %v\ngotDelegatorShares: %v\nvalidator: %v\nbond: %v\n",
|
require.Equal(t, expDelegatorShares.Int64(), gotDelegatorShares.Int64(),
|
||||||
// i, expDelegatorShares, gotDelegatorShares, validator, bond)
|
"i: %v\nexpDelegatorShares: %v\ngotDelegatorShares: %v\nvalidator: %v\nbond: %v\n",
|
||||||
// require.Equal(t, expDelegatorAcc.Int64(), gotDelegatorAcc.Int64(),
|
i, expDelegatorShares, gotDelegatorShares, validator, bond)
|
||||||
// "i: %v\nexpDelegatorAcc: %v\ngotDelegatorAcc: %v\nvalidator: %v\nbond: %v\n",
|
require.Equal(t, expDelegatorAcc.Int64(), gotDelegatorAcc.Int64(),
|
||||||
// i, expDelegatorAcc, gotDelegatorAcc, validator, bond)
|
"i: %v\nexpDelegatorAcc: %v\ngotDelegatorAcc: %v\nvalidator: %v\nbond: %v\n",
|
||||||
// }
|
i, expDelegatorAcc, gotDelegatorAcc, validator, bond)
|
||||||
//
|
}
|
||||||
// // these are more than we have bonded now
|
|
||||||
// errorCases := []sdk.Int{
|
// these are more than we have bonded now
|
||||||
// //1<<64 - 1, // more than int64 power
|
errorCases := []sdk.Int{
|
||||||
// //1<<63 + 1, // more than int64 power
|
//1<<64 - 1, // more than int64 power
|
||||||
// sdk.TokensFromConsensusPower(1<<63 - 1),
|
//1<<63 + 1, // more than int64 power
|
||||||
// sdk.TokensFromConsensusPower(1 << 31),
|
sdk.TokensFromConsensusPower(1<<63 - 1),
|
||||||
// initBond,
|
sdk.TokensFromConsensusPower(1 << 31),
|
||||||
// }
|
initBond,
|
||||||
//
|
}
|
||||||
// for _, c := range errorCases {
|
|
||||||
// unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, c)
|
for _, c := range errorCases {
|
||||||
// msgUndelegate := NewMsgUndelegate(delegatorAddr, validatorAddr, unbondAmt)
|
unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, c)
|
||||||
// res, err = handler(ctx, msgUndelegate)
|
msgUndelegate := types.NewMsgUndelegate(delegatorAddr, validatorAddr, unbondAmt)
|
||||||
// require.Error(t, err)
|
res, err = handler(ctx, msgUndelegate)
|
||||||
// require.Nil(t, res)
|
require.Error(t, err)
|
||||||
// }
|
require.Nil(t, res)
|
||||||
//
|
}
|
||||||
// leftBonded := initBond.Sub(unbondAmt.Amount.Mul(sdk.NewInt(numUnbonds)))
|
|
||||||
//
|
leftBonded := initBond.Sub(unbondAmt.Amount.Mul(sdk.NewInt(numUnbonds)))
|
||||||
// // should be able to unbond remaining
|
|
||||||
// unbondAmt = sdk.NewCoin(sdk.DefaultBondDenom, leftBonded)
|
// should be able to unbond remaining
|
||||||
// msgUndelegate = NewMsgUndelegate(delegatorAddr, validatorAddr, unbondAmt)
|
unbondAmt = sdk.NewCoin(sdk.DefaultBondDenom, leftBonded)
|
||||||
// res, err = handler(ctx, msgUndelegate)
|
msgUndelegate = types.NewMsgUndelegate(delegatorAddr, validatorAddr, unbondAmt)
|
||||||
// require.NoError(t, err, "msgUnbond: %v\nshares: %s\nleftBonded: %s\n", msgUndelegate, unbondAmt, leftBonded)
|
res, err = handler(ctx, msgUndelegate)
|
||||||
// require.NotNil(t, res, "msgUnbond: %v\nshares: %s\nleftBonded: %s\n", msgUndelegate, unbondAmt, leftBonded)
|
require.NoError(t, err, "msgUnbond: %v\nshares: %s\nleftBonded: %s\n", msgUndelegate, unbondAmt, leftBonded)
|
||||||
//}
|
require.NotNil(t, res, "msgUnbond: %v\nshares: %s\nleftBonded: %s\n", msgUndelegate, unbondAmt, leftBonded)
|
||||||
//
|
}
|
||||||
|
|
||||||
//func TestMultipleMsgCreateValidator(t *testing.T) {
|
//func TestMultipleMsgCreateValidator(t *testing.T) {
|
||||||
// initPower := int64(1000)
|
// initPower := int64(1000)
|
||||||
// initTokens := sdk.TokensFromConsensusPower(initPower)
|
// initTokens := sdk.TokensFromConsensusPower(initPower)
|
||||||
|
|
Loading…
Reference in New Issue