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 { func (k Keeper) AccountIsVesting(ctx sdk.Context, addr sdk.AccAddress) bool {
vv := k.GetAccountFromAuthKeeper(ctx, addr) vv := k.GetAccountFromAuthKeeper(ctx, addr)
if !vv.DebtAfterFailedVesting.IsZero() { if !vv.DebtAfterFailedVesting.IsZero() {
return false return true
} }
for _, p := range vv.VestingPeriodProgress { for _, p := range vv.VestingPeriodProgress {
if !p.PeriodComplete { 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) ak.SetAccount(ctx, vva)
keeper.SetValidatorVestingAccountKey(ctx, vva.Address) 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 { for i := range vva.VestingPeriodProgress {
vva.VestingPeriodProgress[i] = types.VestingProgress{true, true} vva.VestingPeriodProgress[i] = types.VestingProgress{true, true}
ak.SetAccount(ctx, vva) ak.SetAccount(ctx, vva)
} }
require.Equal(t, true, keeper.AccountIsVesting(ctx, vva.Address)) require.Equal(t, false, keeper.AccountIsVesting(ctx, vva.Address))
} }