Fix broken object storage playlist on file removal

This commit is contained in:
Chocobozzz
2024-08-19 16:00:40 +02:00
parent bd60f178af
commit b2bb45cf91
4 changed files with 193 additions and 165 deletions

View File

@@ -1,16 +1,18 @@
/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
import { expect } from 'chai'
import { HttpStatusCode } from '@peertube/peertube-models'
import { areMockObjectStorageTestsDisabled } from '@peertube/peertube-node-utils'
import {
cleanupTests,
createMultipleServers,
doubleFollow,
makeRawRequest,
ObjectStorageCommand,
PeerTubeServer,
setAccessTokensToServers,
waitJobs
} from '@peertube/peertube-server-commands'
import { expect } from 'chai'
describe('Test videos files', function () {
let servers: PeerTubeServer[]
@@ -28,173 +30,199 @@ describe('Test videos files', function () {
await servers[0].config.enableTranscoding({ hls: true, webVideo: true, resolutions: 'max' })
})
describe('When deleting all files', function () {
let validId1: string
let validId2: string
function runTests (objectStorage?: ObjectStorageCommand) {
before(async function () {
this.timeout(360_000)
describe('When deleting all files', function () {
let validId1: string
let validId2: string
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'video 1' })
validId1 = uuid
}
before(async function () {
this.timeout(360_000)
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'video 2' })
validId2 = uuid
}
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'video 1' })
validId1 = uuid
}
await waitJobs(servers)
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'video 2' })
validId2 = uuid
}
await waitJobs(servers)
})
it('Should delete web video files', async function () {
this.timeout(30_000)
await servers[0].videos.removeAllWebVideoFiles({ videoId: validId1 })
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: validId1 })
expect(video.files).to.have.lengthOf(0)
expect(video.streamingPlaylists).to.have.lengthOf(1)
}
})
it('Should delete HLS files', async function () {
this.timeout(30_000)
await servers[0].videos.removeHLSPlaylist({ videoId: validId2 })
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: validId2 })
expect(video.files).to.have.length.above(0)
expect(video.streamingPlaylists).to.have.lengthOf(0)
}
})
})
it('Should delete web video files', async function () {
this.timeout(30_000)
describe('When deleting a specific file', function () {
let webVideoId: string
let hlsId: string
await servers[0].videos.removeAllWebVideoFiles({ videoId: validId1 })
before(async function () {
this.timeout(120_000)
await waitJobs(servers)
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'web-video' })
webVideoId = uuid
}
for (const server of servers) {
const video = await server.videos.get({ id: validId1 })
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'hls' })
hlsId = uuid
}
expect(video.files).to.have.lengthOf(0)
expect(video.streamingPlaylists).to.have.lengthOf(1)
}
await waitJobs(servers)
})
it('Shoulde delete a web video file', async function () {
this.timeout(30_000)
const video = await servers[0].videos.get({ id: webVideoId })
const files = video.files
const toDelete = files[0]
await servers[0].videos.removeWebVideoFile({ videoId: webVideoId, fileId: toDelete.id })
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: webVideoId })
expect(video.files).to.have.lengthOf(files.length - 1)
expect(video.files.find(f => f.resolution.id === toDelete.resolution.id)).to.not.exist
}
})
it('Should delete all web video files', async function () {
this.timeout(30_000)
const video = await servers[0].videos.get({ id: webVideoId })
const files = video.files
for (const file of files) {
await servers[0].videos.removeWebVideoFile({ videoId: webVideoId, fileId: file.id })
}
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: webVideoId })
expect(video.files).to.have.lengthOf(0)
}
})
it('Should delete a hls file', async function () {
this.timeout(30_000)
const video = await servers[0].videos.get({ id: hlsId })
const files = video.streamingPlaylists[0].files
const toDelete = files[0]
await servers[0].videos.removeHLSFile({ videoId: hlsId, fileId: toDelete.id })
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: hlsId })
expect(video.streamingPlaylists[0].files).to.have.lengthOf(files.length - 1)
expect(video.streamingPlaylists[0].files.find(f => f.resolution.id === toDelete.resolution.id)).to.not.exist
const m3u8Content = await servers[0].streamingPlaylists.get({ url: video.streamingPlaylists[0].playlistUrl })
await makeRawRequest({ url: video.streamingPlaylists[0].segmentsSha256Url, expectedStatus: HttpStatusCode.OK_200 })
expect(m3u8Content.includes(`-${toDelete.resolution.id}.m3u8`)).to.be.false
expect(m3u8Content.includes(`-${video.streamingPlaylists[0].files[0].resolution.id}.m3u8`)).to.be.true
}
})
it('Should delete all hls files', async function () {
this.timeout(30_000)
const video = await servers[0].videos.get({ id: hlsId })
const files = video.streamingPlaylists[0].files
for (const file of files) {
await servers[0].videos.removeHLSFile({ videoId: hlsId, fileId: file.id })
}
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: hlsId })
expect(video.streamingPlaylists).to.have.lengthOf(0)
}
})
it('Should not delete last file of a video', async function () {
this.timeout(60_000)
const webVideoOnly = await servers[0].videos.get({ id: hlsId })
const hlsOnly = await servers[0].videos.get({ id: webVideoId })
for (let i = 0; i < 4; i++) {
await servers[0].videos.removeWebVideoFile({ videoId: webVideoOnly.id, fileId: webVideoOnly.files[i].id })
await servers[0].videos.removeHLSFile({ videoId: hlsOnly.id, fileId: hlsOnly.streamingPlaylists[0].files[i].id })
}
const expectedStatus = HttpStatusCode.BAD_REQUEST_400
await servers[0].videos.removeWebVideoFile({ videoId: webVideoOnly.id, fileId: webVideoOnly.files[4].id, expectedStatus })
await servers[0].videos.removeHLSFile({ videoId: hlsOnly.id, fileId: hlsOnly.streamingPlaylists[0].files[4].id, expectedStatus })
})
})
}
it('Should delete HLS files', async function () {
this.timeout(30_000)
await servers[0].videos.removeHLSPlaylist({ videoId: validId2 })
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: validId2 })
expect(video.files).to.have.length.above(0)
expect(video.streamingPlaylists).to.have.lengthOf(0)
}
})
describe('Using filesystem', function () {
runTests()
})
describe('When deleting a specific file', function () {
let webVideoId: string
let hlsId: string
describe('Using object storage', function () {
if (areMockObjectStorageTestsDisabled()) return
before(async function () {
this.timeout(120_000)
this.timeout(120000)
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'web-video' })
webVideoId = uuid
}
const configOverride = objectStorage.getDefaultMockConfig()
await objectStorage.prepareDefaultMockBuckets()
{
const { uuid } = await servers[0].videos.quickUpload({ name: 'hls' })
hlsId = uuid
}
await waitJobs(servers)
await servers[0].kill()
await servers[0].run(configOverride)
})
it('Shoulde delete a web video file', async function () {
this.timeout(30_000)
const objectStorage = new ObjectStorageCommand()
const video = await servers[0].videos.get({ id: webVideoId })
const files = video.files
const toDelete = files[0]
await servers[0].videos.removeWebVideoFile({ videoId: webVideoId, fileId: toDelete.id })
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: webVideoId })
expect(video.files).to.have.lengthOf(files.length - 1)
expect(video.files.find(f => f.resolution.id === toDelete.resolution.id)).to.not.exist
}
})
it('Should delete all web video files', async function () {
this.timeout(30_000)
const video = await servers[0].videos.get({ id: webVideoId })
const files = video.files
for (const file of files) {
await servers[0].videos.removeWebVideoFile({ videoId: webVideoId, fileId: file.id })
}
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: webVideoId })
expect(video.files).to.have.lengthOf(0)
}
})
it('Should delete a hls file', async function () {
this.timeout(30_000)
const video = await servers[0].videos.get({ id: hlsId })
const files = video.streamingPlaylists[0].files
const toDelete = files[0]
await servers[0].videos.removeHLSFile({ videoId: hlsId, fileId: toDelete.id })
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: hlsId })
expect(video.streamingPlaylists[0].files).to.have.lengthOf(files.length - 1)
expect(video.streamingPlaylists[0].files.find(f => f.resolution.id === toDelete.resolution.id)).to.not.exist
const { text } = await makeRawRequest({ url: video.streamingPlaylists[0].playlistUrl, expectedStatus: HttpStatusCode.OK_200 })
expect(text.includes(`-${toDelete.resolution.id}.m3u8`)).to.be.false
expect(text.includes(`-${video.streamingPlaylists[0].files[0].resolution.id}.m3u8`)).to.be.true
}
})
it('Should delete all hls files', async function () {
this.timeout(30_000)
const video = await servers[0].videos.get({ id: hlsId })
const files = video.streamingPlaylists[0].files
for (const file of files) {
await servers[0].videos.removeHLSFile({ videoId: hlsId, fileId: file.id })
}
await waitJobs(servers)
for (const server of servers) {
const video = await server.videos.get({ id: hlsId })
expect(video.streamingPlaylists).to.have.lengthOf(0)
}
})
it('Should not delete last file of a video', async function () {
this.timeout(60_000)
const webVideoOnly = await servers[0].videos.get({ id: hlsId })
const hlsOnly = await servers[0].videos.get({ id: webVideoId })
for (let i = 0; i < 4; i++) {
await servers[0].videos.removeWebVideoFile({ videoId: webVideoOnly.id, fileId: webVideoOnly.files[i].id })
await servers[0].videos.removeHLSFile({ videoId: hlsOnly.id, fileId: hlsOnly.streamingPlaylists[0].files[i].id })
}
const expectedStatus = HttpStatusCode.BAD_REQUEST_400
await servers[0].videos.removeWebVideoFile({ videoId: webVideoOnly.id, fileId: webVideoOnly.files[4].id, expectedStatus })
await servers[0].videos.removeHLSFile({ videoId: hlsOnly.id, fileId: hlsOnly.streamingPlaylists[0].files[4].id, expectedStatus })
})
runTests(objectStorage)
})
after(async function () {