throttle_test.js
1.19 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
var async = require('async');
var Throttle = require('../index.js').Throttle;
var ThrottleGroup = require('../index.js').ThrottleGroup;
var sendStr = (function() {
var s = '0123456789xyzXYZ?!\0åéîõü$£€*<>'; // 30 characters
for (var i = 0, str = ''; i < 1000; i++)
str += s;
return str; // 30K characters
})();
var opts = {rate: 100000}; // 100 KiB/s
var testSendRecv = function(t, cb) {
var recvStr = '';
t.on('data', function(chunk) {
recvStr += chunk;
});
t.on('end', function() {
cb(sendStr == recvStr);
});
t.write(sendStr, function() {
t.end();
});
};
exports.testThrottle = function(test) {
var t = new Throttle(opts);
test.expect(1);
testSendRecv(t, function(ok) {
test.ok(ok, "received string should equal sent string");
test.done();
});
};
exports.testGroupThrottle = function(test) {
var tg = new ThrottleGroup(opts);
test.expect(3);
async.each([1, 2, 3], function(i, done) {
testSendRecv(tg.throttle(), function(ok) {
test.ok(ok, "received string should equal sent string");
done();
});
}, function() {
test.done();
});
};