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