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