fixed linter

Author: sstanculeanu

epochStart/mock/rounderStub.go

@@ -8,11 +8,12 @@ import (
 type RoundHandlerStub struct {
 	RoundIndex int64
 
-	IndexCalled         func() int64
-	TimeDurationCalled  func() time.Duration
-	TimeStampCalled     func() time.Time
-	UpdateRoundCalled   func(time.Time, time.Time)
-	RemainingTimeCalled func(startTime time.Time, maxTime time.Duration) time.Duration
+	IndexCalled                func() int64
+	TimeDurationCalled         func() time.Duration
+	TimeStampCalled            func() time.Time
+	UpdateRoundCalled          func(time.Time, time.Time)
+	RemainingTimeCalled        func(startTime time.Time, maxTime time.Duration) time.Duration
+	GetTimeStampForRoundCalled func(round uint64) uint64
 }
 
 // Index -
@@ -61,6 +62,15 @@ func (rndm *RoundHandlerStub) RemainingTime(startTime time.Time, maxTime time.Du
 	return 4000 * time.Millisecond
 }
 
+// GetTimeStampForRound -
+func (rndm *RoundHandlerStub) GetTimeStampForRound(round uint64) uint64 {
+	if rndm.GetTimeStampForRoundCalled != nil {
+		return rndm.GetTimeStampForRoundCalled(round)
+	}
+
+	return uint64(time.Unix(0, 0).UnixMilli())
+}
+
 // IsInterfaceNil returns true if there is no value under the interface
 func (rndm *RoundHandlerStub) IsInterfaceNil() bool {
 	return rndm == nil

process/mock/roundStub.go

@@ -6,11 +6,12 @@ import (
 
 // RoundStub -
 type RoundStub struct {
-	IndexCalled         func() int64
-	TimeDurationCalled  func() time.Duration
-	TimeStampCalled     func() time.Time
-	UpdateRoundCalled   func(time.Time, time.Time)
-	RemainingTimeCalled func(time.Time, time.Duration) time.Duration
+	IndexCalled                func() int64
+	TimeDurationCalled         func() time.Duration
+	TimeStampCalled            func() time.Time
+	UpdateRoundCalled          func(time.Time, time.Time)
+	RemainingTimeCalled        func(time.Time, time.Duration) time.Duration
+	GetTimeStampForRoundCalled func(round uint64) uint64
 }
 
 // Index -
@@ -38,6 +39,15 @@ func (rnds *RoundStub) RemainingTime(startTime time.Time, maxTime time.Duration)
 	return rnds.RemainingTimeCalled(startTime, maxTime)
 }
 
+// GetTimeStampForRound -
+func (rnds *RoundStub) GetTimeStampForRound(round uint64) uint64 {
+	if rnds.GetTimeStampForRoundCalled != nil {
+		return rnds.GetTimeStampForRoundCalled(round)
+	}
+
+	return uint64(time.Unix(0, 0).UnixMilli())
+}
+
 // IsInterfaceNil --
 func (rnds *RoundStub) IsInterfaceNil() bool {
 	return rnds == nil