]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/tests/cli/optimize-old-videos.ts
Introduce CLI command
[github/Chocobozzz/PeerTube.git] / server / tests / cli / optimize-old-videos.ts
1 /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
2
3 import 'mocha'
4 import * as chai from 'chai'
5 import { join } from 'path'
6 import {
7 buildServerDirectory,
8 cleanupTests,
9 doubleFollow,
10 flushAndRunMultipleServers,
11 generateHighBitrateVideo,
12 getVideo,
13 getVideosList,
14 ServerInfo,
15 setAccessTokensToServers,
16 uploadVideo,
17 viewVideo,
18 wait
19 } from '../../../shared/extra-utils'
20 import { waitJobs } from '../../../shared/extra-utils/server/jobs'
21 import { getMaxBitrate, Video, VideoDetails, VideoResolution } from '../../../shared/models/videos'
22 import { getVideoFileBitrate, getVideoFileFPS, getVideoFileResolution } from '../../helpers/ffprobe-utils'
23 import { VIDEO_TRANSCODING_FPS } from '../../initializers/constants'
24
25 const expect = chai.expect
26
27 describe('Test optimize old videos', function () {
28 let servers: ServerInfo[] = []
29
30 before(async function () {
31 this.timeout(200000)
32
33 // Run server 2 to have transcoding enabled
34 servers = await flushAndRunMultipleServers(2)
35 await setAccessTokensToServers(servers)
36
37 await doubleFollow(servers[0], servers[1])
38
39 let tempFixturePath: string
40
41 {
42 tempFixturePath = await generateHighBitrateVideo()
43
44 const bitrate = await getVideoFileBitrate(tempFixturePath)
45 expect(bitrate).to.be.above(getMaxBitrate(VideoResolution.H_1080P, 25, VIDEO_TRANSCODING_FPS))
46 }
47
48 // Upload two videos for our needs
49 await uploadVideo(servers[0].url, servers[0].accessToken, { name: 'video1', fixture: tempFixturePath })
50 await uploadVideo(servers[0].url, servers[0].accessToken, { name: 'video2', fixture: tempFixturePath })
51
52 await waitJobs(servers)
53 })
54
55 it('Should have two video files on each server', async function () {
56 this.timeout(30000)
57
58 for (const server of servers) {
59 const res = await getVideosList(server.url)
60 const videos = res.body.data
61 expect(videos).to.have.lengthOf(2)
62
63 for (const video of videos) {
64 const res2 = await getVideo(server.url, video.uuid)
65 const videoDetail: VideoDetails = res2.body
66 expect(videoDetail.files).to.have.lengthOf(1)
67 }
68 }
69 })
70
71 it('Should run optimize script', async function () {
72 this.timeout(200000)
73
74 await servers[0].cliCommand.execWithEnv('npm run optimize-old-videos')
75 await waitJobs(servers)
76
77 for (const server of servers) {
78 const res = await getVideosList(server.url)
79 const videos: Video[] = res.body.data
80
81 expect(videos).to.have.lengthOf(2)
82
83 for (const video of videos) {
84 await viewVideo(server.url, video.uuid)
85
86 // Refresh video
87 await waitJobs(servers)
88 await wait(5000)
89 await waitJobs(servers)
90
91 const res2 = await getVideo(server.url, video.uuid)
92 const videosDetails: VideoDetails = res2.body
93
94 expect(videosDetails.files).to.have.lengthOf(1)
95 const file = videosDetails.files[0]
96
97 expect(file.size).to.be.below(8000000)
98
99 const path = buildServerDirectory(servers[0], join('videos', video.uuid + '-' + file.resolution.id + '.mp4'))
100 const bitrate = await getVideoFileBitrate(path)
101 const fps = await getVideoFileFPS(path)
102 const resolution = await getVideoFileResolution(path)
103
104 expect(resolution.videoFileResolution).to.equal(file.resolution.id)
105 expect(bitrate).to.be.below(getMaxBitrate(resolution.videoFileResolution, fps, VIDEO_TRANSCODING_FPS))
106 }
107 }
108 })
109
110 after(async function () {
111 await cleanupTests(servers)
112 })
113 })