]>
Commit | Line | Data |
---|---|---|
1 | /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await,@typescript-eslint/no-floating-promises */ | |
2 | ||
3 | import 'mocha' | |
4 | import magnetUtil from 'magnet-uri' | |
5 | import WebTorrent from 'webtorrent' | |
6 | import { cleanupTests, createSingleServer, killallServers, PeerTubeServer, setAccessTokensToServers } from '@shared/server-commands' | |
7 | ||
8 | describe('Test tracker', function () { | |
9 | let server: PeerTubeServer | |
10 | let badMagnet: string | |
11 | let goodMagnet: string | |
12 | ||
13 | before(async function () { | |
14 | this.timeout(60000) | |
15 | server = await createSingleServer(1) | |
16 | await setAccessTokensToServers([ server ]) | |
17 | ||
18 | { | |
19 | const { uuid } = await server.videos.upload() | |
20 | const video = await server.videos.get({ id: uuid }) | |
21 | goodMagnet = video.files[0].magnetUri | |
22 | ||
23 | const parsed = magnetUtil.decode(goodMagnet) | |
24 | parsed.infoHash = '010597bb88b1968a5693a4fa8267c592ca65f2e9' | |
25 | ||
26 | badMagnet = magnetUtil.encode(parsed) | |
27 | } | |
28 | }) | |
29 | ||
30 | it('Should succeed with the correct infohash', function (done) { | |
31 | this.timeout(10000) | |
32 | const webtorrent = new WebTorrent() | |
33 | ||
34 | const torrent = webtorrent.add(goodMagnet) | |
35 | ||
36 | torrent.on('error', done) | |
37 | torrent.on('warning', warn => { | |
38 | const message = typeof warn === 'string' ? warn : warn.message | |
39 | if (message.includes('Unknown infoHash ')) return done(new Error('Error on infohash')) | |
40 | }) | |
41 | ||
42 | torrent.on('done', done) | |
43 | }) | |
44 | ||
45 | it('Should disable the tracker', function (done) { | |
46 | this.timeout(20000) | |
47 | ||
48 | const errCb = () => done(new Error('Tracker is enabled')) | |
49 | ||
50 | killallServers([ server ]) | |
51 | .then(() => server.run({ tracker: { enabled: false } })) | |
52 | .then(() => { | |
53 | const webtorrent = new WebTorrent() | |
54 | ||
55 | const torrent = webtorrent.add(goodMagnet) | |
56 | ||
57 | torrent.on('error', done) | |
58 | torrent.on('warning', warn => { | |
59 | const message = typeof warn === 'string' ? warn : warn.message | |
60 | if (message.includes('disabled ')) { | |
61 | torrent.off('done', errCb) | |
62 | ||
63 | return done() | |
64 | } | |
65 | }) | |
66 | ||
67 | torrent.on('done', errCb) | |
68 | }) | |
69 | }) | |
70 | ||
71 | it('Should return an error when adding an incorrect infohash', function (done) { | |
72 | this.timeout(20000) | |
73 | ||
74 | killallServers([ server ]) | |
75 | .then(() => server.run()) | |
76 | .then(() => { | |
77 | const webtorrent = new WebTorrent() | |
78 | ||
79 | const torrent = webtorrent.add(badMagnet) | |
80 | ||
81 | torrent.on('error', done) | |
82 | torrent.on('warning', warn => { | |
83 | const message = typeof warn === 'string' ? warn : warn.message | |
84 | if (message.includes('Unknown infoHash ')) return done() | |
85 | }) | |
86 | ||
87 | torrent.on('done', () => done(new Error('No error on infohash'))) | |
88 | }) | |
89 | }) | |
90 | ||
91 | it('Should block the IP after the failed infohash', function (done) { | |
92 | const webtorrent = new WebTorrent() | |
93 | ||
94 | const torrent = webtorrent.add(goodMagnet) | |
95 | ||
96 | torrent.on('error', done) | |
97 | torrent.on('warning', warn => { | |
98 | const message = typeof warn === 'string' ? warn : warn.message | |
99 | if (message.includes('Unsupported tracker protocol')) return done() | |
100 | }) | |
101 | }) | |
102 | ||
103 | after(async function () { | |
104 | await cleanupTests([ server ]) | |
105 | }) | |
106 | }) |