Update node_lifecycle_test.go

smartcontractkit · Jul 8, 2024 · 169e44b · 169e44b
1 parent 80ddd26
commit 169e44b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/common/client/node_lifecycle_test.go b/common/client/node_lifecycle_test.go
@@ -86,8 +86,8 @@ func TestUnit_NodeLifecycle_aliveLoop(t *testing.T) {
 	newSubscribedNode := func(t *testing.T, opts testNodeOpts) testNode {
 		sub := mocks.NewSubscription(t)
 		sub.On("Err").Return(nil)
-		sub.On("Unsubscribe")
-		opts.rpc.On("SubscribeToHeads", mock.Anything).Return(make(<-chan Head), sub, nil)
+		sub.On("Unsubscribe").Once()
+		opts.rpc.On("SubscribeToHeads", mock.Anything).Return(make(<-chan Head), sub, nil).Once()
 		return newDialedNode(t, opts)
 	}
 	t.Run("Stays alive and waits for signal", func(t *testing.T) {