2 import { expect } from 'chai'
3 import { checkPeerTubeRunnerCacheIsEmpty, checkVideoDuration, expectStartWith, PeerTubeRunnerProcess } from '@server/tests/shared'
4 import { areMockObjectStorageTestsDisabled, getAllFiles, wait } from '@shared/core-utils'
11 setAccessTokensToServers,
12 setDefaultVideoChannel,
15 } from '@shared/server-commands'
17 describe('Test studio transcoding in peertube-runner program', function () {
18 let servers: PeerTubeServer[] = []
19 let peertubeRunner: PeerTubeRunnerProcess
21 function runSuite (options: {
22 objectStorage?: ObjectStorageCommand
24 const { objectStorage } = options
26 it('Should run a complex studio transcoding', async function () {
29 const { uuid } = await servers[0].videos.quickUpload({ name: 'mp4', fixture: 'video_short.mp4' })
30 await waitJobs(servers)
32 const video = await servers[0].videos.get({ id: uuid })
33 const oldFileUrls = getAllFiles(video).map(f => f.fileUrl)
35 await servers[0].videoStudio.createEditionTasks({ videoId: uuid, tasks: VideoStudioCommand.getComplexTask() })
36 await waitJobs(servers, { runnerJobs: true })
38 for (const server of servers) {
39 const video = await server.videos.get({ id: uuid })
40 const files = getAllFiles(video)
42 for (const f of files) {
43 expect(oldFileUrls).to.not.include(f.fileUrl)
47 for (const webtorrentFile of video.files) {
48 expectStartWith(webtorrentFile.fileUrl, objectStorage.getMockWebVideosBaseUrl())
51 for (const hlsFile of video.streamingPlaylists[0].files) {
52 expectStartWith(hlsFile.fileUrl, objectStorage.getMockPlaylistBaseUrl())
56 await checkVideoDuration(server, uuid, 9)
61 before(async function () {
64 servers = await createMultipleServers(2)
66 await setAccessTokensToServers(servers)
67 await setDefaultVideoChannel(servers)
69 await doubleFollow(servers[0], servers[1])
71 await servers[0].config.enableTranscoding(true, true)
72 await servers[0].config.enableStudio()
73 await servers[0].config.enableRemoteStudio()
75 const registrationToken = await servers[0].runnerRegistrationTokens.getFirstRegistrationToken()
77 peertubeRunner = new PeerTubeRunnerProcess(servers[0])
78 await peertubeRunner.runServer()
79 await peertubeRunner.registerPeerTubeInstance({ registrationToken, runnerName: 'runner' })
82 describe('With videos on local filesystem storage', function () {
86 describe('With videos on object storage', function () {
87 if (areMockObjectStorageTestsDisabled()) return
89 const objectStorage = new ObjectStorageCommand()
91 before(async function () {
92 await objectStorage.prepareDefaultMockBuckets()
94 await servers[0].kill()
96 await servers[0].run(objectStorage.getDefaultMockConfig())
98 // Wait for peertube runner socket reconnection
102 runSuite({ objectStorage })
104 after(async function () {
105 await objectStorage.cleanupMock()
109 describe('Check cleanup', function () {
111 it('Should have an empty cache directory', async function () {
112 await checkPeerTubeRunnerCacheIsEmpty(peertubeRunner)
116 after(async function () {
117 if (peertubeRunner) {
118 await peertubeRunner.unregisterPeerTubeInstance({ runnerName: 'runner' })
119 peertubeRunner.kill()
122 await cleanupTests(servers)