2 import { expect } from 'chai'
3 import { wait } from '@shared/core-utils'
4 import { JobState, JobType, RunnerJobState } from '../../models'
5 import { PeerTubeServer } from './server'
7 async function waitJobs (
8 serversArg: PeerTubeServer[] | PeerTubeServer,
10 skipDelayed?: boolean // default false
11 runnerJobs?: boolean // default false
14 const { skipDelayed = false, runnerJobs = false } = options
16 const pendingJobWait = process.env.NODE_PENDING_JOB_WAIT
17 ? parseInt(process.env.NODE_PENDING_JOB_WAIT, 10)
20 let servers: PeerTubeServer[]
22 if (Array.isArray(serversArg) === false) servers = [ serversArg as PeerTubeServer ]
23 else servers = serversArg as PeerTubeServer[]
25 const states: JobState[] = [ 'waiting', 'active' ]
26 if (!skipDelayed) states.push('delayed')
28 const repeatableJobs: JobType[] = [ 'videos-views-stats', 'activitypub-cleaner' ]
29 let pendingRequests: boolean
31 function tasksBuilder () {
32 const tasks: Promise<any>[] = []
34 // Check if each server has pending request
35 for (const server of servers) {
36 for (const state of states) {
38 const jobPromise = server.jobs.list({
43 }).then(body => body.data)
44 .then(jobs => jobs.filter(j => !repeatableJobs.includes(j.type)))
46 if (jobs.length !== 0) {
47 pendingRequests = true
51 tasks.push(jobPromise)
54 const debugPromise = server.debug.getDebug()
56 if (obj.activityPubMessagesWaiting !== 0) {
57 pendingRequests = true
60 tasks.push(debugPromise)
63 const runnerJobsPromise = server.runnerJobs.list({ count: 100 })
65 for (const job of data) {
66 if (job.state.id !== RunnerJobState.COMPLETED) {
67 pendingRequests = true
71 tasks.push(runnerJobsPromise)
80 pendingRequests = false
81 await Promise.all(tasksBuilder())
83 // Retry, in case of new jobs were created
84 if (pendingRequests === false) {
85 await wait(pendingJobWait)
86 await Promise.all(tasksBuilder())
89 if (pendingRequests) {
90 await wait(pendingJobWait)
92 } while (pendingRequests)
95 async function expectNoFailedTranscodingJob (server: PeerTubeServer) {
96 const { data } = await server.jobs.listFailed({ jobType: 'video-transcoding' })
97 expect(data).to.have.lengthOf(0)
100 // ---------------------------------------------------------------------------
104 expectNoFailedTranscodingJob