From ae464ada60b00bd2901ba2d580f849ab5a1fdbae Mon Sep 17 00:00:00 2001
From: rafaelsideguide <150964962+rafaelsideguide@users.noreply.github.com>
Date: Wed, 9 Oct 2024 15:06:29 -0300
Subject: [PATCH] tests: teamIds

---
 apps/api/src/services/queue-worker.ts | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/apps/api/src/services/queue-worker.ts b/apps/api/src/services/queue-worker.ts
index bff51f74..2d2ce680 100644
--- a/apps/api/src/services/queue-worker.ts
+++ b/apps/api/src/services/queue-worker.ts
@@ -139,11 +139,28 @@ const workerFun = async (
 
       if (job.data && job.data.team_id && job.data.plan) {
         const concurrencyLimiterThrottledKey = "concurrency-limiter:" + job.data.team_id + ":throttled";
-        const concurrencyLimit = getRateLimiterPoints(RateLimiterMode.Scrape, undefined, job.data.plan);
+        let concurrencyLimit = getRateLimiterPoints(RateLimiterMode.Scrape, undefined, job.data.plan);
         const now = Date.now();
         const stalledJobTimeoutMs = 2 * 60 * 1000;
         const throttledJobTimeoutMs = 10 * 60 * 1000;
 
+        const testTeamIds = [
+          '5a30bfe9-3639-4715-9791-d63fa1c54f16',
+          'cca91fd2-7b7f-4336-b2ad-be065a6786b5',
+          'f69cf2ce-c58b-42df-8a07-e64e9bfd4881',
+          '5e342025-51c9-46db-9095-019ab8ba9ce0',
+          '2c789018-5cd7-49dc-b56e-3d5e87cd6c38',
+          '71e160b6-6b34-4729-8daf-5edcd1788547',
+          '2fd66839-4b30-41c3-9e95-8808a3a45fb8',
+          'f20fb283-02bf-4868-817d-c94b818cb3e1',
+          'fa28a33b-2b28-4169-b81f-ff52b8a91aed',
+          '97722c34-2dd7-46c9-b07e-38bead23a5d8'
+        ]
+
+        if (job.data && job.data.team_id && testTeamIds.includes(job.data.team_id)) {
+          concurrencyLimit = Infinity;
+        }
+
         redisConnection.zremrangebyscore(concurrencyLimiterThrottledKey, -Infinity, now);
         redisConnection.zremrangebyscore(concurrencyLimiterKey, -Infinity, now);
         const activeJobsOfTeam = await redisConnection.zrangebyscore(concurrencyLimiterKey, now, Infinity);