]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blame_incremental - server/tests/api/live/live.ts
Try to fix live test
[github/Chocobozzz/PeerTube.git] / server / tests / api / live / live.ts
... / ...
CommitLineData
1/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
2
3import 'mocha'
4import * as chai from 'chai'
5import { FfmpegCommand } from 'fluent-ffmpeg'
6import { join } from 'path'
7import { ffprobePromise, getVideoStreamFromFile } from '@server/helpers/ffprobe-utils'
8import { getLiveNotificationSocket } from '@shared/extra-utils/socket/socket-io'
9import { LiveVideo, LiveVideoCreate, Video, VideoDetails, VideoPrivacy, VideoState, VideoStreamingPlaylistType } from '@shared/models'
10import {
11 addVideoToBlacklist,
12 buildServerDirectory,
13 checkLiveCleanup,
14 checkLiveSegmentHash,
15 checkResolutionsInMasterPlaylist,
16 cleanupTests,
17 createLive,
18 doubleFollow,
19 flushAndRunMultipleServers,
20 getLive,
21 getPlaylist,
22 getVideo,
23 getVideoIdFromUUID,
24 getVideosList,
25 killallServers,
26 makeRawRequest,
27 removeVideo,
28 reRunServer,
29 sendRTMPStream,
30 sendRTMPStreamInVideo,
31 ServerInfo,
32 setAccessTokensToServers,
33 setDefaultVideoChannel,
34 stopFfmpeg,
35 testFfmpegStreamError,
36 testImage,
37 updateCustomSubConfig,
38 updateLive,
39 viewVideo,
40 wait,
41 waitJobs,
42 waitUntilLiveEnded,
43 waitUntilLivePublished,
44 waitUntilLiveStarts,
45 waitUntilLog
46} from '../../../../shared/extra-utils'
47
48const expect = chai.expect
49
50describe('Test live', function () {
51 let servers: ServerInfo[] = []
52
53 before(async function () {
54 this.timeout(120000)
55
56 servers = await flushAndRunMultipleServers(2)
57
58 // Get the access tokens
59 await setAccessTokensToServers(servers)
60 await setDefaultVideoChannel(servers)
61
62 await updateCustomSubConfig(servers[0].url, servers[0].accessToken, {
63 live: {
64 enabled: true,
65 allowReplay: true,
66 transcoding: {
67 enabled: false
68 }
69 }
70 })
71
72 // Server 1 and server 2 follow each other
73 await doubleFollow(servers[0], servers[1])
74 })
75
76 describe('Live creation, update and delete', function () {
77 let liveVideoUUID: string
78
79 it('Should create a live with the appropriate parameters', async function () {
80 this.timeout(20000)
81
82 const attributes: LiveVideoCreate = {
83 category: 1,
84 licence: 2,
85 language: 'fr',
86 description: 'super live description',
87 support: 'support field',
88 channelId: servers[0].videoChannel.id,
89 nsfw: false,
90 waitTranscoding: false,
91 name: 'my super live',
92 tags: [ 'tag1', 'tag2' ],
93 commentsEnabled: false,
94 downloadEnabled: false,
95 saveReplay: true,
96 privacy: VideoPrivacy.PUBLIC,
97 previewfile: 'video_short1-preview.webm.jpg',
98 thumbnailfile: 'video_short1.webm.jpg'
99 }
100
101 const res = await createLive(servers[0].url, servers[0].accessToken, attributes)
102 liveVideoUUID = res.body.video.uuid
103
104 await waitJobs(servers)
105
106 for (const server of servers) {
107 const resVideo = await getVideo(server.url, liveVideoUUID)
108 const video: VideoDetails = resVideo.body
109
110 expect(video.category.id).to.equal(1)
111 expect(video.licence.id).to.equal(2)
112 expect(video.language.id).to.equal('fr')
113 expect(video.description).to.equal('super live description')
114 expect(video.support).to.equal('support field')
115
116 expect(video.channel.name).to.equal(servers[0].videoChannel.name)
117 expect(video.channel.host).to.equal(servers[0].videoChannel.host)
118
119 expect(video.nsfw).to.be.false
120 expect(video.waitTranscoding).to.be.false
121 expect(video.name).to.equal('my super live')
122 expect(video.tags).to.deep.equal([ 'tag1', 'tag2' ])
123 expect(video.commentsEnabled).to.be.false
124 expect(video.downloadEnabled).to.be.false
125 expect(video.privacy.id).to.equal(VideoPrivacy.PUBLIC)
126
127 await testImage(server.url, 'video_short1-preview.webm', video.previewPath)
128 await testImage(server.url, 'video_short1.webm', video.thumbnailPath)
129
130 const resLive = await getLive(server.url, server.accessToken, liveVideoUUID)
131 const live: LiveVideo = resLive.body
132
133 if (server.url === servers[0].url) {
134 expect(live.rtmpUrl).to.equal('rtmp://' + server.hostname + ':' + servers[0].rtmpPort + '/live')
135 expect(live.streamKey).to.not.be.empty
136 } else {
137 expect(live.rtmpUrl).to.be.null
138 expect(live.streamKey).to.be.null
139 }
140
141 expect(live.saveReplay).to.be.true
142 }
143 })
144
145 it('Should have a default preview and thumbnail', async function () {
146 this.timeout(20000)
147
148 const attributes: LiveVideoCreate = {
149 name: 'default live thumbnail',
150 channelId: servers[0].videoChannel.id,
151 privacy: VideoPrivacy.UNLISTED,
152 nsfw: true
153 }
154
155 const res = await createLive(servers[0].url, servers[0].accessToken, attributes)
156 const videoId = res.body.video.uuid
157
158 await waitJobs(servers)
159
160 for (const server of servers) {
161 const resVideo = await getVideo(server.url, videoId)
162 const video: VideoDetails = resVideo.body
163
164 expect(video.privacy.id).to.equal(VideoPrivacy.UNLISTED)
165 expect(video.nsfw).to.be.true
166
167 await makeRawRequest(server.url + video.thumbnailPath, 200)
168 await makeRawRequest(server.url + video.previewPath, 200)
169 }
170 })
171
172 it('Should not have the live listed since nobody streams into', async function () {
173 for (const server of servers) {
174 const res = await getVideosList(server.url)
175
176 expect(res.body.total).to.equal(0)
177 expect(res.body.data).to.have.lengthOf(0)
178 }
179 })
180
181 it('Should not be able to update a live of another server', async function () {
182 await updateLive(servers[1].url, servers[1].accessToken, liveVideoUUID, { saveReplay: false }, 403)
183 })
184
185 it('Should update the live', async function () {
186 this.timeout(10000)
187
188 await updateLive(servers[0].url, servers[0].accessToken, liveVideoUUID, { saveReplay: false })
189 await waitJobs(servers)
190 })
191
192 it('Have the live updated', async function () {
193 for (const server of servers) {
194 const res = await getLive(server.url, server.accessToken, liveVideoUUID)
195 const live: LiveVideo = res.body
196
197 if (server.url === servers[0].url) {
198 expect(live.rtmpUrl).to.equal('rtmp://' + server.hostname + ':' + servers[0].rtmpPort + '/live')
199 expect(live.streamKey).to.not.be.empty
200 } else {
201 expect(live.rtmpUrl).to.be.null
202 expect(live.streamKey).to.be.null
203 }
204
205 expect(live.saveReplay).to.be.false
206 }
207 })
208
209 it('Delete the live', async function () {
210 this.timeout(10000)
211
212 await removeVideo(servers[0].url, servers[0].accessToken, liveVideoUUID)
213 await waitJobs(servers)
214 })
215
216 it('Should have the live deleted', async function () {
217 for (const server of servers) {
218 await getVideo(server.url, liveVideoUUID, 404)
219 await getLive(server.url, server.accessToken, liveVideoUUID, 404)
220 }
221 })
222 })
223
224 describe('Stream checks', function () {
225 let liveVideo: LiveVideo & VideoDetails
226 let rtmpUrl: string
227
228 before(function () {
229 rtmpUrl = 'rtmp://' + servers[0].hostname + ':' + servers[0].rtmpPort + ''
230 })
231
232 async function createLiveWrapper () {
233 const liveAttributes = {
234 name: 'user live',
235 channelId: servers[0].videoChannel.id,
236 privacy: VideoPrivacy.PUBLIC,
237 saveReplay: false
238 }
239
240 const res = await createLive(servers[0].url, servers[0].accessToken, liveAttributes)
241 const uuid = res.body.video.uuid
242
243 const resLive = await getLive(servers[0].url, servers[0].accessToken, uuid)
244 const resVideo = await getVideo(servers[0].url, uuid)
245
246 return Object.assign(resVideo.body, resLive.body) as LiveVideo & VideoDetails
247 }
248
249 it('Should not allow a stream without the appropriate path', async function () {
250 this.timeout(30000)
251
252 liveVideo = await createLiveWrapper()
253
254 const command = sendRTMPStream(rtmpUrl + '/bad-live', liveVideo.streamKey)
255 await testFfmpegStreamError(command, true)
256 })
257
258 it('Should not allow a stream without the appropriate stream key', async function () {
259 this.timeout(30000)
260
261 const command = sendRTMPStream(rtmpUrl + '/live', 'bad-stream-key')
262 await testFfmpegStreamError(command, true)
263 })
264
265 it('Should succeed with the correct params', async function () {
266 this.timeout(30000)
267
268 const command = sendRTMPStream(rtmpUrl + '/live', liveVideo.streamKey)
269 await testFfmpegStreamError(command, false)
270 })
271
272 it('Should not allow a stream on a live that was blacklisted', async function () {
273 this.timeout(30000)
274
275 liveVideo = await createLiveWrapper()
276
277 await addVideoToBlacklist(servers[0].url, servers[0].accessToken, liveVideo.uuid)
278
279 const command = sendRTMPStream(rtmpUrl + '/live', liveVideo.streamKey)
280 await testFfmpegStreamError(command, true)
281 })
282
283 it('Should not allow a stream on a live that was deleted', async function () {
284 this.timeout(30000)
285
286 liveVideo = await createLiveWrapper()
287
288 await removeVideo(servers[0].url, servers[0].accessToken, liveVideo.uuid)
289
290 const command = sendRTMPStream(rtmpUrl + '/live', liveVideo.streamKey)
291 await testFfmpegStreamError(command, true)
292 })
293 })
294
295 describe('Live transcoding', function () {
296 let liveVideoId: string
297
298 async function createLiveWrapper (saveReplay: boolean) {
299 const liveAttributes = {
300 name: 'live video',
301 channelId: servers[0].videoChannel.id,
302 privacy: VideoPrivacy.PUBLIC,
303 saveReplay
304 }
305
306 const res = await createLive(servers[0].url, servers[0].accessToken, liveAttributes)
307 return res.body.video.uuid
308 }
309
310 async function testVideoResolutions (liveVideoId: string, resolutions: number[]) {
311 for (const server of servers) {
312 const resList = await getVideosList(server.url)
313 const videos: Video[] = resList.body.data
314
315 expect(videos.find(v => v.uuid === liveVideoId)).to.exist
316
317 const resVideo = await getVideo(server.url, liveVideoId)
318 const video: VideoDetails = resVideo.body
319
320 expect(video.streamingPlaylists).to.have.lengthOf(1)
321
322 const hlsPlaylist = video.streamingPlaylists.find(s => s.type === VideoStreamingPlaylistType.HLS)
323 expect(hlsPlaylist).to.exist
324
325 // Only finite files are displayed
326 expect(hlsPlaylist.files).to.have.lengthOf(0)
327
328 await checkResolutionsInMasterPlaylist(hlsPlaylist.playlistUrl, resolutions)
329
330 for (let i = 0; i < resolutions.length; i++) {
331 const segmentName = `${i}-000001.ts`
332 await waitUntilLog(servers[0], `${video.uuid}/${segmentName}`, 2, false)
333
334 const res = await getPlaylist(`${servers[0].url}/static/streaming-playlists/hls/${video.uuid}/${i}.m3u8`)
335 const subPlaylist = res.text
336
337 expect(subPlaylist).to.contain(segmentName)
338
339 const baseUrlAndPath = servers[0].url + '/static/streaming-playlists/hls'
340 await checkLiveSegmentHash(baseUrlAndPath, video.uuid, segmentName, hlsPlaylist)
341 }
342 }
343 }
344
345 function updateConf (resolutions: number[]) {
346 return updateCustomSubConfig(servers[0].url, servers[0].accessToken, {
347 live: {
348 enabled: true,
349 allowReplay: true,
350 maxDuration: null,
351 transcoding: {
352 enabled: true,
353 resolutions: {
354 '240p': resolutions.includes(240),
355 '360p': resolutions.includes(360),
356 '480p': resolutions.includes(480),
357 '720p': resolutions.includes(720),
358 '1080p': resolutions.includes(1080),
359 '2160p': resolutions.includes(2160)
360 }
361 }
362 }
363 })
364 }
365
366 before(async function () {
367 await updateConf([])
368 })
369
370 it('Should enable transcoding without additional resolutions', async function () {
371 this.timeout(30000)
372
373 liveVideoId = await createLiveWrapper(false)
374
375 const command = await sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoId)
376 await waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoId)
377 await waitJobs(servers)
378
379 await testVideoResolutions(liveVideoId, [ 720 ])
380
381 await stopFfmpeg(command)
382 })
383
384 it('Should enable transcoding with some resolutions', async function () {
385 this.timeout(30000)
386
387 const resolutions = [ 240, 480 ]
388 await updateConf(resolutions)
389 liveVideoId = await createLiveWrapper(false)
390
391 const command = await sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoId)
392 await waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoId)
393 await waitJobs(servers)
394
395 await testVideoResolutions(liveVideoId, resolutions)
396
397 await stopFfmpeg(command)
398 })
399
400 it('Should enable transcoding with some resolutions and correctly save them', async function () {
401 this.timeout(120000)
402
403 const resolutions = [ 240, 360, 720 ]
404
405 await updateConf(resolutions)
406 liveVideoId = await createLiveWrapper(true)
407
408 const command = await sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoId, 'video_short2.webm')
409 await waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoId)
410 await waitJobs(servers)
411
412 await testVideoResolutions(liveVideoId, resolutions)
413
414 await stopFfmpeg(command)
415 await waitUntilLivePublished(servers[0].url, servers[0].accessToken, liveVideoId)
416
417 await waitJobs(servers)
418
419 const bitrateLimits = {
420 720: 4000 * 1000, // 60FPS
421 360: 1100 * 1000,
422 240: 600 * 1000
423 }
424
425 for (const server of servers) {
426 const resVideo = await getVideo(server.url, liveVideoId)
427 const video: VideoDetails = resVideo.body
428
429 expect(video.duration).to.be.greaterThan(1)
430 expect(video.files).to.have.lengthOf(0)
431
432 const hlsPlaylist = video.streamingPlaylists.find(s => s.type === VideoStreamingPlaylistType.HLS)
433
434 expect(hlsPlaylist.files).to.have.lengthOf(resolutions.length)
435
436 for (const resolution of resolutions) {
437 const file = hlsPlaylist.files.find(f => f.resolution.id === resolution)
438
439 expect(file).to.exist
440 expect(file.size).to.be.greaterThan(1)
441
442 if (resolution >= 720) {
443 expect(file.fps).to.be.approximately(60, 2)
444 } else {
445 expect(file.fps).to.be.approximately(30, 2)
446 }
447
448 const filename = `${video.uuid}-${resolution}-fragmented.mp4`
449 const segmentPath = buildServerDirectory(servers[0], join('streaming-playlists', 'hls', video.uuid, filename))
450
451 const probe = await ffprobePromise(segmentPath)
452 const videoStream = await getVideoStreamFromFile(segmentPath, probe)
453
454 expect(probe.format.bit_rate).to.be.below(bitrateLimits[videoStream.height])
455
456 await makeRawRequest(file.torrentUrl, 200)
457 await makeRawRequest(file.fileUrl, 200)
458 }
459 }
460 })
461
462 it('Should correctly have cleaned up the live files', async function () {
463 this.timeout(30000)
464
465 await checkLiveCleanup(servers[0], liveVideoId, [ 240, 360, 720 ])
466 })
467 })
468
469 describe('Live views', function () {
470 let liveVideoId: string
471 let command: FfmpegCommand
472
473 async function countViews (expected: number) {
474 for (const server of servers) {
475 const res = await getVideo(server.url, liveVideoId)
476 const video: VideoDetails = res.body
477
478 expect(video.views).to.equal(expected)
479 }
480 }
481
482 before(async function () {
483 this.timeout(30000)
484
485 const liveAttributes = {
486 name: 'live video',
487 channelId: servers[0].videoChannel.id,
488 privacy: VideoPrivacy.PUBLIC
489 }
490
491 const res = await createLive(servers[0].url, servers[0].accessToken, liveAttributes)
492 liveVideoId = res.body.video.uuid
493
494 command = await sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoId)
495 await waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoId)
496 await waitJobs(servers)
497 })
498
499 it('Should display no views for a live', async function () {
500 await countViews(0)
501 })
502
503 it('Should view a live twice and display 1 view', async function () {
504 this.timeout(30000)
505
506 await viewVideo(servers[0].url, liveVideoId)
507 await viewVideo(servers[0].url, liveVideoId)
508
509 await wait(7000)
510
511 await waitJobs(servers)
512
513 await countViews(1)
514 })
515
516 it('Should wait and display 0 views', async function () {
517 this.timeout(30000)
518
519 await wait(7000)
520 await waitJobs(servers)
521
522 await countViews(0)
523 })
524
525 it('Should view a live on a remote and on local and display 2 views', async function () {
526 this.timeout(30000)
527
528 await viewVideo(servers[0].url, liveVideoId)
529 await viewVideo(servers[1].url, liveVideoId)
530 await viewVideo(servers[1].url, liveVideoId)
531
532 await wait(7000)
533 await waitJobs(servers)
534
535 await countViews(2)
536 })
537
538 after(async function () {
539 await stopFfmpeg(command)
540 })
541 })
542
543 describe('Live socket messages', function () {
544
545 async function createLiveWrapper () {
546 const liveAttributes = {
547 name: 'live video',
548 channelId: servers[0].videoChannel.id,
549 privacy: VideoPrivacy.PUBLIC
550 }
551
552 const res = await createLive(servers[0].url, servers[0].accessToken, liveAttributes)
553 return res.body.video.uuid
554 }
555
556 it('Should correctly send a message when the live starts and ends', async function () {
557 this.timeout(60000)
558
559 const localStateChanges: VideoState[] = []
560 const remoteStateChanges: VideoState[] = []
561
562 const liveVideoUUID = await createLiveWrapper()
563 await waitJobs(servers)
564
565 {
566 const videoId = await getVideoIdFromUUID(servers[0].url, liveVideoUUID)
567
568 const localSocket = getLiveNotificationSocket(servers[0].url)
569 localSocket.on('state-change', data => localStateChanges.push(data.state))
570 localSocket.emit('subscribe', { videoId })
571 }
572
573 {
574 const videoId = await getVideoIdFromUUID(servers[1].url, liveVideoUUID)
575
576 const remoteSocket = getLiveNotificationSocket(servers[1].url)
577 remoteSocket.on('state-change', data => remoteStateChanges.push(data.state))
578 remoteSocket.emit('subscribe', { videoId })
579 }
580
581 const command = await sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoUUID)
582 await waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoUUID)
583 await waitJobs(servers)
584
585 for (const stateChanges of [ localStateChanges, remoteStateChanges ]) {
586 expect(stateChanges).to.have.lengthOf(1)
587 expect(stateChanges[0]).to.equal(VideoState.PUBLISHED)
588 }
589
590 await stopFfmpeg(command)
591 await waitUntilLiveEnded(servers[0].url, servers[0].accessToken, liveVideoUUID)
592
593 await waitJobs(servers)
594
595 for (const stateChanges of [ localStateChanges, remoteStateChanges ]) {
596 expect(stateChanges).to.have.lengthOf(2)
597 expect(stateChanges[1]).to.equal(VideoState.LIVE_ENDED)
598 }
599 })
600
601 it('Should not receive a notification after unsubscribe', async function () {
602 this.timeout(60000)
603
604 const stateChanges: VideoState[] = []
605
606 const liveVideoUUID = await createLiveWrapper()
607 await waitJobs(servers)
608
609 const videoId = await getVideoIdFromUUID(servers[0].url, liveVideoUUID)
610
611 const socket = getLiveNotificationSocket(servers[0].url)
612 socket.on('state-change', data => stateChanges.push(data.state))
613 socket.emit('subscribe', { videoId })
614
615 const command = await sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoUUID)
616 await waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoUUID)
617 await waitJobs(servers)
618
619 expect(stateChanges).to.have.lengthOf(1)
620 socket.emit('unsubscribe', { videoId })
621
622 await stopFfmpeg(command)
623 await waitJobs(servers)
624
625 expect(stateChanges).to.have.lengthOf(1)
626 })
627 })
628
629 describe('After a server restart', function () {
630 let liveVideoId: string
631 let liveVideoReplayId: string
632
633 async function createLiveWrapper (saveReplay: boolean) {
634 const liveAttributes = {
635 name: 'live video',
636 channelId: servers[0].videoChannel.id,
637 privacy: VideoPrivacy.PUBLIC,
638 saveReplay
639 }
640
641 const res = await createLive(servers[0].url, servers[0].accessToken, liveAttributes)
642 return res.body.video.uuid
643 }
644
645 before(async function () {
646 this.timeout(60000)
647
648 liveVideoId = await createLiveWrapper(false)
649 liveVideoReplayId = await createLiveWrapper(true)
650
651 await Promise.all([
652 sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoId),
653 sendRTMPStreamInVideo(servers[0].url, servers[0].accessToken, liveVideoReplayId)
654 ])
655
656 await Promise.all([
657 waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoId),
658 waitUntilLiveStarts(servers[0].url, servers[0].accessToken, liveVideoReplayId)
659 ])
660
661 await killallServers([ servers[0] ])
662 await reRunServer(servers[0])
663
664 await wait(5000)
665 })
666
667 it('Should cleanup lives', async function () {
668 this.timeout(60000)
669
670 const res = await getVideo(servers[0].url, liveVideoId)
671 const video: VideoDetails = res.body
672
673 expect(video.state.id).to.equal(VideoState.LIVE_ENDED)
674 })
675
676 it('Should save a live replay', async function () {
677 this.timeout(60000)
678
679 await waitJobs(servers)
680
681 const res = await getVideo(servers[0].url, liveVideoReplayId)
682 const video: VideoDetails = res.body
683
684 expect(video.state.id).to.equal(VideoState.PUBLISHED)
685 })
686 })
687
688 after(async function () {
689 await cleanupTests(servers)
690 })
691})