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 })