server/tests/cli/create-transcoding-job.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 {
   6   cleanupTests,
   7   doubleFollow,
   8   createMultipleServers,
   9   PeerTubeServer,
  10   setAccessTokensToServers,
  11   waitJobs
  12 } from '../../../shared/extra-utils'
  13
  14 const expect = chai.expect
  15
  16 describe('Test create transcoding jobs', function () {
  17   let servers: PeerTubeServer[] = []
  18   const videosUUID: string[] = []
  19
  20   const config = {
  21     transcoding: {
  22       enabled: false,
  23       resolutions: {
  24         '240p': true,
  25         '360p': true,
  26         '480p': true,
  27         '720p': true,
  28         '1080p': true,
  29         '1440p': true,
  30         '2160p': true
  31       },
  32       hls: {
  33         enabled: false
  34       }
  35     }
  36   }
  37
  38   before(async function () {
  39     this.timeout(60000)
  40
  41     // Run server 2 to have transcoding enabled
  42     servers = await createMultipleServers(2)
  43     await setAccessTokensToServers(servers)
  44
  45     await servers[0].config.updateCustomSubConfig({ newConfig: config })
  46
  47     await doubleFollow(servers[0], servers[1])
  48
  49     for (let i = 1; i <= 5; i++) {
  50       const { uuid } = await servers[0].videos.upload({ attributes: { name: 'video' + i } })
  51       videosUUID.push(uuid)
  52     }
  53
  54     await waitJobs(servers)
  55   })
  56
  57   it('Should have two video files on each server', async function () {
  58     this.timeout(30000)
  59
  60     for (const server of servers) {
  61       const { data } = await server.videos.list()
  62       expect(data).to.have.lengthOf(videosUUID.length)
  63
  64       for (const video of data) {
  65         const videoDetail = await server.videos.get({ id: video.uuid })
  66         expect(videoDetail.files).to.have.lengthOf(1)
  67         expect(videoDetail.streamingPlaylists).to.have.lengthOf(0)
  68       }
  69     }
  70   })
  71
  72   it('Should run a transcoding job on video 2', async function () {
  73     this.timeout(60000)
  74
  75     await servers[0].cli.execWithEnv(`npm run create-transcoding-job -- -v ${videosUUID[1]}`)
  76     await waitJobs(servers)
  77
  78     for (const server of servers) {
  79       const { data } = await server.videos.list()
  80
  81       let infoHashes: { [id: number]: string }
  82
  83       for (const video of data) {
  84         const videoDetail = await server.videos.get({ id: video.uuid })
  85
  86         if (video.uuid === videosUUID[1]) {
  87           expect(videoDetail.files).to.have.lengthOf(4)
  88           expect(videoDetail.streamingPlaylists).to.have.lengthOf(0)
  89
  90           if (!infoHashes) {
  91             infoHashes = {}
  92
  93             for (const file of videoDetail.files) {
  94               infoHashes[file.resolution.id.toString()] = file.magnetUri
  95             }
  96           } else {
  97             for (const resolution of Object.keys(infoHashes)) {
  98               const file = videoDetail.files.find(f => f.resolution.id.toString() === resolution)
  99               expect(file.magnetUri).to.equal(infoHashes[resolution])
 100             }
 101           }
 102         } else {
 103           expect(videoDetail.files).to.have.lengthOf(1)
 104           expect(videoDetail.streamingPlaylists).to.have.lengthOf(0)
 105         }
 106       }
 107     }
 108   })
 109
 110   it('Should run a transcoding job on video 1 with resolution', async function () {
 111     this.timeout(60000)
 112
 113     await servers[0].cli.execWithEnv(`npm run create-transcoding-job -- -v ${videosUUID[0]} -r 480`)
 114
 115     await waitJobs(servers)
 116
 117     for (const server of servers) {
 118       const { data } = await server.videos.list()
 119       expect(data).to.have.lengthOf(videosUUID.length)
 120
 121       const videoDetails = await server.videos.get({ id: videosUUID[0] })
 122
 123       expect(videoDetails.files).to.have.lengthOf(2)
 124       expect(videoDetails.files[0].resolution.id).to.equal(720)
 125       expect(videoDetails.files[1].resolution.id).to.equal(480)
 126
 127       expect(videoDetails.streamingPlaylists).to.have.lengthOf(0)
 128     }
 129   })
 130
 131   it('Should generate an HLS resolution', async function () {
 132     this.timeout(120000)
 133
 134     await servers[0].cli.execWithEnv(`npm run create-transcoding-job -- -v ${videosUUID[2]} --generate-hls -r 480`)
 135
 136     await waitJobs(servers)
 137
 138     for (const server of servers) {
 139       const videoDetails = await server.videos.get({ id: videosUUID[2] })
 140
 141       expect(videoDetails.files).to.have.lengthOf(1)
 142       expect(videoDetails.streamingPlaylists).to.have.lengthOf(1)
 143
 144       const files = videoDetails.streamingPlaylists[0].files
 145       expect(files).to.have.lengthOf(1)
 146       expect(files[0].resolution.id).to.equal(480)
 147     }
 148   })
 149
 150   it('Should not duplicate an HLS resolution', async function () {
 151     this.timeout(120000)
 152
 153     await servers[0].cli.execWithEnv(`npm run create-transcoding-job -- -v ${videosUUID[2]} --generate-hls -r 480`)
 154
 155     await waitJobs(servers)
 156
 157     for (const server of servers) {
 158       const videoDetails = await server.videos.get({ id: videosUUID[2] })
 159
 160       const files = videoDetails.streamingPlaylists[0].files
 161       expect(files).to.have.lengthOf(1)
 162       expect(files[0].resolution.id).to.equal(480)
 163     }
 164   })
 165
 166   it('Should generate all HLS resolutions', async function () {
 167     this.timeout(120000)
 168
 169     await servers[0].cli.execWithEnv(`npm run create-transcoding-job -- -v ${videosUUID[3]} --generate-hls`)
 170
 171     await waitJobs(servers)
 172
 173     for (const server of servers) {
 174       const videoDetails = await server.videos.get({ id: videosUUID[3] })
 175
 176       expect(videoDetails.files).to.have.lengthOf(1)
 177       expect(videoDetails.streamingPlaylists).to.have.lengthOf(1)
 178
 179       const files = videoDetails.streamingPlaylists[0].files
 180       expect(files).to.have.lengthOf(4)
 181     }
 182   })
 183
 184   it('Should optimize the video file and generate HLS videos if enabled in config', async function () {
 185     this.timeout(120000)
 186
 187     config.transcoding.hls.enabled = true
 188     await servers[0].config.updateCustomSubConfig({ newConfig: config })
 189
 190     await servers[0].cli.execWithEnv(`npm run create-transcoding-job -- -v ${videosUUID[4]}`)
 191
 192     await waitJobs(servers)
 193
 194     for (const server of servers) {
 195       const videoDetails = await server.videos.get({ id: videosUUID[4] })
 196
 197       expect(videoDetails.files).to.have.lengthOf(4)
 198       expect(videoDetails.streamingPlaylists).to.have.lengthOf(1)
 199       expect(videoDetails.streamingPlaylists[0].files).to.have.lengthOf(4)
 200     }
 201   })
 202
 203   after(async function () {
 204     await cleanupTests(servers)
 205   })
 206 })