Remove Once() in mock call

stellar · Nov 18, 2024 · 9f27e10 · 9f27e10
1 parent a477ab0
commit 9f27e10
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/services/ingest_test.go b/internal/services/ingest_test.go
@@ -406,7 +406,7 @@ func TestTrackRPCServiceHealth_HealthyService(t *testing.T) {
 		OldestLedger:          1,
 		LedgerRetentionWindow: 0,
 	}
-	mockRPCService.On("GetHealth").Return(healthResult, nil).Once()
+	mockRPCService.On("GetHealth").Return(healthResult, nil)
 
 	go func() {
 		select {
@@ -433,7 +433,10 @@ func TestTrackRPCServiceHealth_UnhealthyService(t *testing.T) {
 	defer cancel()
 
 	mockRPCService := &RPCServiceMock{}
-	mockRPCService.On("GetHealth").Return(entities.RPCGetHealthResult{}, errors.New("rpc error"))
+	mockRPCService.On("GetHealth").Return(
+		entities.RPCGetHealthResult{},
+		errors.New("rpc error"),
+	)
 
 	mockAppTracker := &apptracker.MockAppTracker{}
 	mockAppTracker.On("CaptureMessage", "rpc service unhealthy for over 1m0s")