Random listen for mocked servers

pull/4483/head
Chocobozzz 2021-10-22 14:31:38 +02:00
parent 5480933b7f
commit d1bfbdeb20
No known key found for this signature in database
GPG Key ID: 583A612D890159BE
7 changed files with 95 additions and 87 deletions

View File

@ -182,7 +182,7 @@ function runTestSuite (options: {
})
it('Should upload a video and move it to the object storage without transcoding', async function () {
this.timeout(20000)
this.timeout(40000)
const { uuid } = await servers[0].videos.quickUpload({ name: 'video 1' })
uuidsToDelete.push(uuid)

View File

@ -1,41 +1,39 @@
import express from 'express'
import { Server } from 'http'
import { randomInt } from '@shared/core-utils'
import { terminateServer } from './utils'
import { getPort, randomListen, terminateServer } from './utils'
export class MockInstancesIndex {
private server: Server
private readonly indexInstances: { host: string, createdAt: string }[] = []
initialize () {
return new Promise<number>(res => {
const app = express()
async initialize () {
const app = express()
app.use('/', (req: express.Request, res: express.Response, next: express.NextFunction) => {
if (process.env.DEBUG) console.log('Receiving request on mocked server %s.', req.url)
app.use('/', (req: express.Request, res: express.Response, next: express.NextFunction) => {
if (process.env.DEBUG) console.log('Receiving request on mocked server %s.', req.url)
return next()
})
app.get('/api/v1/instances/hosts', (req: express.Request, res: express.Response) => {
const since = req.query.since
const filtered = this.indexInstances.filter(i => {
if (!since) return true
return i.createdAt > since
})
return res.json({
total: filtered.length,
data: filtered
})
})
const port = 42000 + randomInt(1, 1000)
this.server = app.listen(port, () => res(port))
return next()
})
app.get('/api/v1/instances/hosts', (req: express.Request, res: express.Response) => {
const since = req.query.since
const filtered = this.indexInstances.filter(i => {
if (!since) return true
return i.createdAt > since
})
return res.json({
total: filtered.length,
data: filtered
})
})
this.server = await randomListen(app)
return getPort(this.server)
}
addInstance (host: string) {

View File

@ -1,30 +1,31 @@
import express from 'express'
import { randomInt } from '@shared/core-utils'
import { Server } from 'http'
import { getPort, randomListen } from './utils'
export class MockJoinPeerTubeVersions {
private server: Server
private latestVersion: string
initialize () {
return new Promise<number>(res => {
const app = express()
async initialize () {
const app = express()
app.use('/', (req: express.Request, res: express.Response, next: express.NextFunction) => {
if (process.env.DEBUG) console.log('Receiving request on mocked server %s.', req.url)
app.use('/', (req: express.Request, res: express.Response, next: express.NextFunction) => {
if (process.env.DEBUG) console.log('Receiving request on mocked server %s.', req.url)
return next()
})
app.get('/versions.json', (req: express.Request, res: express.Response) => {
return res.json({
peertube: {
latestVersion: this.latestVersion
}
})
})
const port = 43000 + randomInt(1, 1000)
app.listen(port, () => res(port))
return next()
})
app.get('/versions.json', (req: express.Request, res: express.Response) => {
return res.json({
peertube: {
latestVersion: this.latestVersion
}
})
})
this.server = await randomListen(app)
return getPort(this.server)
}
setLatestVersion (latestVersion: string) {

View File

@ -2,39 +2,37 @@ import express from 'express'
import got, { RequestError } from 'got'
import { Server } from 'http'
import { pipeline } from 'stream'
import { randomInt } from '@shared/core-utils'
import { ObjectStorageCommand } from '../server'
import { terminateServer } from './utils'
import { getPort, randomListen, terminateServer } from './utils'
export class MockObjectStorage {
private server: Server
initialize () {
return new Promise<number>(res => {
const app = express()
async initialize () {
const app = express()
app.get('/:bucketName/:path(*)', (req: express.Request, res: express.Response, next: express.NextFunction) => {
const url = `http://${req.params.bucketName}.${ObjectStorageCommand.getEndpointHost()}/${req.params.path}`
app.get('/:bucketName/:path(*)', (req: express.Request, res: express.Response, next: express.NextFunction) => {
const url = `http://${req.params.bucketName}.${ObjectStorageCommand.getEndpointHost()}/${req.params.path}`
if (process.env.DEBUG) {
console.log('Receiving request on mocked server %s.', req.url)
console.log('Proxifying request to %s', url)
if (process.env.DEBUG) {
console.log('Receiving request on mocked server %s.', req.url)
console.log('Proxifying request to %s', url)
}
return pipeline(
got.stream(url, { throwHttpErrors: false }),
res,
(err: RequestError) => {
if (!err) return
console.error('Pipeline failed.', err)
}
return pipeline(
got.stream(url, { throwHttpErrors: false }),
res,
(err: RequestError) => {
if (!err) return
console.error('Pipeline failed.', err)
}
)
})
const port = 44000 + randomInt(1, 1000)
this.server = app.listen(port, () => res(port))
)
})
this.server = await randomListen(app)
return getPort(this.server)
}
terminate () {

View File

@ -1,7 +1,6 @@
import express, { Request, Response } from 'express'
import { Server } from 'http'
import { randomInt } from '@shared/core-utils'
import { terminateServer } from './utils'
import { getPort, randomListen, terminateServer } from './utils'
type BlocklistResponse = {
data: {
@ -15,17 +14,16 @@ export class MockBlocklist {
private body: BlocklistResponse
private server: Server
initialize () {
return new Promise<number>(res => {
const app = express()
async initialize () {
const app = express()
app.get('/blocklist', (req: Request, res: Response) => {
return res.json(this.body)
})
const port = 45000 + randomInt(1, 1000)
this.server = app.listen(port, () => res(port))
app.get('/blocklist', (req: Request, res: Response) => {
return res.json(this.body)
})
this.server = await randomListen(app)
return getPort(this.server)
}
replace (body: BlocklistResponse) {

View File

@ -1,18 +1,15 @@
import { createServer, Server } from 'http'
import proxy from 'proxy'
import { randomInt } from '@shared/core-utils'
import { terminateServer } from './utils'
import { getPort, terminateServer } from './utils'
class MockProxy {
private server: Server
initialize () {
return new Promise<number>(res => {
const port = 46000 + randomInt(1, 1000)
this.server = proxy(createServer())
this.server.listen(port, () => res(port))
this.server.listen(0, () => res(getPort(this.server)))
})
}

View File

@ -1,4 +1,18 @@
import { Express } from 'express'
import { Server } from 'http'
import { AddressInfo } from 'net'
function randomListen (app: Express) {
return new Promise<Server>(res => {
const server = app.listen(0, () => res(server))
})
}
function getPort (server: Server) {
const address = server.address() as AddressInfo
return address.port
}
function terminateServer (server: Server) {
if (!server) return Promise.resolve()
@ -13,5 +27,7 @@ function terminateServer (server: Server) {
}
export {
randomListen,
getPort,
terminateServer
}