fix: set previous block time correctly on block one (#868)

* fix: set previous block time correctly on block one

* fix failing tests

x/validator-vesting/abci.go

@@ -2,9 +2,6 @@ package validatorvesting
 
 import (
 	"bytes"
-	"time"
-
-	tmtime "github.com/tendermint/tendermint/types/time"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
@@ -15,9 +12,10 @@ import (
 
 // BeginBlocker updates the vote signing information for each validator vesting account, updates account when period changes, and updates the previousBlockTime value in the store.
 func BeginBlocker(ctx sdk.Context, req abci.RequestBeginBlock, k keeper.Keeper) {
-	previousBlockTime := tmtime.Canonical(time.Unix(0, 0))
+	previousBlockTime, found := k.GetPreviousBlockTime(ctx)
-	if ctx.BlockHeight() > 1 {
+	if !found {
-		previousBlockTime = k.GetPreviousBlockTime(ctx)
+		k.SetPreviousBlockTime(ctx, ctx.BlockTime())
+		return
 	}
 
 	currentBlockTime := ctx.BlockTime()

x/validator-vesting/abci_test.go

@@ -69,7 +69,7 @@ func TestBeginBlockerZeroHeight(t *testing.T) {
 			}},
 		},
 	}
-
+	vvk.SetPreviousBlockTime(ctx, now)
 	BeginBlocker(ctx, req, vvk)
 
 	vva = vvk.GetAccountFromAuthKeeper(ctx, vva.Address)
@@ -147,6 +147,7 @@ func TestBeginBlockerSignedBlock(t *testing.T) {
 		},
 	}
 
+	vvk.SetPreviousBlockTime(ctx, now)
 	BeginBlocker(ctx, req, vvk)
 	height++
 	blockTime = addHour(blockTime)

x/validator-vesting/genesis.go

@@ -22,6 +22,9 @@ func InitGenesis(ctx sdk.Context, keeper Keeper, accountKeeper types.AccountKeep
 
 // ExportGenesis returns empty genesis state because auth exports all the genesis state we need.
 func ExportGenesis(ctx sdk.Context, keeper Keeper) types.GenesisState {
-	prevBlockTime := keeper.GetPreviousBlockTime(ctx)
+	prevBlockTime, found := keeper.GetPreviousBlockTime(ctx)
+	if !found {
+		prevBlockTime = ctx.BlockTime()
+	}
 	return NewGenesisState(prevBlockTime)
 }

x/validator-vesting/keeper/keeper.go

@@ -43,14 +43,14 @@ func (k Keeper) Logger(ctx sdk.Context) log.Logger {
 }
 
 // GetPreviousBlockTime get the blocktime for the previous block
-func (k Keeper) GetPreviousBlockTime(ctx sdk.Context) (blockTime time.Time) {
+func (k Keeper) GetPreviousBlockTime(ctx sdk.Context) (blockTime time.Time, found bool) {
 	store := ctx.KVStore(k.storeKey)
 	b := store.Get(types.BlocktimeKey)
 	if b == nil {
-		panic("Previous block time not set")
+		return time.Time{}, false
 	}
 	k.cdc.MustUnmarshalBinaryLengthPrefixed(b, &blockTime)
-	return blockTime
+	return blockTime, true
 }
 
 // SetPreviousBlockTime set the time of the previous block

x/validator-vesting/keeper/keeper_test.go

@@ -67,19 +67,22 @@ func TestGetSetPreviousBlock(t *testing.T) {
 	now := tmtime.Now()
 
 	// require panic if the previous blocktime was never set
-	require.Panics(t, func() { keeper.GetPreviousBlockTime(ctx) })
+	_, found := keeper.GetPreviousBlockTime(ctx)
+	require.False(t, found)
 
 	// require that passing a valid time to SetPreviousBlockTime does not panic
 	require.NotPanics(t, func() { keeper.SetPreviousBlockTime(ctx, now) })
 
 	// require that the value from GetPreviousBlockTime equals what was set
-	bpt := keeper.GetPreviousBlockTime(ctx)
+	bpt, found := keeper.GetPreviousBlockTime(ctx)
+	require.True(t, found)
 	require.Equal(t, now, bpt)
 
 	// require that the zero value is safe
 	require.NotPanics(t, func() { keeper.SetPreviousBlockTime(ctx, tmtime.Canonical(time.Unix(0, 0))) })
 
-	bpt = keeper.GetPreviousBlockTime(ctx)
+	bpt, found = keeper.GetPreviousBlockTime(ctx)
+	require.True(t, found)
 	require.Equal(t, tmtime.Canonical(time.Unix(0, 0)), bpt)
 
 }