前端代码

node_modules/limiter/test/ratelimiter-test.js

@@ -0,0 +1,20 @@
+var vows = require('vows');
+var assert = require('assert');
+var RateLimiter = require('../lib/rateLimiter');
+
+vows.describe('RateLimiter').addBatch({
+  'interval validation': {
+    'invalid interval': function() {
+      assert.throws(function() { new RateLimiter(1, 'junk'); }, /interval/);
+    },
+    'valid intervals': function() {
+      assert.doesNotThrow(function() { new RateLimiter(1, 'sec'); });
+      assert.doesNotThrow(function() { new RateLimiter(1, 'second'); });
+      assert.doesNotThrow(function() { new RateLimiter(1, 'min'); });
+      assert.doesNotThrow(function() { new RateLimiter(1, 'minute'); });
+      assert.doesNotThrow(function() { new RateLimiter(1, 'hr'); });
+      assert.doesNotThrow(function() { new RateLimiter(1, 'hour'); });
+      assert.doesNotThrow(function() { new RateLimiter(1, 'day'); });
+    }
+  }
+}).export(module);

node_modules/limiter/test/tokenbucket-test.js

@@ -0,0 +1,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);