tests and code styling

index.js

@@ -13,7 +13,7 @@ var parseTorrent = require('parse-torrent')
 var speedometer = require('speedometer')
 var zeroFill = require('zero-fill')
 var path = require('path')
-var ThrottleGroup = require('stream-throttle').ThrottleGroup 
+var ThrottleGroup = require('stream-throttle').ThrottleGroup
 
 var Torrent = require('./lib/torrent')
 
@@ -65,7 +65,7 @@ function WebTorrent (opts) {
   self.downloadSpeed = speedometer()
   self.uploadSpeed = speedometer()
 
-  self.throttleGroups = { 
+  self.throttleGroups = {
     down: new ThrottleGroup({rate: Number.MAX_VALUE, chunksize: 512}),
     up: new ThrottleGroup({rate: Number.MAX_VALUE, chunksize: 512})
   }
@@ -303,31 +303,31 @@ WebTorrent.prototype.destroy = function (cb) {
  * Set global download throttle in Bps
  * @param  {Number} rate
  */
- WebTorrent.prototype.throttleDownload = function (rate) {
-   if (!Number(rate)) return
-   var self = this
-   self.downloadThrottleRate = rate
-   self.throttleGroups.down = new ThrottleGroup({rate: rate})
-   self.torrents.forEach(function (torrent) {
-     if (torrent.swarm && torrent.swarm.updateDownloadThrottle) {
-       torrent.swarm.updateDownloadThrottle()
-     }
-   })
- }
+WebTorrent.prototype.throttleDownload = function (rate) {
+  if (!Number(rate)) return
+  var self = this
+  self.downloadThrottleRate = rate
+  self.throttleGroups.down = new ThrottleGroup({rate: rate})
+  self.torrents.forEach(function (torrent) {
+    if (torrent.swarm && torrent.swarm.updateDownloadThrottle) {
+      torrent.swarm.updateDownloadThrottle()
+    }
+  })
+}
 
  /**
   * Set global upload throttle in Bps
   * @param  {Number} rate
   */
- WebTorrent.prototype.throttleUpload = function (rate) {
-   var self = this
-   if (!Number(rate)) return
-   self.uploadThrottleRate = rate
-   self.throttleGroups.up = new ThrottleGroup({rate: rate})
-   self.torrents.forEach(function (torrent) {
-     if (torrent.swarm && torrent.swarm.updateUploadThrottle) {
-       torrent.swarm.updateUploadThrottle()
-     }
-   })
- }
+WebTorrent.prototype.throttleUpload = function (rate) {
+  var self = this
+  if (!Number(rate)) return
+  self.uploadThrottleRate = rate
+  self.throttleGroups.up = new ThrottleGroup({rate: rate})
+  self.torrents.forEach(function (torrent) {
+    if (torrent.swarm && torrent.swarm.updateUploadThrottle) {
+      torrent.swarm.updateUploadThrottle()
+    }
+  })
+}
 

test/throttle.js

@@ -0,0 +1,18 @@
+var test = require('tape')
+var WebTorrent = require('../')
+
+test('client download/upload throttle setting and rate reporting', function (t) {
+  var client = new WebTorrent({ dht: false, tracker: false })
+
+  client.on('error', function (err) { t.fail(err) })
+  client.on('warning', function (err) { t.fail(err) })
+
+  client.throttleDownload(100000)
+  client.throttleUpload(10000)
+
+  t.equal(client.downloadThrottleRate, 100000)
+  t.equal(client.uploadThrottleRate, 10000)
+
+  client.destroy()
+  t.end()
+})