server/tests/api/server/jobs.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 { dateIsValid } from '@server/tests/shared'
   6 import { wait } from '@shared/core-utils'
   7 import {
   8   cleanupTests,
   9   createMultipleServers,
  10   doubleFollow,
  11   PeerTubeServer,
  12   setAccessTokensToServers,
  13   waitJobs
  14 } from '@shared/server-commands'
  15
  16 const expect = chai.expect
  17
  18 describe('Test jobs', function () {
  19   let servers: PeerTubeServer[]
  20
  21   before(async function () {
  22     this.timeout(120000)
  23
  24     servers = await createMultipleServers(2)
  25
  26     await setAccessTokensToServers(servers)
  27
  28     // Server 1 and server 2 follow each other
  29     await doubleFollow(servers[0], servers[1])
  30   })
  31
  32   it('Should create some jobs', async function () {
  33     this.timeout(120000)
  34
  35     await servers[1].videos.upload({ attributes: { name: 'video1' } })
  36     await servers[1].videos.upload({ attributes: { name: 'video2' } })
  37
  38     await waitJobs(servers)
  39   })
  40
  41   it('Should list jobs', async function () {
  42     const body = await servers[1].jobs.list({ state: 'completed' })
  43     expect(body.total).to.be.above(2)
  44     expect(body.data).to.have.length.above(2)
  45   })
  46
  47   it('Should list jobs with sort, pagination and job type', async function () {
  48     {
  49       const body = await servers[1].jobs.list({
  50         state: 'completed',
  51         start: 1,
  52         count: 2,
  53         sort: 'createdAt'
  54       })
  55       expect(body.total).to.be.above(2)
  56       expect(body.data).to.have.lengthOf(2)
  57
  58       let job = body.data[0]
  59       // Skip repeat jobs
  60       if (job.type === 'videos-views-stats') job = body.data[1]
  61
  62       expect(job.state).to.equal('completed')
  63       expect(dateIsValid(job.createdAt as string)).to.be.true
  64       expect(dateIsValid(job.processedOn as string)).to.be.true
  65       expect(dateIsValid(job.finishedOn as string)).to.be.true
  66     }
  67
  68     {
  69       const body = await servers[1].jobs.list({
  70         state: 'completed',
  71         start: 0,
  72         count: 100,
  73         sort: 'createdAt',
  74         jobType: 'activitypub-http-broadcast'
  75       })
  76       expect(body.total).to.be.above(2)
  77
  78       for (const j of body.data) {
  79         expect(j.type).to.equal('activitypub-http-broadcast')
  80       }
  81     }
  82   })
  83
  84   it('Should list all jobs', async function () {
  85     const body = await servers[1].jobs.list()
  86     expect(body.total).to.be.above(2)
  87
  88     const jobs = body.data
  89     expect(jobs).to.have.length.above(2)
  90
  91     expect(jobs.find(j => j.state === 'completed')).to.not.be.undefined
  92   })
  93
  94   it('Should pause the job queue', async function () {
  95     this.timeout(120000)
  96
  97     const { uuid } = await servers[1].videos.upload({ attributes: { name: 'video2' } })
  98     await waitJobs(servers)
  99
 100     await servers[1].jobs.pauseJobQueue()
 101     await servers[1].videos.runTranscoding({ videoId: uuid, transcodingType: 'hls' })
 102
 103     await wait(5000)
 104
 105     {
 106       const body = await servers[1].jobs.list({ state: 'waiting', jobType: 'video-transcoding' })
 107       // waiting includes waiting-children
 108       expect(body.data).to.have.lengthOf(4)
 109     }
 110
 111     {
 112       const body = await servers[1].jobs.list({ state: 'waiting-children', jobType: 'video-transcoding' })
 113       expect(body.data).to.have.lengthOf(1)
 114     }
 115   })
 116
 117   it('Should resume the job queue', async function () {
 118     this.timeout(120000)
 119
 120     await servers[1].jobs.resumeJobQueue()
 121
 122     await waitJobs(servers)
 123
 124     const body = await servers[1].jobs.list({ state: 'waiting', jobType: 'video-transcoding' })
 125     expect(body.data).to.have.lengthOf(0)
 126   })
 127
 128   after(async function () {
 129     await cleanupTests(servers)
 130   })
 131 })