mirror of https://git.tuxpa.in/a/code-server.git
38 lines
1.2 KiB
TypeScript
38 lines
1.2 KiB
TypeScript
import { RateLimiter } from "../../../src/node/routes/login"
|
|
|
|
describe("login", () => {
|
|
describe("RateLimiter", () => {
|
|
it("should allow one try ", () => {
|
|
const limiter = new RateLimiter()
|
|
expect(limiter.removeToken()).toBe(true)
|
|
})
|
|
|
|
it("should pull tokens from both limiters (minute & hour)", () => {
|
|
const limiter = new RateLimiter()
|
|
|
|
// Try twice, which pulls two from the minute bucket
|
|
limiter.removeToken()
|
|
limiter.removeToken()
|
|
|
|
// Check that we can still try
|
|
// which should be true since there are 12 remaining in the hour bucket
|
|
expect(limiter.canTry()).toBe(true)
|
|
expect(limiter.removeToken()).toBe(true)
|
|
})
|
|
|
|
it("should not allow more than 14 tries in less than an hour", () => {
|
|
const limiter = new RateLimiter()
|
|
|
|
// The limiter allows 2 tries per minute plus 12 per hour
|
|
// so if we run it 15 times, 14 should return true and the last
|
|
// should return false
|
|
for (let i = 1; i <= 14; i++) {
|
|
expect(limiter.removeToken()).toBe(true)
|
|
}
|
|
|
|
expect(limiter.canTry()).toBe(false)
|
|
expect(limiter.removeToken()).toBe(false)
|
|
})
|
|
})
|
|
})
|