Template Upload

node_modules/limiter/test/tokenbucket-test.js

@@ -0,0 +1,70 @@
+var vows = require('vows');
+var assert = require('assert');
+
+var TIMING_EPSILON = 10;
+
+var TokenBucket = require('../lib/tokenBucket');
+var gBucket;
+var gStart;
+
+vows.describe('TokenBucket').addBatch({
+  'capacity 10, 1 per 100ms': {
+    topic: new TokenBucket(10, 1, 100),
+
+    'is initialized empty': function(bucket) {
+      gBucket = bucket;
+      assert.equal(bucket.bucketSize, 10);
+      assert.equal(bucket.tokensPerInterval, 1);
+      assert.equal(bucket.content, 0);
+    },
+    'removing 10 tokens': {
+      topic: function(bucket) {
+        gStart = +new Date();
+        bucket.removeTokens(10, this.callback);
+      },
+      'takes 1 second': function(remainingTokens) {
+        var duration = +new Date() - gStart;
+        var diff = Math.abs(1000 - duration);
+        assert.ok(diff < TIMING_EPSILON, diff+'');
+        assert.equal(remainingTokens, 0);
+      },
+      'and removing another 10 tokens': {
+        topic: function() {
+          gStart = +new Date();
+          assert.equal(gBucket.content, 0);
+          gBucket.removeTokens(10, this.callback);
+        },
+        'takes 1 second': function() {
+          var duration = +new Date() - gStart;
+          var diff = Math.abs(1000 - duration);
+          assert.ok(diff < TIMING_EPSILON, diff+'');
+        }
+      },
+      'and waiting 2 seconds': {
+        topic: function() {
+          var self = this;
+          setTimeout(function() {
+            gStart = +new Date();
+            gBucket.removeTokens(10, self.callback);
+          }, 2000);
+        },
+        'gives us only 10 tokens': function(remainingTokens) {
+          var duration = +new Date() - gStart;
+          assert.ok(duration < TIMING_EPSILON, duration+'');
+          assert.equal(remainingTokens, 0);
+        },
+        'and removing 1 token': {
+          topic: function() {
+            gStart = +new Date();
+            gBucket.removeTokens(1, this.callback);
+          },
+          'takes 100ms': function() {
+            var duration = +new Date() - gStart;
+            var diff = Math.abs(100 - duration);
+            assert.ok(diff < TIMING_EPSILON, diff+'');
+          }
+        }
+      }
+    }
+  },
+}).export(module);