AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-14 06:25:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Nick:

Browse Source
This commit is contained in:
Nicolas 2024-08-27 15:36:46 -03:00
parent c009013ff6
commit a0f9a81ee6
2 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
apps/api/src/services/rate-limiter.test.ts
View File

@ -65,7 +65,7 @@ describe("Rate Limiter Service", () => {
"test-prefix:someToken", "test-prefix:someToken",
"standard" "standard"
); );
expect(limiter2.points).toBe(50); expect(limiter2.points).toBe(100);
const limiter3 = getRateLimiter( const limiter3 = getRateLimiter(
"search" as RateLimiterMode, "search" as RateLimiterMode,
@ -188,14 +188,13 @@ describe("Rate Limiter Service", () => {
"test-prefix:someTokenXY", "test-prefix:someTokenXY",
"hobby" "hobby"
); );
// expect hobby to have 100 points expect(limiter.points).toBe(20);
expect(limiter.points).toBe(10);
const consumePoints = 5; const consumePoints = 5;
const res = await limiter.consume("test-prefix:someTokenXY", consumePoints); const res = await limiter.consume("test-prefix:someTokenXY", consumePoints);
expect(res.consumedPoints).toBe(5); expect(res.consumedPoints).toBe(5);
expect(res.remainingPoints).toBe(5); expect(res.remainingPoints).toBe(15);
}); });
it("should return the correct rate limiter for 'crawl' mode", () => { it("should return the correct rate limiter for 'crawl' mode", () => {
@ -227,7 +226,7 @@ describe("Rate Limiter Service", () => {
"test-prefix:someToken", "test-prefix:someToken",
"free" "free"
); );
expect(limiter.points).toBe(5); expect(limiter.points).toBe(10);
const limiter2 = getRateLimiter( const limiter2 = getRateLimiter(
"scrape" as RateLimiterMode, "scrape" as RateLimiterMode,
@ -241,7 +240,14 @@ describe("Rate Limiter Service", () => {
"test-prefix:someToken", "test-prefix:someToken",
"standard" "standard"
); );
expect(limiter3.points).toBe(50); expect(limiter3.points).toBe(100);
const limiter4 = getRateLimiter(
"scrape" as RateLimiterMode,
"test-prefix:someToken",
"growth"
);
expect(limiter4.points).toBe(1000);
}); });
it("should return the correct rate limiter for 'search' mode", () => { it("should return the correct rate limiter for 'search' mode", () => {

2
apps/api/src/services/rate-limiter.ts
View File

@ -115,7 +115,7 @@ export function getRateLimiter(
return testSuiteRateLimiter; return testSuiteRateLimiter;
} }
if(teamId === process.env.DEV_B_TEAM_ID) { if (teamId && teamId === process.env.DEV_B_TEAM_ID) {
return devBRateLimiter; return devBRateLimiter;
} }