Server: add script that try to simulate a real world situation
Each pod randomly upload or remove a video and send the information to its friend
This commit is contained in:
Chocobozzz
parent
b359546358
commit
8f68c31a71
|
|
@ -67,6 +67,7 @@
|
|||
},
|
||||
"devDependencies": {
|
||||
"chai": "^3.3.0",
|
||||
"commander": "^2.9.0",
|
||||
"livereload": "^0.4.1",
|
||||
"mocha": "^2.3.3",
|
||||
"standard": "^7.0.1",
|
||||
|
|
|
|||
|
|
@ -0,0 +1,215 @@
|
|||
'use strict'
|
||||
|
||||
const async = require('async')
|
||||
const program = require('commander')
|
||||
const isEqual = require('lodash/isEqual')
|
||||
|
||||
process.env.NODE_ENV = 'test'
|
||||
const constants = require('../../initializers/constants')
|
||||
|
||||
const utils = require('../api/utils')
|
||||
|
||||
program
|
||||
.option('-c, --create [weight]', 'Weight for creating videos')
|
||||
.option('-r, --remove [weight]', 'Weight for removing videos')
|
||||
.option('-p, --pods [n]', 'Number of pods to run (3 or 6)', /^3|6$/, 3)
|
||||
.option('-a, --action [interval]', 'Interval in ms for an action')
|
||||
.option('-i, --integrity [interval]', 'Interval in ms for an integrity check')
|
||||
.option('-f, --flush', 'Flush datas on exit')
|
||||
.parse(process.argv)
|
||||
|
||||
const createWeight = parseInt(program.create) || 5
|
||||
const removeWeight = parseInt(program.remove) || 4
|
||||
const flushAtExit = program.flush || false
|
||||
const actionInterval = parseInt(program.action) || 500
|
||||
let integrityInterval = parseInt(program.integrity) || 60000
|
||||
|
||||
const numberOfPods = 6
|
||||
// Wait requests between pods
|
||||
const requestsMaxPerInterval = constants.INTERVAL / actionInterval
|
||||
const intervalsToMakeAllRequests = Math.ceil(requestsMaxPerInterval / constants.REQUESTS_LIMIT)
|
||||
const waitForBeforeIntegrityCheck = (intervalsToMakeAllRequests * constants.INTERVAL) + 1000
|
||||
|
||||
integrityInterval += waitForBeforeIntegrityCheck
|
||||
|
||||
console.log('Create weight: %d, remove weight: %d.', createWeight, removeWeight)
|
||||
if (flushAtExit) {
|
||||
console.log('Program will flush data on exit.')
|
||||
} else {
|
||||
console.log('Program will not flush data on exit.')
|
||||
}
|
||||
console.log('Interval in ms for each action: %d.', actionInterval)
|
||||
console.log('Interval in ms for each integrity check: %d.', integrityInterval)
|
||||
console.log('Will wait %d ms before an integrity check.', waitForBeforeIntegrityCheck)
|
||||
|
||||
console.log('Run servers...')
|
||||
runServers(numberOfPods, function (err, servers) {
|
||||
if (err) throw err
|
||||
|
||||
process.on('exit', function () {
|
||||
exitServers(servers, flushAtExit)
|
||||
})
|
||||
process.on('SIGINT', goodbye)
|
||||
process.on('SIGTERM', goodbye)
|
||||
|
||||
console.log('Servers runned')
|
||||
|
||||
let checking = false
|
||||
|
||||
setInterval(function () {
|
||||
if (checking === true) return
|
||||
|
||||
const rand = getRandomInt(0, createWeight + removeWeight)
|
||||
|
||||
if (rand < createWeight) {
|
||||
upload(servers, getRandomNumServer(servers))
|
||||
} else {
|
||||
remove(servers, getRandomNumServer(servers))
|
||||
}
|
||||
}, actionInterval)
|
||||
|
||||
setInterval(function () {
|
||||
console.log('Checking integrity...')
|
||||
checking = true
|
||||
|
||||
setTimeout(function () {
|
||||
checkIntegrity(servers, function () {
|
||||
checking = false
|
||||
})
|
||||
}, waitForBeforeIntegrityCheck)
|
||||
}, integrityInterval)
|
||||
})
|
||||
|
||||
// ----------------------------------------------------------------------------
|
||||
|
||||
function getRandomInt (min, max) {
|
||||
return Math.floor(Math.random() * (max - min)) + min
|
||||
}
|
||||
|
||||
function getRandomNumServer (servers) {
|
||||
return getRandomInt(0, servers.length)
|
||||
}
|
||||
|
||||
function runServers (numberOfPods, callback) {
|
||||
let servers = null
|
||||
|
||||
async.series([
|
||||
// Run servers
|
||||
function (next) {
|
||||
utils.flushAndRunMultipleServers(numberOfPods, function (serversRun) {
|
||||
servers = serversRun
|
||||
next()
|
||||
})
|
||||
},
|
||||
// Get the access tokens
|
||||
function (next) {
|
||||
async.each(servers, function (server, callbackEach) {
|
||||
utils.loginAndGetAccessToken(server, function (err, accessToken) {
|
||||
if (err) return callbackEach(err)
|
||||
|
||||
server.accessToken = accessToken
|
||||
callbackEach()
|
||||
})
|
||||
}, next)
|
||||
},
|
||||
function (next) {
|
||||
const server = servers[1]
|
||||
utils.makeFriends(server.url, server.accessToken, next)
|
||||
},
|
||||
function (next) {
|
||||
const server = servers[0]
|
||||
utils.makeFriends(server.url, server.accessToken, next)
|
||||
},
|
||||
function (next) {
|
||||
setTimeout(next, 1000)
|
||||
},
|
||||
function (next) {
|
||||
const server = servers[3]
|
||||
utils.makeFriends(server.url, server.accessToken, next)
|
||||
},
|
||||
function (next) {
|
||||
const server = servers[5]
|
||||
utils.makeFriends(server.url, server.accessToken, next)
|
||||
},
|
||||
function (next) {
|
||||
const server = servers[4]
|
||||
utils.makeFriends(server.url, server.accessToken, next)
|
||||
},
|
||||
function (next) {
|
||||
setTimeout(next, 1000)
|
||||
}
|
||||
], function (err) {
|
||||
return callback(err, servers)
|
||||
})
|
||||
}
|
||||
|
||||
function exitServers (servers, callback) {
|
||||
if (!callback) callback = function () {}
|
||||
|
||||
servers.forEach(function (server) {
|
||||
if (server.app) process.kill(-server.app.pid)
|
||||
})
|
||||
|
||||
if (flushAtExit) utils.flushTests(callback)
|
||||
}
|
||||
|
||||
function upload (servers, numServer, callback) {
|
||||
if (!callback) callback = function () {}
|
||||
|
||||
const name = 'my super name for pod 1'
|
||||
const description = 'my super description for pod 1'
|
||||
const tags = [ 'tag1p1', 'tag2p1' ]
|
||||
const file = 'video_short1.webm'
|
||||
|
||||
console.log('Upload video to server ' + numServer)
|
||||
|
||||
utils.uploadVideo(servers[numServer].url, servers[numServer].accessToken, name, description, tags, file, callback)
|
||||
}
|
||||
|
||||
function remove (servers, numServer, callback) {
|
||||
if (!callback) callback = function () {}
|
||||
|
||||
utils.getVideosList(servers[numServer].url, function (err, res) {
|
||||
if (err) throw err
|
||||
|
||||
const videos = res.body.data
|
||||
if (videos.length === 0) return callback()
|
||||
|
||||
const toRemove = videos[getRandomInt(0, videos.length)].id
|
||||
|
||||
console.log('Removing video from server ' + numServer)
|
||||
utils.removeVideo(servers[numServer].url, servers[numServer].accessToken, toRemove, callback)
|
||||
})
|
||||
}
|
||||
|
||||
function checkIntegrity (servers, callback) {
|
||||
const videos = []
|
||||
async.each(servers, function (server, callback) {
|
||||
utils.getAllVideosListBy(server.url, function (err, res) {
|
||||
if (err) throw err
|
||||
const serverVideos = res.body.data
|
||||
for (const serverVideo of serverVideos) {
|
||||
delete serverVideo.id
|
||||
delete serverVideo.isLocal
|
||||
delete serverVideo.thumbnailPath
|
||||
}
|
||||
|
||||
videos.push(serverVideos)
|
||||
callback()
|
||||
})
|
||||
}, function () {
|
||||
for (const video of videos) {
|
||||
if (!isEqual(video, videos[0])) {
|
||||
console.error('Integrity not ok!')
|
||||
process.exit(-1)
|
||||
}
|
||||
}
|
||||
|
||||
console.log('Integrity ok.')
|
||||
return callback()
|
||||
})
|
||||
}
|
||||
|
||||
function goodbye () {
|
||||
return process.exit(-1)
|
||||
}
|
||||
Loading…
Reference in New Issue