Filter tracker based on infohash

2 files changed, 72 insertions, 0 deletions

diff --git a/server/tests/api/index-fast.ts b/server/tests/api/index-fast.ts
index 531a09b82..02ffdd4f1 100644
--- a/server/tests/api/index-fast.ts
+++ b/server/tests/api/index-fast.ts
@@ -15,3 +15,4 @@ import './server/email'
 import './server/config'
 import './server/reverse-proxy'
 import './search/search-videos'
+import './server/tracker'
diff --git a/server/tests/api/server/tracker.ts b/server/tests/api/server/tracker.ts
new file mode 100644
index 000000000..5d61c1558
--- /dev/null
+++ b/server/tests/api/server/tracker.ts
@@ -0,0 +1,71 @@
+/* tslint:disable:no-unused-expression */
+
+import * as magnetUtil from 'magnet-uri'
+import * as chai from 'chai'
+import 'mocha'
+import { getVideo, killallServers, runServer, ServerInfo, uploadVideo } from '../../utils'
+import { flushTests, setAccessTokensToServers } from '../../utils/index'
+import { VideoDetails } from '../../../../shared/models/videos'
+import * as WebTorrent from 'webtorrent'
+
+describe('Test tracker', function () {
+  let server: ServerInfo
+  let badMagnet: string
+  let goodMagnet: string
+
+  before(async function () {
+    this.timeout(60000)
+
+    await flushTests()
+    server = await runServer(1)
+    await setAccessTokensToServers([ server ])
+
+    {
+      const res = await uploadVideo(server.url, server.accessToken, {})
+      const videoUUID = res.body.video.uuid
+
+      const resGet = await getVideo(server.url, videoUUID)
+      const video: VideoDetails = resGet.body
+      goodMagnet = video.files[0].magnetUri
+
+      const parsed = magnetUtil.decode(goodMagnet)
+      parsed.infoHash = '010597bb88b1968a5693a4fa8267c592ca65f2e9'
+
+      badMagnet = magnetUtil.encode(parsed)
+    }
+  })
+
+  it('Should return an error when adding an incorrect infohash', done => {
+    this.timeout(10000)
+    const webtorrent = new WebTorrent()
+
+    const torrent = webtorrent.add(badMagnet)
+
+    torrent.on('error', done)
+    torrent.on('warning', warn => {
+      const message = typeof warn === 'string' ? warn : warn.message
+      if (message.indexOf('Unknown infoHash ') !== -1) return done()
+    })
+
+    torrent.on('done', () => done(new Error('No error on infohash')))
+  })
+
+  it('Should succeed with the correct infohash', done => {
+    this.timeout(10000)
+    const webtorrent = new WebTorrent()
+
+    const torrent = webtorrent.add(goodMagnet)
+
+    torrent.on('error', done)
+    torrent.on('warning', warn => {
+      const message = typeof warn === 'string' ? warn : warn.message
+      if (message.indexOf('Unknown infoHash ') !== -1) return done(new Error('Error on infohash'))
+    })
+
+    torrent.on('done', done)
+  })
+
+  after(async function () {
+    killallServers([ server ])
+  })
+})