aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/tests/cli/create-move-video-storage-job.ts
blob: 8b66f24bfe3a8e6218fe048db2d0405974eb2798 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */

import 'mocha'

import {
  areObjectStorageTestsDisabled,
  cleanupTests,
  createMultipleServers,
  doubleFollow,
  expectStartWith,
  makeRawRequest,
  ObjectStorageCommand,
  PeerTubeServer,
  setAccessTokensToServers,
  waitJobs
} from '@shared/server-commands'
import { HttpStatusCode, VideoDetails } from '@shared/models'

async function checkFiles (origin: PeerTubeServer, video: VideoDetails, inObjectStorage: boolean) {
  for (const file of video.files) {
    const start = inObjectStorage
      ? ObjectStorageCommand.getWebTorrentBaseUrl()
      : origin.url

    expectStartWith(file.fileUrl, start)

    await makeRawRequest(file.fileUrl, HttpStatusCode.OK_200)
  }

  const start = inObjectStorage
    ? ObjectStorageCommand.getPlaylistBaseUrl()
    : origin.url

  const hls = video.streamingPlaylists[0]
  expectStartWith(hls.playlistUrl, start)
  expectStartWith(hls.segmentsSha256Url, start)

  for (const file of hls.files) {
    expectStartWith(file.fileUrl, start)

    await makeRawRequest(file.fileUrl, HttpStatusCode.OK_200)
  }
}

describe('Test create move video storage job', function () {
  if (areObjectStorageTestsDisabled()) return

  let servers: PeerTubeServer[] = []
  const uuids: string[] = []

  before(async function () {
    this.timeout(360000)

    // Run server 2 to have transcoding enabled
    servers = await createMultipleServers(2)
    await setAccessTokensToServers(servers)

    await doubleFollow(servers[0], servers[1])

    await ObjectStorageCommand.prepareDefaultBuckets()

    await servers[0].config.enableTranscoding()

    for (let i = 0; i < 3; i++) {
      const { uuid } = await servers[0].videos.upload({ attributes: { name: 'video' + i } })
      uuids.push(uuid)
    }

    await waitJobs(servers)

    await servers[0].kill()
    await servers[0].run(ObjectStorageCommand.getDefaultConfig())
  })

  it('Should move only one file', async function () {
    this.timeout(120000)

    const command = `npm run create-move-video-storage-job -- --to-object-storage -v ${uuids[1]}`
    await servers[0].cli.execWithEnv(command, ObjectStorageCommand.getDefaultConfig())
    await waitJobs(servers)

    for (const server of servers) {
      const video = await server.videos.get({ id: uuids[1] })

      await checkFiles(servers[0], video, true)

      for (const id of [ uuids[0], uuids[2] ]) {
        const video = await server.videos.get({ id })

        await checkFiles(servers[0], video, false)
      }
    }
  })

  it('Should move all files', async function () {
    this.timeout(120000)

    const command = `npm run create-move-video-storage-job -- --to-object-storage --all-videos`
    await servers[0].cli.execWithEnv(command, ObjectStorageCommand.getDefaultConfig())
    await waitJobs(servers)

    for (const server of servers) {
      for (const id of [ uuids[0], uuids[2] ]) {
        const video = await server.videos.get({ id })

        await checkFiles(servers[0], video, true)
      }
    }
  })

  after(async function () {
    await cleanupTests(servers)
  })
})