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   cleanupTests,
   8   doubleFollow,
   9   createMultipleServers,
  10   generateHighBitrateVideo,
  11   PeerTubeServer,
  12   setAccessTokensToServers,
  13   wait,
  14   waitJobs
  15 } from '@shared/extra-utils'
  16 import { getMaxBitrate, VideoResolution } from '@shared/models'
  17 import { getVideoFileBitrate, getVideoFileFPS, getVideoFileResolution } from '../../helpers/ffprobe-utils'
  18 import { VIDEO_TRANSCODING_FPS } from '../../initializers/constants'
  19
  20 const expect = chai.expect
  21
  22 describe('Test optimize old videos', function () {
  23   let servers: PeerTubeServer[] = []
  24
  25   before(async function () {
  26     this.timeout(200000)
  27
  28     // Run server 2 to have transcoding enabled
  29     servers = await createMultipleServers(2)
  30     await setAccessTokensToServers(servers)
  31
  32     await doubleFollow(servers[0], servers[1])
  33
  34     let tempFixturePath: string
  35
  36     {
  37       tempFixturePath = await generateHighBitrateVideo()
  38
  39       const bitrate = await getVideoFileBitrate(tempFixturePath)
  40       expect(bitrate).to.be.above(getMaxBitrate(VideoResolution.H_1080P, 25, VIDEO_TRANSCODING_FPS))
  41     }
  42
  43     // Upload two videos for our needs
  44     await servers[0].videos.upload({ attributes: { name: 'video1', fixture: tempFixturePath } })
  45     await servers[0].videos.upload({ attributes: { name: 'video2', fixture: tempFixturePath } })
  46
  47     await waitJobs(servers)
  48   })
  49
  50   it('Should have two video files on each server', async function () {
  51     this.timeout(30000)
  52
  53     for (const server of servers) {
  54       const { data } = await server.videos.list()
  55       expect(data).to.have.lengthOf(2)
  56
  57       for (const video of data) {
  58         const videoDetails = await server.videos.get({ id: video.uuid })
  59         expect(videoDetails.files).to.have.lengthOf(1)
  60       }
  61     }
  62   })
  63
  64   it('Should run optimize script', async function () {
  65     this.timeout(200000)
  66
  67     await servers[0].cli.execWithEnv('npm run optimize-old-videos')
  68     await waitJobs(servers)
  69
  70     for (const server of servers) {
  71       const { data } = await server.videos.list()
  72       expect(data).to.have.lengthOf(2)
  73
  74       for (const video of data) {
  75         await server.videos.view({ id: video.uuid })
  76
  77         // Refresh video
  78         await waitJobs(servers)
  79         await wait(5000)
  80         await waitJobs(servers)
  81
  82         const videoDetails = await server.videos.get({ id: video.uuid })
  83
  84         expect(videoDetails.files).to.have.lengthOf(1)
  85         const file = videoDetails.files[0]
  86
  87         expect(file.size).to.be.below(8000000)
  88
  89         const path = servers[0].servers.buildDirectory(join('videos', video.uuid + '-' + file.resolution.id + '.mp4'))
  90         const bitrate = await getVideoFileBitrate(path)
  91         const fps = await getVideoFileFPS(path)
  92         const resolution = await getVideoFileResolution(path)
  93
  94         expect(resolution.videoFileResolution).to.equal(file.resolution.id)
  95         expect(bitrate).to.be.below(getMaxBitrate(resolution.videoFileResolution, fps, VIDEO_TRANSCODING_FPS))
  96       }
  97     }
  98   })
  99
 100   after(async function () {
 101     await cleanupTests(servers)
 102   })
 103 })