mirror of https://github.com/Chocobozzz/PeerTube
98 lines
2.7 KiB
TypeScript
98 lines
2.7 KiB
TypeScript
import { exec } from 'child_process'
|
|
import { copy, ensureDir, readFile, remove } from 'fs-extra'
|
|
import { basename, join } from 'path'
|
|
import { isGithubCI, root, wait } from '@shared/core-utils'
|
|
import { getFileSize } from '@shared/extra-utils'
|
|
import { HttpStatusCode } from '@shared/models'
|
|
import { AbstractCommand, OverrideCommandOptions } from '../shared'
|
|
|
|
export class ServersCommand extends AbstractCommand {
|
|
|
|
static flushTests (internalServerNumber: number) {
|
|
return new Promise<void>((res, rej) => {
|
|
const suffix = ` -- ${internalServerNumber}`
|
|
|
|
return exec('npm run clean:server:test' + suffix, (err, _stdout, stderr) => {
|
|
if (err || stderr) return rej(err || new Error(stderr))
|
|
|
|
return res()
|
|
})
|
|
})
|
|
}
|
|
|
|
ping (options: OverrideCommandOptions = {}) {
|
|
return this.getRequestBody({
|
|
...options,
|
|
|
|
path: '/api/v1/ping',
|
|
implicitToken: false,
|
|
defaultExpectedStatus: HttpStatusCode.OK_200
|
|
})
|
|
}
|
|
|
|
cleanupTests () {
|
|
const promises: Promise<any>[] = []
|
|
|
|
const saveGithubLogsIfNeeded = async () => {
|
|
if (!isGithubCI()) return
|
|
|
|
await ensureDir('artifacts')
|
|
|
|
const origin = this.buildDirectory('logs/peertube.log')
|
|
const destname = `peertube-${this.server.internalServerNumber}.log`
|
|
console.log('Saving logs %s.', destname)
|
|
|
|
await copy(origin, join('artifacts', destname))
|
|
}
|
|
|
|
if (this.server.parallel) {
|
|
const promise = saveGithubLogsIfNeeded()
|
|
.then(() => ServersCommand.flushTests(this.server.internalServerNumber))
|
|
|
|
promises.push(promise)
|
|
}
|
|
|
|
if (this.server.customConfigFile) {
|
|
promises.push(remove(this.server.customConfigFile))
|
|
}
|
|
|
|
return promises
|
|
}
|
|
|
|
async waitUntilLog (str: string, count = 1, strictCount = true) {
|
|
const logfile = this.buildDirectory('logs/peertube.log')
|
|
|
|
while (true) {
|
|
const buf = await readFile(logfile)
|
|
|
|
const matches = buf.toString().match(new RegExp(str, 'g'))
|
|
if (matches && matches.length === count) return
|
|
if (matches && strictCount === false && matches.length >= count) return
|
|
|
|
await wait(1000)
|
|
}
|
|
}
|
|
|
|
buildDirectory (directory: string) {
|
|
return join(root(), 'test' + this.server.internalServerNumber, directory)
|
|
}
|
|
|
|
buildWebVideoFilePath (fileUrl: string) {
|
|
return this.buildDirectory(join('web-videos', basename(fileUrl)))
|
|
}
|
|
|
|
buildFragmentedFilePath (videoUUID: string, fileUrl: string) {
|
|
return this.buildDirectory(join('streaming-playlists', 'hls', videoUUID, basename(fileUrl)))
|
|
}
|
|
|
|
getLogContent () {
|
|
return readFile(this.buildDirectory('logs/peertube.log'))
|
|
}
|
|
|
|
async getServerFileSize (subPath: string) {
|
|
const path = this.server.servers.buildDirectory(subPath)
|
|
|
|
return getFileSize(path)
|
|
}
|
|
}
|