fix: get vesting status return values

This commit is contained in:
Kevin Davis 2019-10-10 10:59:05 -04:00
parent e05ea73da0
commit 37bd179c8e
2 changed files with 5 additions and 5 deletions

View File

@ -209,12 +209,12 @@ func (k Keeper) GetPeriodEndTimes(ctx sdk.Context, addr sdk.AccAddress) []int64
func (k Keeper) AccountIsVesting(ctx sdk.Context, addr sdk.AccAddress) bool {
vv := k.GetAccountFromAuthKeeper(ctx, addr)
if !vv.DebtAfterFailedVesting.IsZero() {
return false
return true
}
for _, p := range vv.VestingPeriodProgress {
if !p.PeriodComplete {
return false
return true
}
}
return true
return false
}

View File

@ -111,13 +111,13 @@ func TestAccountIsVesting(t *testing.T) {
ak.SetAccount(ctx, vva)
keeper.SetValidatorVestingAccountKey(ctx, vva.Address)
require.Equal(t, false, keeper.AccountIsVesting(ctx, vva.Address))
require.Equal(t, true, keeper.AccountIsVesting(ctx, vva.Address))
for i := range vva.VestingPeriodProgress {
vva.VestingPeriodProgress[i] = types.VestingProgress{true, true}
ak.SetAccount(ctx, vva)
}
require.Equal(t, true, keeper.AccountIsVesting(ctx, vva.Address))
require.Equal(t, false, keeper.AccountIsVesting(ctx, vva.Address))
}