add throttler factory to tests

Author: BonapartePC

app/cluster/integration_test.go

@@ -34,6 +34,7 @@ import (
 	"github.com/rudderlabs/rudder-server/router"
 	"github.com/rudderlabs/rudder-server/router/batchrouter"
 	routermanager "github.com/rudderlabs/rudder-server/router/manager"
+	"github.com/rudderlabs/rudder-server/router/throttler"
 	destinationdebugger "github.com/rudderlabs/rudder-server/services/debugger/destination"
 	transformationdebugger "github.com/rudderlabs/rudder-server/services/debugger/transformation"
 	"github.com/rudderlabs/rudder-server/services/fileuploader"
@@ -217,6 +218,7 @@ func TestDynamicClusterManager(t *testing.T) {
 	processor.BackendConfig = mockBackendConfig
 	processor.Transformer = mockTransformer
 	mockBackendConfig.EXPECT().WaitForConfig(gomock.Any()).Times(1)
+	throttlerFactory, _ := throttler.NewFactory(config.New(), nil)
 
 	rtFactory := &router.Factory{
 		Logger:           logger.NOP,
@@ -226,6 +228,7 @@ func TestDynamicClusterManager(t *testing.T) {
 		ProcErrorDB:      readErrDB,
 		TransientSources: transientsource.NewEmptyService(),
 		RsourcesService:  mockRsourcesService,
+		ThrottlerFactory: throttlerFactory,
 	}
 	brtFactory := &batchrouter.Factory{
 		Reporting:        &reporting.NOOP{},

router/handle.go

@@ -296,11 +296,7 @@ func (rt *Handle) commitStatusList(workerJobStatuses *[]workerJobStatus) {
 		if err != nil {
 			errorCode = 200
 		}
-		if rt.throttlerFactory != nil {
-			rt.throttlerFactory.Get(rt.destType, parameters.DestinationID).ResponseCodeReceived(errorCode) // send response code to throttler
-		} else {
-			rt.logger.Debugf("[%v Router] :: ThrottlerFactory is nil. Not sending response code to throttler", rt.destType)
-		}
+		rt.throttlerFactory.Get(rt.destType, parameters.DestinationID).ResponseCodeReceived(errorCode) // send response code to throttler
 		// Update metrics maps
 		// REPORTING - ROUTER - START
 		workspaceID := workerJobStatus.status.WorkspaceId

router/manager/manager_test.go

@@ -28,6 +28,7 @@ import (
 	mocksBackendConfig "github.com/rudderlabs/rudder-server/mocks/backend-config"
 	"github.com/rudderlabs/rudder-server/router"
 	"github.com/rudderlabs/rudder-server/router/batchrouter"
+	"github.com/rudderlabs/rudder-server/router/throttler"
 	"github.com/rudderlabs/rudder-server/services/rsources"
 	"github.com/rudderlabs/rudder-server/services/transientsource"
 	"github.com/rudderlabs/rudder-server/utils/pubsub"
@@ -193,6 +194,7 @@ func TestRouterManager(t *testing.T) {
 	mockRtDB := &mockJobsDB{JobsDB: rtDB}
 	brtDB := jobsdb.NewForReadWrite("batch_rt")
 	errDB := jobsdb.NewForReadWrite("proc_error")
+	throttlerFactory, _ := throttler.NewFactory(config.New(), nil)
 	defer rtDB.Close()
 	defer brtDB.Close()
 	defer errDB.Close()
@@ -204,6 +206,7 @@ func TestRouterManager(t *testing.T) {
 		ProcErrorDB:      errDB,
 		TransientSources: transientsource.NewEmptyService(),
 		RsourcesService:  mockRsourcesService,
+		ThrottlerFactory: throttlerFactory,
 	}
 	brtFactory := &batchrouter.Factory{
 		Reporting:        &reporting.NOOP{},