1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
import { expect } from 'chai'
import { wait } from '@shared/core-utils'
import {
cleanupTests,
createSingleServer,
PeerTubeServer,
setAccessTokensToServers,
setDefaultVideoChannel,
waitJobs
} from '@shared/server-commands'
describe('Test runner socket', function () {
let server: PeerTubeServer
let runnerToken: string
before(async function () {
this.timeout(120_000)
server = await createSingleServer(1)
await setAccessTokensToServers([ server ])
await setDefaultVideoChannel([ server ])
await server.config.enableTranscoding(true, true)
await server.config.enableRemoteTranscoding()
runnerToken = await server.runners.autoRegisterRunner()
})
it('Should throw an error without runner token', function (done) {
const localSocket = server.socketIO.getRunnersSocket({ runnerToken: null })
localSocket.on('connect_error', err => {
expect(err.message).to.contain('No runner token provided')
done()
})
})
it('Should throw an error with a bad runner token', function (done) {
const localSocket = server.socketIO.getRunnersSocket({ runnerToken: 'ergag' })
localSocket.on('connect_error', err => {
expect(err.message).to.contain('Invalid runner token')
done()
})
})
it('Should not send ping if there is no available jobs', async function () {
let pings = 0
const localSocket = server.socketIO.getRunnersSocket({ runnerToken })
localSocket.on('available-jobs', () => pings++)
expect(pings).to.equal(0)
})
it('Should send a ping on available job', async function () {
let pings = 0
const localSocket = server.socketIO.getRunnersSocket({ runnerToken })
localSocket.on('available-jobs', () => pings++)
await server.videos.quickUpload({ name: 'video1' })
// Wait for debounce
await wait(1000)
await waitJobs([ server ])
expect(pings).to.equal(1)
await server.videos.quickUpload({ name: 'video2' })
// Wait for debounce
await wait(1000)
await waitJobs([ server ])
expect(pings).to.equal(2)
await server.runnerJobs.cancelAllJobs()
})
it('Should send a ping when a child is ready', async function () {
let pings = 0
const localSocket = server.socketIO.getRunnersSocket({ runnerToken })
localSocket.on('available-jobs', () => pings++)
await server.videos.quickUpload({ name: 'video3' })
// Wait for debounce
await wait(1000)
await waitJobs([ server ])
expect(pings).to.equal(1)
await server.runnerJobs.autoProcessWebVideoJob(runnerToken)
// Wait for debounce
await wait(1000)
await waitJobs([ server ])
expect(pings).to.equal(2)
})
it('Should not send a ping if the ended job does not have a child', async function () {
let pings = 0
const localSocket = server.socketIO.getRunnersSocket({ runnerToken })
localSocket.on('available-jobs', () => pings++)
const { availableJobs } = await server.runnerJobs.request({ runnerToken })
const job = availableJobs.find(j => j.type === 'vod-web-video-transcoding')
await server.runnerJobs.autoProcessWebVideoJob(runnerToken, job.uuid)
// Wait for debounce
await wait(1000)
await waitJobs([ server ])
expect(pings).to.equal(0)
})
after(async function () {
await cleanupTests([ server ])
})
})
|
