From 9f27e10179a3505923b6b1f8896beb727c70e15c Mon Sep 17 00:00:00 2001
From: Aditya Vyas <aditya.vyas@stellar.org>
Date: Mon, 18 Nov 2024 22:53:52 +0530
Subject: [PATCH] Remove Once() in mock call

---
 internal/services/ingest_test.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/internal/services/ingest_test.go b/internal/services/ingest_test.go
index 98377bb..573bc95 100644
--- 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")