improve tests to pass when verifying torrent data

1 files changed, 11 insertions, 2 deletions

diff --git a/test/download-tracker-magnet.js b/test/download-tracker-magnet.js
index 43ed393..0aac1fb 100644
--- a/test/download-tracker-magnet.js
+++ b/test/download-tracker-magnet.js
@@ -27,7 +27,7 @@ function magnetDownloadTest (t, serverType) {
   auto({
     tracker: function (cb) {
       var tracker = new TrackerServer(
-        serverType === 'udp' ? { http: false } : { udp: false }
+        serverType === 'udp' ? { http: false, ws: false } : { udp: false, ws: false }
       )
 
       tracker.on('error', function (err) { t.fail(err) })
@@ -84,13 +84,22 @@ function magnetDownloadTest (t, serverType) {
           file.getBuffer(function (err, buf) {
             if (err) throw err
             t.deepEqual(buf, leavesFile, 'downloaded correct content')
+            gotBuffer = true
+            maybeDone()
           })
         })
 
         torrent.once('done', function () {
           t.pass('client2 downloaded torrent from client1')
-          cb(null, client2)
+          torrentDone = true
+          maybeDone()
         })
+
+        var gotBuffer = false
+        var torrentDone = false
+        function maybeDone () {
+          if (gotBuffer && torrentDone) cb(null, client2)
+        }
       })
     }]