PeerTube/server/tests/api/videos/video-hls.ts

219 lines
6.7 KiB
TypeScript
Raw Normal View History

2019-01-29 08:37:25 +01:00
/* tslint:disable:no-unused-expression */
import * as chai from 'chai'
import 'mocha'
import {
checkDirectoryIsEmpty,
checkSegmentHash,
checkTmpIsEmpty,
cleanupTests,
2019-01-29 08:37:25 +01:00
doubleFollow,
flushAndRunMultipleServers,
getPlaylist,
getVideo, makeGetRequest, makeRawRequest,
2019-01-29 08:37:25 +01:00
removeVideo,
ServerInfo,
setAccessTokensToServers, updateCustomSubConfig,
2019-01-29 08:37:25 +01:00
updateVideo,
uploadVideo,
waitJobs, webtorrentAdd
} from '../../../../shared/extra-utils'
2019-01-29 08:37:25 +01:00
import { VideoDetails } from '../../../../shared/models/videos'
import { VideoStreamingPlaylistType } from '../../../../shared/models/videos/video-streaming-playlist.type'
import { join } from 'path'
2019-05-17 11:56:12 +02:00
import { DEFAULT_AUDIO_RESOLUTION } from '../../../initializers/constants'
2019-01-29 08:37:25 +01:00
const expect = chai.expect
async function checkHlsPlaylist (servers: ServerInfo[], videoUUID: string, hlsOnly: boolean, resolutions = [ 240, 360, 480, 720 ]) {
2019-01-29 08:37:25 +01:00
for (const server of servers) {
const resVideoDetails = await getVideo(server.url, videoUUID)
const videoDetails: VideoDetails = resVideoDetails.body
const baseUrl = `http://${videoDetails.account.host}`
2019-01-29 08:37:25 +01:00
expect(videoDetails.streamingPlaylists).to.have.lengthOf(1)
const hlsPlaylist = videoDetails.streamingPlaylists.find(p => p.type === VideoStreamingPlaylistType.HLS)
expect(hlsPlaylist).to.not.be.undefined
const hlsFiles = hlsPlaylist.files
expect(hlsFiles).to.have.lengthOf(resolutions.length)
if (hlsOnly) expect(videoDetails.files).to.have.lengthOf(0)
else expect(videoDetails.files).to.have.lengthOf(resolutions.length)
for (const resolution of resolutions) {
const file = hlsFiles.find(f => f.resolution.id === resolution)
expect(file).to.not.be.undefined
expect(file.magnetUri).to.have.lengthOf.above(2)
expect(file.torrentUrl).to.equal(`${baseUrl}/static/torrents/${videoDetails.uuid}-${file.resolution.id}-hls.torrent`)
expect(file.fileUrl).to.equal(`${baseUrl}/static/streaming-playlists/hls/${videoDetails.uuid}/${videoDetails.uuid}-${file.resolution.id}-fragmented.mp4`)
expect(file.resolution.label).to.equal(resolution + 'p')
await makeRawRequest(file.torrentUrl, 200)
await makeRawRequest(file.fileUrl, 200)
const torrent = await webtorrentAdd(file.magnetUri, true)
expect(torrent.files).to.be.an('array')
expect(torrent.files.length).to.equal(1)
expect(torrent.files[0].path).to.exist.and.to.not.equal('')
}
2019-01-29 08:37:25 +01:00
{
const res = await getPlaylist(hlsPlaylist.playlistUrl)
2019-01-29 08:37:25 +01:00
const masterPlaylist = res.text
2019-01-29 08:37:25 +01:00
for (const resolution of resolutions) {
2019-11-26 16:25:36 +01:00
const reg = new RegExp('#EXT-X-STREAM-INF:BANDWIDTH=\\d+,RESOLUTION=\\d+x' + resolution + ',FRAME-RATE=\\d+,CODECS="avc1.64001f,mp4a.40.2"')
expect(masterPlaylist).to.match(reg)
expect(masterPlaylist).to.contain(`${resolution}.m3u8`)
2019-01-29 08:37:25 +01:00
expect(masterPlaylist).to.contain(`${resolution}.m3u8`)
}
}
{
for (const resolution of resolutions) {
const res = await getPlaylist(`${baseUrl}/static/streaming-playlists/hls/${videoUUID}/${resolution}.m3u8`)
2019-01-29 08:37:25 +01:00
const subPlaylist = res.text
expect(subPlaylist).to.contain(`${videoUUID}-${resolution}-fragmented.mp4`)
2019-01-29 08:37:25 +01:00
}
}
{
const baseUrlAndPath = baseUrl + '/static/streaming-playlists/hls'
2019-01-29 08:37:25 +01:00
for (const resolution of resolutions) {
await checkSegmentHash(baseUrlAndPath, baseUrlAndPath, videoUUID, resolution, hlsPlaylist)
2019-01-29 08:37:25 +01:00
}
}
}
}
describe('Test HLS videos', function () {
let servers: ServerInfo[] = []
let videoUUID = ''
2019-05-17 11:56:12 +02:00
let videoAudioUUID = ''
2019-01-29 08:37:25 +01:00
function runTestSuite (hlsOnly: boolean) {
it('Should upload a video and transcode it to HLS', async function () {
this.timeout(120000)
2019-01-29 08:37:25 +01:00
const res = await uploadVideo(servers[ 0 ].url, servers[ 0 ].accessToken, { name: 'video 1', fixture: 'video_short.webm' })
videoUUID = res.body.video.uuid
2019-01-29 08:37:25 +01:00
await waitJobs(servers)
2019-01-29 08:37:25 +01:00
await checkHlsPlaylist(servers, videoUUID, hlsOnly)
})
2019-01-29 08:37:25 +01:00
it('Should upload an audio file and transcode it to HLS', async function () {
this.timeout(120000)
2019-01-29 08:37:25 +01:00
const res = await uploadVideo(servers[ 0 ].url, servers[ 0 ].accessToken, { name: 'video audio', fixture: 'sample.ogg' })
videoAudioUUID = res.body.video.uuid
2019-01-29 08:37:25 +01:00
await waitJobs(servers)
2019-01-29 08:37:25 +01:00
await checkHlsPlaylist(servers, videoAudioUUID, hlsOnly, [ DEFAULT_AUDIO_RESOLUTION ])
})
2019-01-29 08:37:25 +01:00
it('Should update the video', async function () {
this.timeout(10000)
2019-05-17 11:56:12 +02:00
await updateVideo(servers[ 0 ].url, servers[ 0 ].accessToken, videoUUID, { name: 'video 1 updated' })
2019-05-17 11:56:12 +02:00
await waitJobs(servers)
2019-05-17 11:56:12 +02:00
await checkHlsPlaylist(servers, videoUUID, hlsOnly)
})
2019-05-17 11:56:12 +02:00
it('Should delete videos', async function () {
this.timeout(10000)
2019-05-17 15:51:42 +02:00
await removeVideo(servers[ 0 ].url, servers[ 0 ].accessToken, videoUUID)
await removeVideo(servers[ 0 ].url, servers[ 0 ].accessToken, videoAudioUUID)
2019-01-29 08:37:25 +01:00
await waitJobs(servers)
2019-01-29 08:37:25 +01:00
for (const server of servers) {
await getVideo(server.url, videoUUID, 404)
await getVideo(server.url, videoAudioUUID, 404)
}
})
2019-01-29 08:37:25 +01:00
it('Should have the playlists/segment deleted from the disk', async function () {
for (const server of servers) {
await checkDirectoryIsEmpty(server, 'videos')
await checkDirectoryIsEmpty(server, join('streaming-playlists', 'hls'))
}
})
2019-05-17 15:51:42 +02:00
it('Should have an empty tmp directory', async function () {
for (const server of servers) {
await checkTmpIsEmpty(server)
}
})
}
2019-01-29 08:37:25 +01:00
before(async function () {
this.timeout(120000)
2019-01-29 08:37:25 +01:00
const configOverride = {
transcoding: {
enabled: true,
allow_audio_files: true,
hls: {
enabled: true
}
}
2019-01-29 08:37:25 +01:00
}
servers = await flushAndRunMultipleServers(2, configOverride)
// Get the access tokens
await setAccessTokensToServers(servers)
// Server 1 and server 2 follow each other
await doubleFollow(servers[0], servers[1])
2019-01-29 08:37:25 +01:00
})
describe('With WebTorrent & HLS enabled', function () {
runTestSuite(false)
2019-01-29 08:37:25 +01:00
})
describe('With only HLS enabled', function () {
before(async function () {
await updateCustomSubConfig(servers[0].url, servers[0].accessToken, {
transcoding: {
enabled: true,
allowAudioFiles: true,
resolutions: {
'240p': true,
'360p': true,
'480p': true,
'720p': true,
'1080p': true,
'2160p': true
},
hls: {
enabled: true
},
webtorrent: {
enabled: false
}
}
})
})
runTestSuite(true)
2019-01-29 08:37:25 +01:00
})
2019-04-24 15:10:37 +02:00
after(async function () {
await cleanupTests(servers)
2019-01-29 08:37:25 +01:00
})
})