server/tests/cli/optimize-old-videos.ts

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