/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */ import 'mocha' import * as chai from 'chai' import { wait } from '@shared/core-utils' import { VideoPrivacy } from '@shared/models' import { cleanupTests, ConfigCommand, createMultipleServers, doubleFollow, PeerTubeServer, setAccessTokensToServers, setDefaultVideoChannel, waitJobs } from '@shared/server-commands' import { checkLiveCleanupAfterSave } from '../../shared' const expect = chai.expect describe('Test live constraints', function () { let servers: PeerTubeServer[] = [] let userId: number let userAccessToken: string let userChannelId: number async function createLiveWrapper (saveReplay: boolean) { const liveAttributes = { name: 'user live', channelId: userChannelId, privacy: VideoPrivacy.PUBLIC, saveReplay } const { uuid } = await servers[0].live.create({ token: userAccessToken, fields: liveAttributes }) return uuid } async function checkSaveReplay (videoId: string, resolutions = [ 720 ]) { for (const server of servers) { const video = await server.videos.get({ id: videoId }) expect(video.isLive).to.be.false expect(video.duration).to.be.greaterThan(0) } await checkLiveCleanupAfterSave(servers[0], videoId, resolutions) } async function waitUntilLivePublishedOnAllServers (videoId: string) { for (const server of servers) { await server.live.waitUntilPublished({ videoId }) } } function updateQuota (options: { total: number, daily: number }) { return servers[0].users.update({ userId, videoQuota: options.total, videoQuotaDaily: options.daily }) } before(async function () { this.timeout(120000) servers = await createMultipleServers(2) // Get the access tokens await setAccessTokensToServers(servers) await setDefaultVideoChannel(servers) await servers[0].config.updateCustomSubConfig({ newConfig: { live: { enabled: true, allowReplay: true, transcoding: { enabled: false } } } }) { const res = await servers[0].users.generate('user1') userId = res.userId userChannelId = res.userChannelId userAccessToken = res.token await updateQuota({ total: 1, daily: -1 }) } // Server 1 and server 2 follow each other await doubleFollow(servers[0], servers[1]) }) it('Should not have size limit if save replay is disabled', async function () { this.timeout(60000) const userVideoLiveoId = await createLiveWrapper(false) await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: false }) }) it('Should have size limit depending on user global quota if save replay is enabled', async function () { this.timeout(60000) // Wait for user quota memoize cache invalidation await wait(5000) const userVideoLiveoId = await createLiveWrapper(true) await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: true }) await waitUntilLivePublishedOnAllServers(userVideoLiveoId) await waitJobs(servers) await checkSaveReplay(userVideoLiveoId) }) it('Should have size limit depending on user daily quota if save replay is enabled', async function () { this.timeout(60000) // Wait for user quota memoize cache invalidation await wait(5000) await updateQuota({ total: -1, daily: 1 }) const userVideoLiveoId = await createLiveWrapper(true) await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: true }) await waitUntilLivePublishedOnAllServers(userVideoLiveoId) await waitJobs(servers) await checkSaveReplay(userVideoLiveoId) }) it('Should succeed without quota limit', async function () { this.timeout(60000) // Wait for user quota memoize cache invalidation await wait(5000) await updateQuota({ total: 10 * 1000 * 1000, daily: -1 }) const userVideoLiveoId = await createLiveWrapper(true) await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: false }) }) it('Should have max duration limit', async function () { this.timeout(60000) await servers[0].config.updateCustomSubConfig({ newConfig: { live: { enabled: true, allowReplay: true, maxDuration: 1, transcoding: { enabled: true, resolutions: ConfigCommand.getCustomConfigResolutions(true) } } } }) const userVideoLiveoId = await createLiveWrapper(true) await servers[0].live.runAndTestStreamError({ token: userAccessToken, videoId: userVideoLiveoId, shouldHaveError: true }) await waitUntilLivePublishedOnAllServers(userVideoLiveoId) await waitJobs(servers) await checkSaveReplay(userVideoLiveoId, [ 720, 480, 360, 240, 144 ]) }) after(async function () { await cleanupTests(servers) }) })