'use strict' const childProcess = require('child_process') const exec = childProcess.exec const fork = childProcess.fork const fs = require('fs') const pathUtils = require('path') const request = require('supertest') const testUtils = { flushTests: flushTests, getFriendsList: getFriendsList, getVideo: getVideo, getVideosList: getVideosList, getVideosListPagination: getVideosListPagination, login: login, loginAndGetAccessToken: loginAndGetAccessToken, makeFriends: makeFriends, quitFriends: quitFriends, removeVideo: removeVideo, flushAndRunMultipleServers: flushAndRunMultipleServers, runServer: runServer, searchVideo: searchVideo, searchVideoWithPagination: searchVideoWithPagination, testImage: testImage, uploadVideo: uploadVideo } // ---------------------- Export functions -------------------- function flushTests (callback) { exec('npm run clean:server:test', callback) } function getFriendsList (url, end) { const path = '/api/v1/pods/' request(url) .get(path) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) .end(end) } function getVideo (url, id, end) { const path = '/api/v1/videos/' + id request(url) .get(path) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) .end(end) } function getVideosList (url, end) { const path = '/api/v1/videos' request(url) .get(path) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) .end(end) } function getVideosListPagination (url, start, count, end) { const path = '/api/v1/videos' request(url) .get(path) .query({ start: start }) .query({ count: count }) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) .end(end) } function login (url, client, user, expectedStatus, end) { if (!end) { end = expectedStatus expectedStatus = 200 } const path = '/api/v1/users/token' const body = { client_id: client.id, client_secret: client.secret, username: user.username, password: user.password, response_type: 'code', grant_type: 'password', scope: 'upload' } request(url) .post(path) .type('form') .send(body) .expect(expectedStatus) .end(end) } function loginAndGetAccessToken (server, callback) { login(server.url, server.client, server.user, 200, function (err, res) { if (err) return callback(err) return callback(null, res.body.access_token) }) } function makeFriends (url, accessToken, expectedStatus, callback) { if (!callback) { callback = expectedStatus expectedStatus = 204 } const path = '/api/v1/pods/makefriends' // The first pod make friend with the third request(url) .get(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(expectedStatus) .end(function (err, res) { if (err) throw err // Wait for the request between pods setTimeout(callback, 1000) }) } function quitFriends (url, accessToken, expectedStatus, callback) { if (!callback) { callback = expectedStatus expectedStatus = 204 } const path = '/api/v1/pods/quitfriends' // The first pod make friend with the third request(url) .get(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(expectedStatus) .end(function (err, res) { if (err) throw err // Wait for the request between pods setTimeout(callback, 1000) }) } function removeVideo (url, token, id, expectedStatus, end) { if (!end) { end = expectedStatus expectedStatus = 204 } const path = '/api/v1/videos' request(url) .delete(path + '/' + id) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + token) .expect(expectedStatus) .end(end) } function flushAndRunMultipleServers (totalServers, serversRun) { let apps = [] let urls = [] let i = 0 function anotherServerDone (number, app, url) { apps[number - 1] = app urls[number - 1] = url i++ if (i === totalServers) { serversRun(apps, urls) } } flushTests(function () { for (let j = 1; j <= totalServers; j++) { // For the virtual buffer setTimeout(function () { runServer(j, function (app, url) { anotherServerDone(j, app, url) }) }, 1000 * j) } }) } function runServer (number, callback) { const server = { app: null, url: `http://localhost:${9000 + number}`, client: { id: null, secret: null }, user: { username: null, password: null } } // These actions are async so we need to be sure that they have both been done const serverRunString = { 'Connected to mongodb': false, 'Server listening on port': false } const regexps = { client_id: 'Client id: ([a-f0-9]+)', client_secret: 'Client secret: (.+)', user_username: 'Username: (.+)', user_password: 'User password: (.+)' } // Share the environment const env = Object.create(process.env) env.NODE_ENV = 'test' env.NODE_APP_INSTANCE = number const options = { silent: true, env: env, detached: true } server.app = fork(pathUtils.join(__dirname, '../../../server.js'), [], options) server.app.stdout.on('data', function onStdout (data) { let dontContinue = false // Capture things if we want to for (const key of Object.keys(regexps)) { const regexp = regexps[key] const matches = data.toString().match(regexp) if (matches !== null) { if (key === 'client_id') server.client.id = matches[1] else if (key === 'client_secret') server.client.secret = matches[1] else if (key === 'user_username') server.user.username = matches[1] else if (key === 'user_password') server.user.password = matches[1] } } // Check if all required sentences are here for (const key of Object.keys(serverRunString)) { if (data.toString().indexOf(key) !== -1) serverRunString[key] = true if (serverRunString[key] === false) dontContinue = true } // If no, there is maybe one thing not already initialized (mongodb...) if (dontContinue === true) return server.app.stdout.removeListener('data', onStdout) callback(server) }) } function searchVideo (url, search, end) { const path = '/api/v1/videos' request(url) .get(path + '/search/' + search) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) .end(end) } function searchVideoWithPagination (url, search, start, count, end) { const path = '/api/v1/videos' request(url) .get(path + '/search/' + search) .query({ start: start }) .query({ count: count }) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) .end(end) } function testImage (url, videoName, imagePath, callback) { request(url) .get(imagePath) .expect(200) .end(function (err, res) { if (err) return callback(err) fs.readFile(pathUtils.join(__dirname, 'fixtures', videoName + '.jpg'), function (err, data) { if (err) return callback(err) callback(null, data.equals(res.body)) }) }) } function uploadVideo (url, accessToken, name, description, fixture, specialStatus, end) { if (!end) { end = specialStatus specialStatus = 204 } const path = '/api/v1/videos' request(url) .post(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .field('name', name) .field('description', description) .attach('videofile', pathUtils.join(__dirname, 'fixtures', fixture)) .expect(specialStatus) .end(end) } // --------------------------------------------------------------------------- module.exports = testUtils