Adjusted existing unit tests for heartbeats

keep-network · May 14, 2024 · 82dc68a · 82dc68a
1 parent 4e32ec6
commit 82dc68a
Show file tree

Hide file tree

Showing 2 changed files with 319 additions and 20 deletions.
diff --git a/pkg/tbtc/heartbeat.go b/pkg/tbtc/heartbeat.go
@@ -63,6 +63,18 @@ type heartbeatSigningExecutor interface {
 	) (*tecdsa.Signature, uint32, uint64, error)
 }
 
+// heartbeatInactivityClaimExecutor is an interface meant to decouple the
+// specific implementation of the inactivity claim executor from the heartbeat
+// action.
+type heartbeatInactivityClaimExecutor interface {
+	claimInactivity(
+		ctx context.Context,
+		inactiveMembersIndexes []group.MemberIndex,
+		heartbeatFailed bool,
+		sessionID *big.Int,
+	) error
+}
+
 // heartbeatAction is a walletAction implementation handling heartbeat requests
 // from the wallet coordinator.
 type heartbeatAction struct {
@@ -75,7 +87,7 @@ type heartbeatAction struct {
 	proposal       *HeartbeatProposal
 	failureCounter *heartbeatFailureCounter
 
-	inactivityClaimExecutor *inactivityClaimExecutor
+	inactivityClaimExecutor heartbeatInactivityClaimExecutor
 
 	startBlock  uint64
 	expiryBlock uint64
@@ -90,7 +102,7 @@ func newHeartbeatAction(
 	signingExecutor heartbeatSigningExecutor,
 	proposal *HeartbeatProposal,
 	failureCounter *heartbeatFailureCounter,
-	inactivityClaimExecutor *inactivityClaimExecutor,
+	inactivityClaimExecutor heartbeatInactivityClaimExecutor,
 	startBlock uint64,
 	expiryBlock uint64,
 	waitForBlockFn waitForBlockFn,