1
0
Fork 0

Add infohash test for VOD

This commit is contained in:
Chocobozzz 2023-03-20 08:57:01 +01:00
parent c32985769e
commit 1c41b5c1ac
No known key found for this signature in database
GPG key ID: 583A612D890159BE

View file

@ -26,8 +26,12 @@ import {
waitJobs, waitJobs,
webtorrentAdd webtorrentAdd
} from '@shared/server-commands' } from '@shared/server-commands'
import { sha1 } from '@shared/extra-utils'
async function checkFiles (options: { async function checkFiles (options: {
server: PeerTubeServer
originServer: PeerTubeServer
video: VideoDetails video: VideoDetails
baseMockUrl?: string baseMockUrl?: string
@ -39,6 +43,8 @@ async function checkFiles (options: {
webtorrentPrefix?: string webtorrentPrefix?: string
}) { }) {
const { const {
server,
originServer,
video, video,
playlistBucket, playlistBucket,
webtorrentBucket, webtorrentBucket,
@ -86,6 +92,7 @@ async function checkFiles (options: {
const resSha = await makeRawRequest({ url: hls.segmentsSha256Url, expectedStatus: HttpStatusCode.OK_200 }) const resSha = await makeRawRequest({ url: hls.segmentsSha256Url, expectedStatus: HttpStatusCode.OK_200 })
expect(JSON.stringify(resSha.body)).to.not.throw expect(JSON.stringify(resSha.body)).to.not.throw
let i = 0
for (const file of hls.files) { for (const file of hls.files) {
expectStartWith(file.fileUrl, start) expectStartWith(file.fileUrl, start)
@ -94,6 +101,15 @@ async function checkFiles (options: {
expectStartWith(location, start) expectStartWith(location, start)
await makeRawRequest({ url: location, expectedStatus: HttpStatusCode.OK_200 }) await makeRawRequest({ url: location, expectedStatus: HttpStatusCode.OK_200 })
if (originServer.internalServerNumber === server.internalServerNumber) {
const infohash = sha1(`${2 + hls.playlistUrl}+V${i}`)
const dbInfohashes = await originServer.sql.getPlaylistInfohash(hls.id)
expect(dbInfohashes).to.include(infohash)
}
i++
} }
} }
@ -198,7 +214,7 @@ function runTestSuite (options: {
for (const server of servers) { for (const server of servers) {
const video = await server.videos.get({ id: uuid }) const video = await server.videos.get({ id: uuid })
const files = await checkFiles({ ...options, video, baseMockUrl }) const files = await checkFiles({ ...options, server, originServer: servers[0], video, baseMockUrl })
deletedUrls = deletedUrls.concat(files) deletedUrls = deletedUrls.concat(files)
} }
@ -214,7 +230,7 @@ function runTestSuite (options: {
for (const server of servers) { for (const server of servers) {
const video = await server.videos.get({ id: uuid }) const video = await server.videos.get({ id: uuid })
const files = await checkFiles({ ...options, video, baseMockUrl }) const files = await checkFiles({ ...options, server, originServer: servers[0], video, baseMockUrl })
deletedUrls = deletedUrls.concat(files) deletedUrls = deletedUrls.concat(files)
} }