import 'mocha'
import * as chai from 'chai'
-import { join } from 'path'
-import { ffprobePromise, getVideoStreamFromFile } from '@server/helpers/ffprobe-utils'
-import { HttpStatusCode } from '@shared/core-utils'
+import { basename, join } from 'path'
+import { ffprobePromise, getVideoStream } from '@server/helpers/ffmpeg'
+import { checkLiveCleanup, checkLiveSegmentHash, checkResolutionsInMasterPlaylist, testImage } from '@server/tests/shared'
+import { wait } from '@shared/core-utils'
+import {
+ HttpStatusCode,
+ LiveVideo,
+ LiveVideoCreate,
+ LiveVideoLatencyMode,
+ VideoDetails,
+ VideoPrivacy,
+ VideoState,
+ VideoStreamingPlaylistType
+} from '@shared/models'
import {
- checkLiveCleanup,
- checkLiveSegmentHash,
- checkResolutionsInMasterPlaylist,
cleanupTests,
+ createMultipleServers,
doubleFollow,
- flushAndRunMultipleServers,
killallServers,
LiveCommand,
makeRawRequest,
- reRunServer,
+ PeerTubeServer,
sendRTMPStream,
- ServerInfo,
setAccessTokensToServers,
setDefaultVideoChannel,
stopFfmpeg,
testFfmpegStreamError,
- testImage,
- wait,
waitJobs,
waitUntilLivePublishedOnAllServers
-} from '@shared/extra-utils'
-import { LiveVideo, LiveVideoCreate, VideoDetails, VideoPrivacy, VideoState, VideoStreamingPlaylistType } from '@shared/models'
+} from '@shared/server-commands'
const expect = chai.expect
describe('Test live', function () {
- let servers: ServerInfo[] = []
+ let servers: PeerTubeServer[] = []
let commands: LiveCommand[]
before(async function () {
this.timeout(120000)
- servers = await flushAndRunMultipleServers(2)
+ servers = await createMultipleServers(2)
// Get the access tokens
await setAccessTokensToServers(servers)
live: {
enabled: true,
allowReplay: true,
+ latencySetting: {
+ enabled: true
+ },
transcoding: {
enabled: false
}
commentsEnabled: false,
downloadEnabled: false,
saveReplay: true,
+ latencyMode: LiveVideoLatencyMode.SMALL_LATENCY,
privacy: VideoPrivacy.PUBLIC,
previewfile: 'video_short1-preview.webm.jpg',
thumbnailfile: 'video_short1.webm.jpg'
expect(live.rtmpUrl).to.equal('rtmp://' + server.hostname + ':' + servers[0].rtmpPort + '/live')
expect(live.streamKey).to.not.be.empty
} else {
- expect(live.rtmpUrl).to.be.null
- expect(live.streamKey).to.be.null
+ expect(live.rtmpUrl).to.not.exist
+ expect(live.streamKey).to.not.exist
}
expect(live.saveReplay).to.be.true
+ expect(live.latencyMode).to.equal(LiveVideoLatencyMode.SMALL_LATENCY)
}
})
it('Should update the live', async function () {
this.timeout(10000)
- await commands[0].update({ videoId: liveVideoUUID, fields: { saveReplay: false } })
+ await commands[0].update({ videoId: liveVideoUUID, fields: { saveReplay: false, latencyMode: LiveVideoLatencyMode.DEFAULT } })
await waitJobs(servers)
})
expect(live.rtmpUrl).to.equal('rtmp://' + server.hostname + ':' + servers[0].rtmpPort + '/live')
expect(live.streamKey).to.not.be.empty
} else {
- expect(live.rtmpUrl).to.be.null
- expect(live.streamKey).to.be.null
+ expect(live.rtmpUrl).to.not.exist
+ expect(live.streamKey).to.not.exist
}
expect(live.saveReplay).to.be.false
+ expect(live.latencyMode).to.equal(LiveVideoLatencyMode.DEFAULT)
}
})
liveVideo = await createLiveWrapper()
- const command = sendRTMPStream(rtmpUrl + '/bad-live', liveVideo.streamKey)
+ const command = sendRTMPStream({ rtmpBaseUrl: rtmpUrl + '/bad-live', streamKey: liveVideo.streamKey })
await testFfmpegStreamError(command, true)
})
it('Should not allow a stream without the appropriate stream key', async function () {
this.timeout(60000)
- const command = sendRTMPStream(rtmpUrl + '/live', 'bad-stream-key')
+ const command = sendRTMPStream({ rtmpBaseUrl: rtmpUrl + '/live', streamKey: 'bad-stream-key' })
await testFfmpegStreamError(command, true)
})
it('Should succeed with the correct params', async function () {
this.timeout(60000)
- const command = sendRTMPStream(rtmpUrl + '/live', liveVideo.streamKey)
+ const command = sendRTMPStream({ rtmpBaseUrl: rtmpUrl + '/live', streamKey: liveVideo.streamKey })
await testFfmpegStreamError(command, false)
})
await servers[0].blacklist.add({ videoId: liveVideo.uuid })
- const command = sendRTMPStream(rtmpUrl + '/live', liveVideo.streamKey)
+ const command = sendRTMPStream({ rtmpBaseUrl: rtmpUrl + '/live', streamKey: liveVideo.streamKey })
await testFfmpegStreamError(command, true)
})
await servers[0].videos.remove({ id: liveVideo.uuid })
- const command = sendRTMPStream(rtmpUrl + '/live', liveVideo.streamKey)
+ const command = sendRTMPStream({ rtmpBaseUrl: rtmpUrl + '/live', streamKey: liveVideo.streamKey })
await testFfmpegStreamError(command, true)
})
})
transcoding: {
enabled: true,
resolutions: {
+ '144p': resolutions.includes(144),
'240p': resolutions.includes(240),
'360p': resolutions.includes(360),
'480p': resolutions.includes(480),
})
it('Should enable transcoding without additional resolutions', async function () {
- this.timeout(60000)
+ this.timeout(120000)
liveVideoId = await createLiveWrapper(false)
})
it('Should enable transcoding with some resolutions', async function () {
- this.timeout(60000)
+ this.timeout(120000)
const resolutions = [ 240, 480 ]
await updateConf(resolutions)
await stopFfmpeg(ffmpegCommand)
})
+ it('Should correctly set the appropriate bitrate depending on the input', async function () {
+ this.timeout(120000)
+
+ liveVideoId = await createLiveWrapper(false)
+
+ const ffmpegCommand = await commands[0].sendRTMPStreamInVideo({
+ videoId: liveVideoId,
+ fixtureName: 'video_short.mp4',
+ copyCodecs: true
+ })
+ await waitUntilLivePublishedOnAllServers(servers, liveVideoId)
+ await waitJobs(servers)
+
+ const video = await servers[0].videos.get({ id: liveVideoId })
+
+ const masterPlaylist = video.streamingPlaylists[0].playlistUrl
+ const probe = await ffprobePromise(masterPlaylist)
+
+ const bitrates = probe.streams.map(s => parseInt(s.tags.variant_bitrate))
+ for (const bitrate of bitrates) {
+ expect(bitrate).to.exist
+ expect(isNaN(bitrate)).to.be.false
+ expect(bitrate).to.be.below(61_000_000) // video_short.mp4 bitrate
+ }
+
+ await stopFfmpeg(ffmpegCommand)
+ })
+
it('Should enable transcoding with some resolutions and correctly save them', async function () {
- this.timeout(200000)
+ this.timeout(400_000)
const resolutions = [ 240, 360, 720 ]
await waitUntilLivePublishedOnAllServers(servers, liveVideoId)
- const bitrateLimits = {
- 720: 5000 * 1000, // 60FPS
- 360: 1100 * 1000,
- 240: 600 * 1000
+ const maxBitrateLimits = {
+ 720: 6500 * 1000, // 60FPS
+ 360: 1250 * 1000,
+ 240: 700 * 1000
+ }
+
+ const minBitrateLimits = {
+ 720: 5500 * 1000,
+ 360: 1000 * 1000,
+ 240: 550 * 1000
}
for (const server of servers) {
await makeRawRequest(hlsPlaylist.playlistUrl, HttpStatusCode.OK_200)
await makeRawRequest(hlsPlaylist.segmentsSha256Url, HttpStatusCode.OK_200)
+ // We should have generated random filenames
+ expect(basename(hlsPlaylist.playlistUrl)).to.not.equal('master.m3u8')
+ expect(basename(hlsPlaylist.segmentsSha256Url)).to.not.equal('segments-sha256.json')
+
expect(hlsPlaylist.files).to.have.lengthOf(resolutions.length)
for (const resolution of resolutions) {
expect(file.fps).to.be.approximately(30, 2)
}
- const filename = `${video.uuid}-${resolution}-fragmented.mp4`
+ const filename = basename(file.fileUrl)
+ expect(filename).to.not.contain(video.uuid)
+
const segmentPath = servers[0].servers.buildDirectory(join('streaming-playlists', 'hls', video.uuid, filename))
const probe = await ffprobePromise(segmentPath)
- const videoStream = await getVideoStreamFromFile(segmentPath, probe)
+ const videoStream = await getVideoStream(segmentPath, probe)
- expect(probe.format.bit_rate).to.be.below(bitrateLimits[videoStream.height])
+ expect(probe.format.bit_rate).to.be.below(maxBitrateLimits[videoStream.height])
+ expect(probe.format.bit_rate).to.be.at.least(minBitrateLimits[videoStream.height])
await makeRawRequest(file.torrentUrl, HttpStatusCode.OK_200)
await makeRawRequest(file.fileUrl, HttpStatusCode.OK_200)
describe('After a server restart', function () {
let liveVideoId: string
let liveVideoReplayId: string
+ let permanentLiveVideoReplayId: string
- async function createLiveWrapper (saveReplay: boolean) {
- const liveAttributes = {
+ let permanentLiveReplayName: string
+
+ let beforeServerRestart: Date
+
+ async function createLiveWrapper (options: { saveReplay: boolean, permanent: boolean }) {
+ const liveAttributes: LiveVideoCreate = {
name: 'live video',
channelId: servers[0].store.channel.id,
privacy: VideoPrivacy.PUBLIC,
- saveReplay
+ saveReplay: options.saveReplay,
+ permanentLive: options.permanent
}
const { uuid } = await commands[0].create({ fields: liveAttributes })
}
before(async function () {
- this.timeout(120000)
+ this.timeout(160000)
- liveVideoId = await createLiveWrapper(false)
- liveVideoReplayId = await createLiveWrapper(true)
+ liveVideoId = await createLiveWrapper({ saveReplay: false, permanent: false })
+ liveVideoReplayId = await createLiveWrapper({ saveReplay: true, permanent: false })
+ permanentLiveVideoReplayId = await createLiveWrapper({ saveReplay: true, permanent: true })
await Promise.all([
commands[0].sendRTMPStreamInVideo({ videoId: liveVideoId }),
+ commands[0].sendRTMPStreamInVideo({ videoId: permanentLiveVideoReplayId }),
commands[0].sendRTMPStreamInVideo({ videoId: liveVideoReplayId })
])
await Promise.all([
commands[0].waitUntilPublished({ videoId: liveVideoId }),
+ commands[0].waitUntilPublished({ videoId: permanentLiveVideoReplayId }),
commands[0].waitUntilPublished({ videoId: liveVideoReplayId })
])
await commands[0].waitUntilSegmentGeneration({ videoUUID: liveVideoId, resolution: 0, segment: 2 })
await commands[0].waitUntilSegmentGeneration({ videoUUID: liveVideoReplayId, resolution: 0, segment: 2 })
+ await commands[0].waitUntilSegmentGeneration({ videoUUID: permanentLiveVideoReplayId, resolution: 0, segment: 2 })
+
+ {
+ const video = await servers[0].videos.get({ id: permanentLiveVideoReplayId })
+ permanentLiveReplayName = video.name + ' - ' + new Date(video.publishedAt).toLocaleString()
+ }
await killallServers([ servers[0] ])
- await reRunServer(servers[0])
+
+ beforeServerRestart = new Date()
+ await servers[0].run()
await wait(5000)
+ await waitJobs(servers)
})
it('Should cleanup lives', async function () {
this.timeout(60000)
await commands[0].waitUntilEnded({ videoId: liveVideoId })
+ await commands[0].waitUntilWaiting({ videoId: permanentLiveVideoReplayId })
})
- it('Should save a live replay', async function () {
- this.timeout(120000)
+ it('Should save a non permanent live replay', async function () {
+ this.timeout(240000)
await commands[0].waitUntilPublished({ videoId: liveVideoReplayId })
+
+ const session = await commands[0].getReplaySession({ videoId: liveVideoReplayId })
+ expect(session.endDate).to.exist
+ expect(new Date(session.endDate)).to.be.above(beforeServerRestart)
+ })
+
+ it('Should have saved a permanent live replay', async function () {
+ this.timeout(120000)
+
+ const { data } = await servers[0].videos.listMyVideos({ sort: '-publishedAt' })
+ expect(data.find(v => v.name === permanentLiveReplayName)).to.exist
})
})