adjust tests more

proto/downtime/v1/downtime.proto

@@ -13,6 +13,8 @@ message Params {
   // downtime_duration defines the minimum time elapsed between blocks
   // that we consider the chain to be down.
   google.protobuf.Duration downtime_duration = 1 [(gogoproto.nullable) = false, (gogoproto.stdduration) = true];
+  // epoch_duration defines an estimation of the time elapsed between epochs
+  google.protobuf.Duration epoch_duration = 2 [(gogoproto.nullable) = false, (gogoproto.stdduration) = true];
 }
 
 // Downtime defines a single downtime record.

x/downtime/keeper/keeper.go

@@ -191,6 +191,15 @@ func (k Keeper) GarbageCollectDowntimes(ctx sdk.Context) {
 	}
 }
 
+func (k Keeper) GetDowntimeFactor(ctx sdk.Context, epoch uint64) uint64 {
+	duration, exists := k.GetDowntime(ctx, epoch)
+	if !exists {
+		return 1
+	}
+	epochDuration := k.GetParams(ctx).EpochDuration
+	return uint64(duration/epochDuration) + 1
+}
+
 func (k Keeper) BeginBlock(ctx sdk.Context) {
 	// this ensures that no matter the outcome, we will
 	// reset the last block time to the current block time.

x/downtime/keeper/keeper_test.go

@@ -116,6 +116,16 @@ func TestBeginBlock(t *testing.T) {
 	require.True(t, hadDowntimes)
 	require.Equal(t, 2*keeper.GetParams(ctx).DowntimeDuration, duration)
 
+	// check downtime factors
+	factor := keeper.GetDowntimeFactor(ctx, epochStartBlock(ctx))
+	require.Equal(t, uint64(1), factor)
+
+	// now let's advance the block until we have 1 entire epoch of downtime
+	ctx = nextBlock(ctx, keeper.GetParams(ctx).EpochDuration-duration)
+	keeper.BeginBlock(ctx)
+	factor = keeper.GetDowntimeFactor(ctx, epochStartBlock(ctx))
+	require.Equal(t, uint64(2), factor)
+
 	// check garbage collection was done, after forcing an epoch to pass
 	for !app.EpochstorageKeeper.IsEpochStart(ctx) {
 		ctx = nextBlock(ctx, 1*time.Second)

x/downtime/v1/params.go

@@ -9,12 +9,15 @@ import (
 
 var (
 	ParamKeyDowntimeDuration        = []byte("DowntimeDuration")
-	DefaultParamKeyDowntimeDuration = 30 * time.Minute
+	DefaultParamKeyDowntimeDuration = 5 * time.Minute
+	ParamKeyEpochDuration           = []byte("EpochDuration")
+	DefaultParamKeyEpochDuration    = 30 * time.Minute
 )
 
 func DefaultParams() Params {
 	return Params{
 		DowntimeDuration: DefaultParamKeyDowntimeDuration,
+		EpochDuration:    DefaultParamKeyEpochDuration,
 	}
 }
 
@@ -23,13 +26,17 @@ var _ paramtypes.ParamSet = (*Params)(nil)
 func (m *Params) ParamSetPairs() paramtypes.ParamSetPairs {
 	return paramtypes.ParamSetPairs{
 		paramtypes.NewParamSetPair(ParamKeyDowntimeDuration, &m.DowntimeDuration, validateDowntimeDuration),
+		paramtypes.NewParamSetPair(ParamKeyEpochDuration, &m.EpochDuration, validateDowntimeDuration),
 	}
 }
 
 func (m *Params) Validate() error {
 	if err := validateDowntimeDuration(m.DowntimeDuration); err != nil {
 		return err
 	}
+	if err := validateDowntimeDuration(m.EpochDuration); err != nil {
+		return err
+	}
 	return nil
 }