tokenbucket-test.js
2.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
var vows = require('vows');
var assert = require('assert');
var TIMING_EPSILON = 10;
var TokenBucket = require('../lib/tokenBucket');
vows.describe('TokenBucket').addBatch({
'capacity 10, 1 per 100ms': {
topic: new TokenBucket(10, 1, 100),
'is initialized empty': function(bucket) {
assert.equal(bucket.bucketSize, 10);
assert.equal(bucket.tokensPerInterval, 1);
assert.equal(bucket.content, 0);
},
'removing 10 tokens': {
topic: function(bucket) {
this.gStart = +new Date();
bucket.removeTokens(10, this.callback);
},
'takes 1 second': function(remainingTokens) {
var duration = +new Date() - this.gStart;
var diff = Math.abs(1000 - duration);
assert.ok(diff < TIMING_EPSILON, diff+'');
assert.equal(remainingTokens, 0);
},
'and removing another 10 tokens': {
topic: function(_, bucket) {
this.gStart = +new Date();
assert.equal(bucket.content, 0);
bucket.removeTokens(10, this.callback);
},
'takes 1 second': function() {
var duration = +new Date() - this.gStart;
var diff = Math.abs(1000 - duration);
assert.ok(diff < TIMING_EPSILON, diff+'');
}
},
'and waiting 2 seconds': {
topic: function(_, bucket) {
var self = this;
setTimeout(function() {
self.gStart = +new Date();
bucket.removeTokens(10, self.callback);
}, 2000);
},
'gives us only 10 tokens': function(remainingTokens) {
var duration = +new Date() - this.gStart;
assert.ok(duration < TIMING_EPSILON, duration+'');
assert.equal(remainingTokens, 0);
},
'and removing 1 token': {
topic: function(_, _, bucket) {
this.gStart = +new Date();
bucket.removeTokens(1, this.callback);
},
'takes 100ms': function() {
var duration = +new Date() - this.gStart;
var diff = Math.abs(100 - duration);
assert.ok(diff < TIMING_EPSILON, diff+'');
}
}
}
}
},
}).export(module);