server/tests/cli/regenerate-thumbnails.ts

   1 import 'mocha'
   2 import { expect } from 'chai'
   3 import { writeFile } from 'fs-extra'
   4 import { basename, join } from 'path'
   5 import { Video, VideoDetails } from '@shared/models'
   6 import {
   7   cleanupTests,
   8   doubleFollow,
   9   flushAndRunMultipleServers,
  10   getVideo,
  11   makeRawRequest,
  12   ServerInfo,
  13   setAccessTokensToServers,
  14   uploadVideoAndGetId,
  15   waitJobs
  16 } from '../../../shared/extra-utils'
  17 import { HttpStatusCode } from '@shared/core-utils'
  18
  19 async function testThumbnail (server: ServerInfo, videoId: number | string) {
  20   const res = await getVideo(server.url, videoId)
  21   const video: VideoDetails = res.body
  22
  23   const res1 = await makeRawRequest(join(server.url, video.thumbnailPath), HttpStatusCode.OK_200)
  24   expect(res1.body).to.not.have.lengthOf(0)
  25
  26   const res2 = await makeRawRequest(join(server.url, video.thumbnailPath), HttpStatusCode.OK_200)
  27   expect(res2.body).to.not.have.lengthOf(0)
  28 }
  29
  30 describe('Test regenerate thumbnails script', function () {
  31   let servers: ServerInfo[]
  32
  33   let video1: Video
  34   let video2: Video
  35   let remoteVideo: Video
  36
  37   let thumbnail1Path: string
  38   let thumbnailRemotePath: string
  39
  40   before(async function () {
  41     this.timeout(60000)
  42
  43     servers = await flushAndRunMultipleServers(2)
  44     await setAccessTokensToServers(servers)
  45
  46     await doubleFollow(servers[0], servers[1])
  47
  48     {
  49       const videoUUID1 = (await uploadVideoAndGetId({ server: servers[0], videoName: 'video 1' })).uuid
  50       video1 = await (getVideo(servers[0].url, videoUUID1).then(res => res.body))
  51
  52       thumbnail1Path = join(servers[0].serversCommand.buildDirectory('thumbnails'), basename(video1.thumbnailPath))
  53
  54       const videoUUID2 = (await uploadVideoAndGetId({ server: servers[0], videoName: 'video 2' })).uuid
  55       video2 = await (getVideo(servers[0].url, videoUUID2).then(res => res.body))
  56     }
  57
  58     {
  59       const videoUUID = (await uploadVideoAndGetId({ server: servers[1], videoName: 'video 3' })).uuid
  60       await waitJobs(servers)
  61
  62       remoteVideo = await (getVideo(servers[0].url, videoUUID).then(res => res.body))
  63
  64       thumbnailRemotePath = join(servers[0].serversCommand.buildDirectory('thumbnails'), basename(remoteVideo.thumbnailPath))
  65     }
  66
  67     await writeFile(thumbnail1Path, '')
  68     await writeFile(thumbnailRemotePath, '')
  69   })
  70
  71   it('Should have empty thumbnails', async function () {
  72     {
  73       const res = await makeRawRequest(join(servers[0].url, video1.thumbnailPath), HttpStatusCode.OK_200)
  74       expect(res.body).to.have.lengthOf(0)
  75     }
  76
  77     {
  78       const res = await makeRawRequest(join(servers[0].url, video2.thumbnailPath), HttpStatusCode.OK_200)
  79       expect(res.body).to.not.have.lengthOf(0)
  80     }
  81
  82     {
  83       const res = await makeRawRequest(join(servers[0].url, remoteVideo.thumbnailPath), HttpStatusCode.OK_200)
  84       expect(res.body).to.have.lengthOf(0)
  85     }
  86   })
  87
  88   it('Should regenerate local thumbnails from the CLI', async function () {
  89     this.timeout(15000)
  90
  91     await servers[0].cliCommand.execWithEnv(`npm run regenerate-thumbnails`)
  92   })
  93
  94   it('Should have generated new thumbnail files', async function () {
  95     await testThumbnail(servers[0], video1.uuid)
  96     await testThumbnail(servers[0], video2.uuid)
  97
  98     const res = await makeRawRequest(join(servers[0].url, remoteVideo.thumbnailPath), HttpStatusCode.OK_200)
  99     expect(res.body).to.have.lengthOf(0)
 100   })
 101
 102   it('Should have deleted old thumbnail files', async function () {
 103     {
 104       await makeRawRequest(join(servers[0].url, video1.thumbnailPath), HttpStatusCode.NOT_FOUND_404)
 105     }
 106
 107     {
 108       await makeRawRequest(join(servers[0].url, video2.thumbnailPath), HttpStatusCode.NOT_FOUND_404)
 109     }
 110
 111     {
 112       const res = await makeRawRequest(join(servers[0].url, remoteVideo.thumbnailPath), HttpStatusCode.OK_200)
 113       expect(res.body).to.have.lengthOf(0)
 114     }
 115   })
 116
 117   after(async function () {
 118     await cleanupTests(servers)
 119   })
 120 })