AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-12 16:09:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update rate-limiter.test.ts

Browse Source
This commit is contained in:
Nicolas 2024-08-27 15:38:01 -03:00
parent bd84e49a77
commit 8c8d0602bb
1 changed files with 12 additions and 6 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", () => {