1 import Bluebird from 'bluebird'
2 import { wait } from '@shared/core-utils'
3 import { createSingleServer, doubleFollow, PeerTubeServer, setAccessTokensToServers, waitJobs } from '@shared/server-commands'
5 let servers: PeerTubeServer[]
6 const viewers: { xForwardedFor: string }[] = []
10 .then(() => process.exit(0))
11 .catch(err => console.error(err))
13 async function run () {
21 async function prepare () {
22 console.log('Preparing servers...')
35 local_buffer_update_interval: '30 minutes',
36 ip_view_expiration: '1 hour'
41 servers = await Promise.all([
42 createSingleServer(1, config, { nodeArgs: [ '--inspect' ] }),
43 createSingleServer(2, config),
44 createSingleServer(3, config)
47 await setAccessTokensToServers(servers)
48 await doubleFollow(servers[0], servers[1])
49 await doubleFollow(servers[0], servers[2])
51 const { uuid } = await servers[0].videos.quickUpload({ name: 'video' })
54 await waitJobs(servers)
56 const THOUSAND_VIEWERS = 2
58 for (let i = 2; i < 252; i++) {
59 for (let j = 2; j < 6; j++) {
60 for (let k = 2; k < THOUSAND_VIEWERS + 2; k++) {
61 viewers.push({ xForwardedFor: `0.${k}.${j}.${i},127.0.0.1` })
66 console.log('Servers preparation finished.')
69 async function runViewers () {
70 console.log('Will run views of %d viewers.', viewers.length)
72 await Bluebird.map(viewers, viewer => {
73 return servers[0].views.simulateView({ id: videoId, xForwardedFor: viewer.xForwardedFor })
74 }, { concurrency: 100 })