1 import * as request from 'supertest'
2 import { Job, JobState } from '../../../../shared/models'
3 import { ServerInfo, wait } from '../index'
5 function getJobsList (url: string, accessToken: string, state: JobState) {
6 const path = '/api/v1/jobs/' + state
10 .set('Accept', 'application/json')
11 .set('Authorization', 'Bearer ' + accessToken)
13 .expect('Content-Type', /json/)
16 function getJobsListPaginationAndSort (url: string, accessToken: string, state: JobState, start: number, count: number, sort: string) {
17 const path = '/api/v1/jobs/' + state
24 .set('Accept', 'application/json')
25 .set('Authorization', 'Bearer ' + accessToken)
27 .expect('Content-Type', /json/)
30 async function waitJobs (serversArg: ServerInfo[] | ServerInfo) {
31 let servers: ServerInfo[]
33 if (Array.isArray(serversArg) === false) servers = [ serversArg as ServerInfo ]
34 else servers = serversArg as ServerInfo[]
36 const states: JobState[] = [ 'waiting', 'active', 'delayed' ]
37 const tasks: Promise<any>[] = []
38 let pendingRequests: boolean
41 pendingRequests = false
43 // Check if each server has pending request
44 for (const server of servers) {
45 for (const state of states) {
46 const p = getJobsListPaginationAndSort(server.url, server.accessToken, state, 0, 10, '-createdAt')
47 .then(res => res.body.data)
48 .then((jobs: Job[]) => jobs.filter(j => j.type !== 'videos-views'))
50 if (jobs.length !== 0) pendingRequests = true
56 await Promise.all(tasks)
58 // Retry, in case of new jobs were created
59 if (pendingRequests === false) {
62 await Promise.all(tasks)
65 if (pendingRequests) {
68 } while (pendingRequests)
71 // ---------------------------------------------------------------------------
76 getJobsListPaginationAndSort