1 /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
3 import { expect } from 'chai'
4 import { wait } from '@shared/core-utils'
5 import { LiveVideoError, VideoPrivacy } from '@shared/models'
12 setAccessTokensToServers,
13 setDefaultVideoChannel,
16 waitUntilLiveReplacedByReplayOnAllServers,
17 waitUntilLiveWaitingOnAllServers
18 } from '@shared/server-commands'
19 import { checkLiveCleanup } from '../../shared'
21 describe('Test live constraints', function () {
22 let servers: PeerTubeServer[] = []
24 let userAccessToken: string
25 let userChannelId: number
27 async function createLiveWrapper (options: {
31 const { replay, permanent } = options
33 const liveAttributes = {
35 channelId: userChannelId,
36 privacy: VideoPrivacy.PUBLIC,
38 permanentLive: permanent
41 const { uuid } = await servers[0].live.create({ token: userAccessToken, fields: liveAttributes })
45 async function checkSaveReplay (videoId: string, resolutions = [ 720 ]) {
46 for (const server of servers) {
47 const video = await server.videos.get({ id: videoId })
48 expect(video.isLive).to.be.false
49 expect(video.duration).to.be.greaterThan(0)
52 await checkLiveCleanup({ server: servers[0], permanent: false, videoUUID: videoId, savedResolutions: resolutions })
55 function updateQuota (options: { total: number, daily: number }) {
56 return servers[0].users.update({
58 videoQuota: options.total,
59 videoQuotaDaily: options.daily
63 before(async function () {
66 servers = await createMultipleServers(2)
68 // Get the access tokens
69 await setAccessTokensToServers(servers)
70 await setDefaultVideoChannel(servers)
72 await servers[0].config.updateCustomSubConfig({
85 const res = await servers[0].users.generate('user1')
87 userChannelId = res.userChannelId
88 userAccessToken = res.token
90 await updateQuota({ total: 1, daily: -1 })
93 // Server 1 and server 2 follow each other
94 await doubleFollow(servers[0], servers[1])
97 it('Should not have size limit if save replay is disabled', async function () {
100 const userVideoLiveoId = await createLiveWrapper({ replay: false, permanent: false })
101 await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: false })
104 it('Should have size limit depending on user global quota if save replay is enabled on non permanent live', async function () {
107 // Wait for user quota memoize cache invalidation
110 const userVideoLiveoId = await createLiveWrapper({ replay: true, permanent: false })
111 await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: true })
113 await waitUntilLiveReplacedByReplayOnAllServers(servers, userVideoLiveoId)
114 await waitJobs(servers)
116 await checkSaveReplay(userVideoLiveoId)
118 const session = await servers[0].live.getReplaySession({ videoId: userVideoLiveoId })
119 expect(session.error).to.equal(LiveVideoError.QUOTA_EXCEEDED)
122 it('Should have size limit depending on user global quota if save replay is enabled on a permanent live', async function () {
125 // Wait for user quota memoize cache invalidation
128 const userVideoLiveoId = await createLiveWrapper({ replay: true, permanent: true })
129 await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: true })
131 await waitJobs(servers)
132 await waitUntilLiveWaitingOnAllServers(servers, userVideoLiveoId)
134 const session = await servers[0].live.findLatestSession({ videoId: userVideoLiveoId })
135 expect(session.error).to.equal(LiveVideoError.QUOTA_EXCEEDED)
138 it('Should have size limit depending on user daily quota if save replay is enabled', async function () {
141 // Wait for user quota memoize cache invalidation
144 await updateQuota({ total: -1, daily: 1 })
146 const userVideoLiveoId = await createLiveWrapper({ replay: true, permanent: false })
147 await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: true })
149 await waitUntilLiveReplacedByReplayOnAllServers(servers, userVideoLiveoId)
150 await waitJobs(servers)
152 await checkSaveReplay(userVideoLiveoId)
154 const session = await servers[0].live.getReplaySession({ videoId: userVideoLiveoId })
155 expect(session.error).to.equal(LiveVideoError.QUOTA_EXCEEDED)
158 it('Should succeed without quota limit', async function () {
161 // Wait for user quota memoize cache invalidation
164 await updateQuota({ total: 10 * 1000 * 1000, daily: -1 })
166 const userVideoLiveoId = await createLiveWrapper({ replay: true, permanent: false })
167 await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: false })
170 it('Should have the same quota in admin and as a user', async function () {
173 const userVideoLiveoId = await createLiveWrapper({ replay: true, permanent: false })
174 const ffmpegCommand = await servers[0].live.sendRTMPStreamInVideo({ token: userAccessToken, videoId: userVideoLiveoId })
176 await servers[0].live.waitUntilPublished({ videoId: userVideoLiveoId })
180 const quotaUser = await servers[0].users.getMyQuotaUsed({ token: userAccessToken })
182 const { data } = await servers[0].users.list()
183 const quotaAdmin = data.find(u => u.username === 'user1')
185 expect(quotaUser.videoQuotaUsed).to.equal(quotaAdmin.videoQuotaUsed)
186 expect(quotaUser.videoQuotaUsedDaily).to.equal(quotaAdmin.videoQuotaUsedDaily)
188 expect(quotaUser.videoQuotaUsed).to.be.above(10)
189 expect(quotaUser.videoQuotaUsedDaily).to.be.above(10)
191 await stopFfmpeg(ffmpegCommand)
194 it('Should have max duration limit', async function () {
197 await servers[0].config.updateCustomSubConfig({
205 resolutions: ConfigCommand.getCustomConfigResolutions(true)
211 const userVideoLiveoId = await createLiveWrapper({ replay: true, permanent: false })
212 await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: true })
214 await waitUntilLiveReplacedByReplayOnAllServers(servers, userVideoLiveoId)
215 await waitJobs(servers)
217 await checkSaveReplay(userVideoLiveoId, [ 720, 480, 360, 240, 144 ])
219 const session = await servers[0].live.getReplaySession({ videoId: userVideoLiveoId })
220 expect(session.error).to.equal(LiveVideoError.DURATION_EXCEEDED)
223 after(async function () {
224 await cleanupTests(servers)