test fix

Author: nischitpra

internal/orchestrator/committer_test.go

@@ -1,11 +1,13 @@
 package orchestrator
 
 import (
+	"context"
 	"math/big"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
+	"github.com/thirdweb-dev/indexer/internal/rpc"
 	"github.com/thirdweb-dev/indexer/internal/storage"
 	mocks "github.com/thirdweb-dev/indexer/test/mocks"
 )
@@ -19,6 +21,10 @@ func TestNewCommitter(t *testing.T) {
 		MainStorage:    mockMainStorage,
 		StagingStorage: mockStagingStorage,
 	}
+
+	// Mock the GetBlocksPerRequest call that happens in NewWorker
+	mockRPC.EXPECT().GetBlocksPerRequest().Return(rpc.BlocksPerRequestConfig{Blocks: 100})
+
 	poller := &Poller{}
 	committer := NewCommitter(mockRPC, mockStorage, poller)
 
@@ -61,20 +67,25 @@ func TestCleanupProcessedStagingBlocks(t *testing.T) {
 		StagingStorage:      mockStagingStorage,
 		OrchestratorStorage: mockOrchestratorStorage,
 	}
+
+	// Mock the GetBlocksPerRequest call that happens in NewWorker
+	mockRPC.EXPECT().GetBlocksPerRequest().Return(rpc.BlocksPerRequestConfig{Blocks: 100})
+
 	poller := &Poller{}
 	committer := NewCommitter(mockRPC, mockStorage, poller)
 
 	chainID := big.NewInt(1)
 	committer.lastCommittedBlock.Store(100)
 	committer.lastPublishedBlock.Store(0)
 
-	committer.cleanupProcessedStagingBlocks()
+	ctx := context.Background()
+	committer.cleanupProcessedStagingBlocks(ctx)
 	mockStagingStorage.AssertNotCalled(t, "DeleteStagingDataOlderThan", mock.Anything, mock.Anything)
 
 	committer.lastPublishedBlock.Store(90)
 	mockRPC.EXPECT().GetChainID().Return(chainID)
 	mockStagingStorage.EXPECT().DeleteStagingDataOlderThan(chainID, big.NewInt(90)).Return(nil)
-	committer.cleanupProcessedStagingBlocks()
+	committer.cleanupProcessedStagingBlocks(ctx)
 }
 
 func TestStartCommitter(t *testing.T) {