mirror of
https://github.com/scratchfoundation/scratch-vm.git
synced 2024-12-22 22:12:28 -05:00
39be6d873c
* Initial version of token bucket * Add rateLimiter util * Remove check for motor.isOn in stopAllMotors * Fix unit test * Fix unit test with stubbed timer, and cleanup * Add comment * Reduce WeDo rate limit to 20 sends/sec * Move rate limit into a constant * Stop button stops motors and tone even if rate limit exceeded
32 lines
1 KiB
JavaScript
32 lines
1 KiB
JavaScript
const test = require('tap').test;
|
|
const RateLimiter = require('../../src/util/rateLimiter.js');
|
|
|
|
test('rate limiter', t => {
|
|
// Create a rate limiter with maximum of 20 sends per second
|
|
const rate = 20;
|
|
const limiter = new RateLimiter(rate);
|
|
|
|
// Simulate time passing with a stubbed timer
|
|
let simulatedTime = Date.now();
|
|
limiter._timer = {timeElapsed: () => simulatedTime};
|
|
|
|
// The rate limiter starts with a number of tokens equal to the max rate
|
|
t.equal(limiter._count, rate);
|
|
|
|
// Running okayToSend a number of times equal to the max rate
|
|
// uses up all of the tokens
|
|
for (let i = 0; i < rate; i++) {
|
|
t.true(limiter.okayToSend());
|
|
// Tokens are counting down
|
|
t.equal(limiter._count, rate - (i + 1));
|
|
}
|
|
t.false(limiter.okayToSend());
|
|
|
|
// Advance the timer enough so we get exactly one more token
|
|
// One extra millisecond is required to get over the threshold
|
|
simulatedTime += (1000 / rate) + 1;
|
|
t.true(limiter.okayToSend());
|
|
t.false(limiter.okayToSend());
|
|
|
|
t.end();
|
|
});
|