Add ability to delete a specific video file

1 files changed, 142 insertions, 30 deletions

diff --git a/server/tests/api/videos/video-files.ts b/server/tests/api/videos/video-files.ts
index b0ef4a2e9..313f020e9 100644
--- a/server/tests/api/videos/video-files.ts
+++ b/server/tests/api/videos/video-files.ts
@@ -2,10 +2,12 @@
 
 import 'mocha'
 import { expect } from 'chai'
+import { HttpStatusCode } from '@shared/models'
 import {
   cleanupTests,
   createMultipleServers,
   doubleFollow,
+  makeRawRequest,
   PeerTubeServer,
   setAccessTokensToServers,
   waitJobs
@@ -13,8 +15,6 @@ import {
 
 describe('Test videos files', function () {
   let servers: PeerTubeServer[]
-  let validId1: string
-  let validId2: string
 
   // ---------------------------------------------------------------
 
@@ -27,48 +27,160 @@ describe('Test videos files', function () {
     await doubleFollow(servers[0], servers[1])
 
     await servers[0].config.enableTranscoding(true, true)
+  })
 
-    {
-      const { uuid } = await servers[0].videos.quickUpload({ name: 'video 1' })
-      validId1 = uuid
-    }
+  describe('When deleting all files', function () {
+    let validId1: string
+    let validId2: string
 
-    {
-      const { uuid } = await servers[0].videos.quickUpload({ name: 'video 2' })
-      validId2 = uuid
-    }
+    before(async function () {
+      {
+        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 webtorrent files', async function () {
+      this.timeout(30_000)
+
+      await servers[0].videos.removeAllWebTorrentFiles({ 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 webtorrent files', async function () {
-    this.timeout(30_000)
+    it('Should delete HLS files', async function () {
+      this.timeout(30_000)
 
-    await servers[0].videos.removeWebTorrentFiles({ videoId: validId1 })
+      await servers[0].videos.removeHLSPlaylist({ videoId: validId2 })
 
-    await waitJobs(servers)
+      await waitJobs(servers)
 
-    for (const server of servers) {
-      const video = await server.videos.get({ id: validId1 })
+      for (const server of servers) {
+        const video = await server.videos.get({ id: validId2 })
 
-      expect(video.files).to.have.lengthOf(0)
-      expect(video.streamingPlaylists).to.have.lengthOf(1)
-    }
+        expect(video.files).to.have.length.above(0)
+        expect(video.streamingPlaylists).to.have.lengthOf(0)
+      }
+    })
   })
 
-  it('Should delete HLS files', async function () {
-    this.timeout(30_000)
+  describe('When deleting a specific file', function () {
+    let webtorrentId: string
+    let hlsId: string
+
+    before(async function () {
+      {
+        const { uuid } = await servers[0].videos.quickUpload({ name: 'webtorrent' })
+        webtorrentId = uuid
+      }
+
+      {
+        const { uuid } = await servers[0].videos.quickUpload({ name: 'hls' })
+        hlsId = uuid
+      }
+
+      await waitJobs(servers)
+    })
+
+    it('Shoulde delete a webtorrent file', async function () {
+      const video = await servers[0].videos.get({ id: webtorrentId })
+      const files = video.files
+
+      await servers[0].videos.removeWebTorrentFile({ videoId: webtorrentId, fileId: files[0].id })
+
+      await waitJobs(servers)
+
+      for (const server of servers) {
+        const video = await server.videos.get({ id: webtorrentId })
+
+        expect(video.files).to.have.lengthOf(files.length - 1)
+        expect(video.files.find(f => f.id === files[0].id)).to.not.exist
+      }
+    })
+
+    it('Should delete all webtorrent files', async function () {
+      const video = await servers[0].videos.get({ id: webtorrentId })
+      const files = video.files
+
+      for (const file of files) {
+        await servers[0].videos.removeWebTorrentFile({ videoId: webtorrentId, fileId: file.id })
+      }
+
+      await waitJobs(servers)
+
+      for (const server of servers) {
+        const video = await server.videos.get({ id: webtorrentId })
+
+        expect(video.files).to.have.lengthOf(0)
+      }
+    })
+
+    it('Should delete a hls file', async function () {
+      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.id === toDelete.id)).to.not.exist
+
+        const { text } = await makeRawRequest(video.streamingPlaylists[0].playlistUrl)
+
+        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 () {
+      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 })
 
-    await servers[0].videos.removeHLSFiles({ videoId: validId2 })
+        expect(video.streamingPlaylists).to.have.lengthOf(0)
+      }
+    })
 
-    await waitJobs(servers)
+    it('Should not delete last file of a video', async function () {
+      const webtorrentOnly = await servers[0].videos.get({ id: hlsId })
+      const hlsOnly = await servers[0].videos.get({ id: webtorrentId })
 
-    for (const server of servers) {
-      const video = await server.videos.get({ id: validId2 })
+      for (let i = 0; i < 4; i++) {
+        await servers[0].videos.removeWebTorrentFile({ videoId: webtorrentOnly.id, fileId: webtorrentOnly.files[i].id })
+        await servers[0].videos.removeHLSFile({ videoId: hlsOnly.id, fileId: hlsOnly.streamingPlaylists[0].files[i].id })
+      }
 
-      expect(video.files).to.have.length.above(0)
-      expect(video.streamingPlaylists).to.have.lengthOf(0)
-    }
+      const expectedStatus = HttpStatusCode.BAD_REQUEST_400
+      await servers[0].videos.removeWebTorrentFile({ videoId: webtorrentOnly.id, fileId: webtorrentOnly.files[4].id, expectedStatus })
+      await servers[0].videos.removeHLSFile({ videoId: hlsOnly.id, fileId: hlsOnly.streamingPlaylists[0].files[4].id, expectedStatus })
+    })
   })
 
   after(async function () {