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/extra-utils'
  19 import { waitJobs } from '../../../shared/extra-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     // Run server 2 to have transcoding enabled
  35     servers = await flushAndRunMultipleServers(2)
  36     await setAccessTokensToServers(servers)
  37
  38     await doubleFollow(servers[0], servers[1])
  39
  40     let tempFixturePath: string
  41
  42     {
  43       tempFixturePath = await generateHighBitrateVideo()
  44
  45       const bitrate = await getVideoFileBitrate(tempFixturePath)
  46       expect(bitrate).to.be.above(getMaxBitrate(VideoResolution.H_1080P, 60, VIDEO_TRANSCODING_FPS))
  47     }
  48
  49     // Upload two videos for our needs
  50     const res1 = await uploadVideo(servers[0].url, servers[0].accessToken, { name: 'video1', fixture: tempFixturePath })
  51     video1UUID = res1.body.video.uuid
  52     const res2 = await uploadVideo(servers[0].url, servers[0].accessToken, { name: 'video2', fixture: tempFixturePath })
  53     video2UUID = res2.body.video.uuid
  54
  55     await waitJobs(servers)
  56   })
  57
  58   it('Should have two video files on each server', async function () {
  59     this.timeout(30000)
  60
  61     for (const server of servers) {
  62       const res = await getVideosList(server.url)
  63       const videos = res.body.data
  64       expect(videos).to.have.lengthOf(2)
  65
  66       for (const video of videos) {
  67         const res2 = await getVideo(server.url, video.uuid)
  68         const videoDetail: VideoDetails = res2.body
  69         expect(videoDetail.files).to.have.lengthOf(1)
  70       }
  71     }
  72   })
  73
  74   it('Should run optimize script', async function () {
  75     this.timeout(120000)
  76
  77     const env = getEnvCli(servers[0])
  78     await execCLI(`${env} npm run optimize-old-videos`)
  79
  80     await waitJobs(servers)
  81
  82     for (const server of servers) {
  83       const res = await getVideosList(server.url)
  84       const videos: Video[] = res.body.data
  85
  86       expect(videos).to.have.lengthOf(2)
  87
  88       for (const video of videos) {
  89         await viewVideo(server.url, video.uuid)
  90
  91         // Refresh video
  92         await waitJobs(servers)
  93         await wait(5000)
  94         await waitJobs(servers)
  95
  96         const res2 = await getVideo(server.url, video.uuid)
  97         const videosDetails: VideoDetails = res2.body
  98
  99         expect(videosDetails.files).to.have.lengthOf(1)
 100         const file = videosDetails.files[0]
 101
 102         expect(file.size).to.be.below(5000000)
 103
 104         const path = join(root(), 'test1', 'videos', video.uuid + '-' + file.resolution.id + '.mp4')
 105         const bitrate = await getVideoFileBitrate(path)
 106         const fps = await getVideoFileFPS(path)
 107         const resolution = await getVideoFileResolution(path)
 108
 109         expect(resolution.videoFileResolution).to.equal(file.resolution.id)
 110         expect(bitrate).to.be.below(getMaxBitrate(resolution.videoFileResolution, fps, VIDEO_TRANSCODING_FPS))
 111       }
 112     }
 113   })
 114
 115   after(function () {
 116     killallServers(servers)
 117   })
 118 })