Function RateLimiterTestSuite

packages/effect/test/RateLimiter.test.ts:157–377

(algorithm: "fixed-window" | "token-bucket")

})

const RateLimiterTestSuite = (algorithm: "fixed-window" | "token-bucket") => {
  it.scoped(`${algorithm} - execute up to max calls immediately`, () =>
    Effect.gen(function*() {
      const limit = yield* (RateLimiter.make({
        limit: 10,
        interval: "1 seconds",
        algorithm
      }))
      const now = yield* (Clock.currentTimeMillis)
      const times = yield* (Effect.forEach(
        Array.range(1, 10),
        () => limit(Clock.currentTimeMillis)
      ))
      const result = Array.every(times, (time) => time === now)
      assertTrue(result)
    }))

  it.scoped(`${algorithm} - is not affected by stream chunk size`, () =>
    Effect.gen(function*() {
      const limiter = yield* (RateLimiter.make({
        limit: 10,
        interval: "1 seconds",
        algorithm
      }))
      const now = yield* (Clock.currentTimeMillis)
      const times1 = yield* (Effect.forEach(
        Array.range(1, 5),
        () => limiter(Clock.currentTimeMillis),
        { concurrency: "unbounded" }
      ))
      const fibers = yield* (Effect.forEach(
        Array.range(1, 15),
        () => Effect.fork(limiter(Clock.currentTimeMillis)),
        { concurrency: "unbounded" }
      ))
      yield* (TestClock.adjust("1 seconds"))
      const times2 = yield* (Effect.forEach(fibers, Fiber.join, { concurrency: "unbounded" }))
      const times = Array.appendAll(times1, times2)
      const result = Array.filter(times, (time) => time === now)
      strictEqual(result.length, 10)
    }))

  it.scoped(`${algorithm} - succeed with the result of the call`, () =>
    Effect.gen(function*() {
      const limit = yield* (RateLimiter.make({
        limit: 10,
        interval: "1 seconds",
        algorithm
      }))
      const result = yield* (limit(Effect.succeed(3)))
      strictEqual(result, 3)
    }))

  it.scoped(`${algorithm} - fail with the result of a failed call`, () =>
    Effect.gen(function*() {
      const limit = yield* (RateLimiter.make({
        limit: 10,
        interval: "1 seconds",