refactor TestUnbondingPeriod and TestValidatorQueue

This commit is contained in:
Jonathan Gimeno 2020-02-26 15:26:50 +01:00
parent 0fcf1e20d6
commit eb26b478b4
1 changed files with 114 additions and 114 deletions

View File

@ -756,120 +756,120 @@ func TestJailValidator(t *testing.T) {
require.NotNil(t, res) require.NotNil(t, res)
} }
//func TestValidatorQueue(t *testing.T) { func TestValidatorQueue(t *testing.T) {
// ctx, _, _, keeper, _ := CreateTestInput(t, false, 1000) app, ctx, delAddrs, valAddrs := bootstrapHandlerGenesisTest(t, 1000, 2, 1000000000)
// handler := NewHandler(keeper) handler := staking.NewHandler(app.StakingKeeper)
// validatorAddr, delegatorAddr := sdk.ValAddress(Addrs[0]), Addrs[1] validatorAddr, delegatorAddr := valAddrs[0], delAddrs[1]
//
// // set the unbonding time // set the unbonding time
// params := keeper.GetParams(ctx) params := app.StakingKeeper.GetParams(ctx)
// params.UnbondingTime = 7 * time.Second params.UnbondingTime = 7 * time.Second
// keeper.SetParams(ctx, params) app.StakingKeeper.SetParams(ctx, params)
//
// // create the validator // create the validator
// valTokens := sdk.TokensFromConsensusPower(10) valTokens := sdk.TokensFromConsensusPower(10)
// msgCreateValidator := NewTestMsgCreateValidator(validatorAddr, PKs[0], valTokens) msgCreateValidator := NewTestMsgCreateValidator(validatorAddr, PKs[0], valTokens)
// res, err := handler(ctx, msgCreateValidator) res, err := handler(ctx, msgCreateValidator)
// require.NoError(t, err) require.NoError(t, err)
// require.NotNil(t, res) require.NotNil(t, res)
//
// // bond a delegator // bond a delegator
// delTokens := sdk.TokensFromConsensusPower(10) delTokens := sdk.TokensFromConsensusPower(10)
// msgDelegate := NewTestMsgDelegate(delegatorAddr, validatorAddr, delTokens) msgDelegate := NewTestMsgDelegate(delegatorAddr, validatorAddr, delTokens)
// res, err = handler(ctx, msgDelegate) res, err = handler(ctx, msgDelegate)
// require.NoError(t, err) require.NoError(t, err)
// require.NotNil(t, res) require.NotNil(t, res)
//
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
//
// // unbond the all self-delegation to put validator in unbonding state // unbond the all self-delegation to put validator in unbonding state
// unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, delTokens) unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, delTokens)
// msgUndelegateValidator := NewMsgUndelegate(sdk.AccAddress(validatorAddr), validatorAddr, unbondAmt) msgUndelegateValidator := types.NewMsgUndelegate(sdk.AccAddress(validatorAddr), validatorAddr, unbondAmt)
// res, err = handler(ctx, msgUndelegateValidator) res, err = handler(ctx, msgUndelegateValidator)
// require.NoError(t, err) require.NoError(t, err)
// require.NotNil(t, res) require.NotNil(t, res)
//
// ts := &gogotypes.Timestamp{} ts := &gogotypes.Timestamp{}
// types.ModuleCdc.MustUnmarshalBinaryLengthPrefixed(res.Data, ts) types.ModuleCdc.MustUnmarshalBinaryLengthPrefixed(res.Data, ts)
//
// finishTime, err := gogotypes.TimestampFromProto(ts) finishTime, err := gogotypes.TimestampFromProto(ts)
// require.NoError(t, err) require.NoError(t, err)
//
// ctx = ctx.WithBlockTime(finishTime) ctx = ctx.WithBlockTime(finishTime)
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
//
// origHeader := ctx.BlockHeader() origHeader := ctx.BlockHeader()
//
// validator, found := keeper.GetValidator(ctx, validatorAddr) validator, found := app.StakingKeeper.GetValidator(ctx, validatorAddr)
// require.True(t, found) require.True(t, found)
// require.True(t, validator.IsUnbonding(), "%v", validator) require.True(t, validator.IsUnbonding(), "%v", validator)
//
// // should still be unbonding at time 6 seconds later // should still be unbonding at time 6 seconds later
// ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 6)) ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 6))
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
//
// validator, found = keeper.GetValidator(ctx, validatorAddr) validator, found = app.StakingKeeper.GetValidator(ctx, validatorAddr)
// require.True(t, found) require.True(t, found)
// require.True(t, validator.IsUnbonding(), "%v", validator) require.True(t, validator.IsUnbonding(), "%v", validator)
//
// // should be in unbonded state at time 7 seconds later // should be in unbonded state at time 7 seconds later
// ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 7)) ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 7))
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
//
// validator, found = keeper.GetValidator(ctx, validatorAddr) validator, found = app.StakingKeeper.GetValidator(ctx, validatorAddr)
// require.True(t, found) require.True(t, found)
// require.True(t, validator.IsUnbonded(), "%v", validator) require.True(t, validator.IsUnbonded(), "%v", validator)
//} }
//
//func TestUnbondingPeriod(t *testing.T) { func TestUnbondingPeriod(t *testing.T) {
// ctx, _, _, keeper, _ := CreateTestInput(t, false, 1000) app, ctx, _, valAddrs := bootstrapHandlerGenesisTest(t, 1000, 1, 1000000000)
// handler := NewHandler(keeper) handler := staking.NewHandler(app.StakingKeeper)
// validatorAddr := sdk.ValAddress(Addrs[0]) validatorAddr := valAddrs[0]
//
// // set the unbonding time // set the unbonding time
// params := keeper.GetParams(ctx) params := app.StakingKeeper.GetParams(ctx)
// params.UnbondingTime = 7 * time.Second params.UnbondingTime = 7 * time.Second
// keeper.SetParams(ctx, params) app.StakingKeeper.SetParams(ctx, params)
//
// // create the validator // create the validator
// valTokens := sdk.TokensFromConsensusPower(10) valTokens := sdk.TokensFromConsensusPower(10)
// msgCreateValidator := NewTestMsgCreateValidator(validatorAddr, PKs[0], valTokens) msgCreateValidator := NewTestMsgCreateValidator(validatorAddr, PKs[0], valTokens)
// res, err := handler(ctx, msgCreateValidator) res, err := handler(ctx, msgCreateValidator)
// require.NoError(t, err) require.NoError(t, err)
// require.NotNil(t, res) require.NotNil(t, res)
//
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
//
// // begin unbonding // begin unbonding
// unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, sdk.TokensFromConsensusPower(10)) unbondAmt := sdk.NewCoin(sdk.DefaultBondDenom, sdk.TokensFromConsensusPower(10))
// msgUndelegate := NewMsgUndelegate(sdk.AccAddress(validatorAddr), validatorAddr, unbondAmt) msgUndelegate := types.NewMsgUndelegate(sdk.AccAddress(validatorAddr), validatorAddr, unbondAmt)
// res, err = handler(ctx, msgUndelegate) res, err = handler(ctx, msgUndelegate)
// require.NoError(t, err) require.NoError(t, err)
// require.NotNil(t, res) require.NotNil(t, res)
//
// origHeader := ctx.BlockHeader() origHeader := ctx.BlockHeader()
//
// _, found := keeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr) _, found := app.StakingKeeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr)
// require.True(t, found, "should not have unbonded") require.True(t, found, "should not have unbonded")
//
// // cannot complete unbonding at same time // cannot complete unbonding at same time
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
// _, found = keeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr) _, found = app.StakingKeeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr)
// require.True(t, found, "should not have unbonded") require.True(t, found, "should not have unbonded")
//
// // cannot complete unbonding at time 6 seconds later // cannot complete unbonding at time 6 seconds later
// ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 6)) ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 6))
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
// _, found = keeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr) _, found = app.StakingKeeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr)
// require.True(t, found, "should not have unbonded") require.True(t, found, "should not have unbonded")
//
// // can complete unbonding at time 7 seconds later // can complete unbonding at time 7 seconds later
// ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 7)) ctx = ctx.WithBlockTime(origHeader.Time.Add(time.Second * 7))
// EndBlocker(ctx, keeper) staking.EndBlocker(ctx, app.StakingKeeper)
// _, found = keeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr) _, found = app.StakingKeeper.GetUnbondingDelegation(ctx, sdk.AccAddress(validatorAddr), validatorAddr)
// require.False(t, found, "should have unbonded") require.False(t, found, "should have unbonded")
//} }
//
//func TestUnbondingFromUnbondingValidator(t *testing.T) { //func TestUnbondingFromUnbondingValidator(t *testing.T) {
// ctx, _, _, keeper, _ := CreateTestInput(t, false, 1000) // ctx, _, _, keeper, _ := CreateTestInput(t, false, 1000)
// handler := NewHandler(keeper) // handler := NewHandler(keeper)