]>
Commit | Line | Data |
---|---|---|
1 | import * as request from 'supertest' | |
2 | import { Job, JobState } from '../../models' | |
3 | import { ServerInfo, wait } from '../index' | |
4 | ||
5 | function getJobsList (url: string, accessToken: string, state: JobState) { | |
6 | const path = '/api/v1/jobs/' + state | |
7 | ||
8 | return request(url) | |
9 | .get(path) | |
10 | .set('Accept', 'application/json') | |
11 | .set('Authorization', 'Bearer ' + accessToken) | |
12 | .expect(200) | |
13 | .expect('Content-Type', /json/) | |
14 | } | |
15 | ||
16 | function getJobsListPaginationAndSort (url: string, accessToken: string, state: JobState, start: number, count: number, sort: string) { | |
17 | const path = '/api/v1/jobs/' + state | |
18 | ||
19 | return request(url) | |
20 | .get(path) | |
21 | .query({ start }) | |
22 | .query({ count }) | |
23 | .query({ sort }) | |
24 | .set('Accept', 'application/json') | |
25 | .set('Authorization', 'Bearer ' + accessToken) | |
26 | .expect(200) | |
27 | .expect('Content-Type', /json/) | |
28 | } | |
29 | ||
30 | async function waitJobs (serversArg: ServerInfo[] | ServerInfo) { | |
31 | let servers: ServerInfo[] | |
32 | ||
33 | if (Array.isArray(serversArg) === false) servers = [ serversArg as ServerInfo ] | |
34 | else servers = serversArg as ServerInfo[] | |
35 | ||
36 | const states: JobState[] = [ 'waiting', 'active', 'delayed' ] | |
37 | const tasks: Promise<any>[] = [] | |
38 | let pendingRequests: boolean | |
39 | ||
40 | do { | |
41 | pendingRequests = false | |
42 | ||
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')) | |
49 | .then(jobs => { | |
50 | if (jobs.length !== 0) pendingRequests = true | |
51 | }) | |
52 | tasks.push(p) | |
53 | } | |
54 | } | |
55 | ||
56 | await Promise.all(tasks) | |
57 | ||
58 | // Retry, in case of new jobs were created | |
59 | if (pendingRequests === false) { | |
60 | await wait(1000) | |
61 | ||
62 | await Promise.all(tasks) | |
63 | } | |
64 | ||
65 | if (pendingRequests) { | |
66 | await wait(1000) | |
67 | } | |
68 | } while (pendingRequests) | |
69 | } | |
70 | ||
71 | // --------------------------------------------------------------------------- | |
72 | ||
73 | export { | |
74 | getJobsList, | |
75 | waitJobs, | |
76 | getJobsListPaginationAndSort | |
77 | } |