server/tests/api/utils.js

   1 'use strict'
   2
   3 const child_process = require('child_process')
   4 const exec = child_process.exec
   5 const fork = child_process.fork
   6 const pathUtils = require('path')
   7 const request = require('supertest')
   8
   9 const testUtils = {
  10   flushTests: flushTests,
  11   getFriendsList: getFriendsList,
  12   getVideo: getVideo,
  13   getVideosList: getVideosList,
  14   login: login,
  15   loginAndGetAccessToken: loginAndGetAccessToken,
  16   makeFriends: makeFriends,
  17   quitFriends: quitFriends,
  18   removeVideo: removeVideo,
  19   flushAndRunMultipleServers: flushAndRunMultipleServers,
  20   runServer: runServer,
  21   searchVideo: searchVideo,
  22   uploadVideo: uploadVideo
  23 }
  24
  25 // ---------------------- Export functions --------------------
  26
  27 function flushTests (callback) {
  28   exec(pathUtils.join(__dirname, '../../../bin/clean_test.sh'), callback)
  29 }
  30
  31 function getFriendsList (url, end) {
  32   const path = '/api/v1/pods/'
  33
  34   request(url)
  35     .get(path)
  36     .set('Accept', 'application/json')
  37     .expect(200)
  38     .expect('Content-Type', /json/)
  39     .end(end)
  40 }
  41
  42 function getVideo (url, id, end) {
  43   const path = '/api/v1/videos/' + id
  44
  45   request(url)
  46     .get(path)
  47     .set('Accept', 'application/json')
  48     .expect(200)
  49     .expect('Content-Type', /json/)
  50     .end(end)
  51 }
  52
  53 function getVideosList (url, end) {
  54   const path = '/api/v1/videos'
  55
  56   request(url)
  57     .get(path)
  58     .set('Accept', 'application/json')
  59     .expect(200)
  60     .expect('Content-Type', /json/)
  61     .end(end)
  62 }
  63
  64 function login (url, client, user, expected_status, end) {
  65   if (!end) {
  66     end = expected_status
  67     expected_status = 200
  68   }
  69
  70   const path = '/api/v1/users/token'
  71
  72   const body = {
  73     client_id: client.id,
  74     client_secret: client.secret,
  75     username: user.username,
  76     password: user.password,
  77     response_type: 'code',
  78     grant_type: 'password',
  79     scope: 'upload'
  80   }
  81
  82   request(url)
  83     .post(path)
  84     .type('form')
  85     .send(body)
  86     .expect(expected_status)
  87     .end(end)
  88 }
  89
  90 function loginAndGetAccessToken (server, callback) {
  91   login(server.url, server.client, server.user, 200, function (err, res) {
  92     if (err) return callback(err)
  93
  94     return callback(null, res.body.access_token)
  95   })
  96 }
  97
  98 function makeFriends (url, expected_status, callback) {
  99   if (!callback) {
 100     callback = expected_status
 101     expected_status = 204
 102   }
 103
 104   const path = '/api/v1/pods/makefriends'
 105
 106   // The first pod make friend with the third
 107   request(url)
 108     .get(path)
 109     .set('Accept', 'application/json')
 110     .expect(expected_status)
 111     .end(function (err, res) {
 112       if (err) throw err
 113
 114       // Wait for the request between pods
 115       setTimeout(callback, 1000)
 116     })
 117 }
 118
 119 function quitFriends (url, callback) {
 120   const path = '/api/v1/pods/quitfriends'
 121
 122   // The first pod make friend with the third
 123   request(url)
 124     .get(path)
 125     .set('Accept', 'application/json')
 126     .expect(204)
 127     .end(function (err, res) {
 128       if (err) throw err
 129
 130       // Wait for the request between pods
 131       setTimeout(callback, 1000)
 132     })
 133 }
 134
 135 function removeVideo (url, token, id, expected_status, end) {
 136   if (!end) {
 137     end = expected_status
 138     expected_status = 204
 139   }
 140
 141   const path = '/api/v1/videos'
 142
 143   request(url)
 144     .delete(path + '/' + id)
 145     .set('Accept', 'application/json')
 146     .set('Authorization', 'Bearer ' + token)
 147     .expect(expected_status)
 148     .end(end)
 149 }
 150
 151 function flushAndRunMultipleServers (total_servers, serversRun) {
 152   let apps = []
 153   let urls = []
 154   let i = 0
 155
 156   function anotherServerDone (number, app, url) {
 157     apps[number - 1] = app
 158     urls[number - 1] = url
 159     i++
 160     if (i === total_servers) {
 161       serversRun(apps, urls)
 162     }
 163   }
 164
 165   flushTests(function () {
 166     for (let j = 1; j <= total_servers; j++) {
 167       // For the virtual buffer
 168       setTimeout(function () {
 169         runServer(j, function (app, url) {
 170           anotherServerDone(j, app, url)
 171         })
 172       }, 1000 * j)
 173     }
 174   })
 175 }
 176
 177 function runServer (number, callback) {
 178   const server = {
 179     app: null,
 180     url: `http://localhost:${9000 + number}`,
 181     client: {
 182       id: null,
 183       secret: null
 184     },
 185     user: {
 186       username: null,
 187       password: null
 188     }
 189   }
 190
 191   // These actions are async so we need to be sure that they have both been done
 192   const server_run_string = {
 193     'Connected to mongodb': false,
 194     'Server listening on port': false
 195   }
 196
 197   const regexps = {
 198     client_id: 'Client id: ([a-f0-9]+)',
 199     client_secret: 'Client secret: (.+)',
 200     user_username: 'Username: (.+)',
 201     user_password: 'User password: (.+)'
 202   }
 203
 204   // Share the environment
 205   const env = Object.create(process.env)
 206   env.NODE_ENV = 'test'
 207   env.NODE_APP_INSTANCE = number
 208   const options = {
 209     silent: true,
 210     env: env,
 211     detached: true
 212   }
 213
 214   server.app = fork(pathUtils.join(__dirname, '../../../server.js'), [], options)
 215   server.app.stdout.on('data', function onStdout (data) {
 216     let dont_continue = false
 217
 218     // Capture things if we want to
 219     for (const key of Object.keys(regexps)) {
 220       const regexp = regexps[key]
 221       const matches = data.toString().match(regexp)
 222       if (matches !== null) {
 223         if (key === 'client_id') server.client.id = matches[1]
 224         else if (key === 'client_secret') server.client.secret = matches[1]
 225         else if (key === 'user_username') server.user.username = matches[1]
 226         else if (key === 'user_password') server.user.password = matches[1]
 227       }
 228     }
 229
 230     // Check if all required sentences are here
 231     for (const key of Object.keys(server_run_string)) {
 232       if (data.toString().indexOf(key) !== -1) server_run_string[key] = true
 233       if (server_run_string[key] === false) dont_continue = true
 234     }
 235
 236     // If no, there is maybe one thing not already initialized (mongodb...)
 237     if (dont_continue === true) return
 238
 239     server.app.stdout.removeListener('data', onStdout)
 240     callback(server)
 241   })
 242 }
 243
 244 function searchVideo (url, search, end) {
 245   const path = '/api/v1/videos'
 246
 247   request(url)
 248     .get(path + '/search/' + search)
 249     .set('Accept', 'application/json')
 250     .expect(200)
 251     .expect('Content-Type', /json/)
 252     .end(end)
 253 }
 254
 255 function uploadVideo (url, access_token, name, description, fixture, special_status, end) {
 256   if (!end) {
 257     end = special_status
 258     special_status = 204
 259   }
 260
 261   const path = '/api/v1/videos'
 262
 263   request(url)
 264     .post(path)
 265     .set('Accept', 'application/json')
 266     .set('Authorization', 'Bearer ' + access_token)
 267     .field('name', name)
 268     .field('description', description)
 269     .attach('videofile', pathUtils.join(__dirname, 'fixtures', fixture))
 270     .expect(special_status)
 271     .end(end)
 272 }
 273
 274 // ---------------------------------------------------------------------------
 275
 276 module.exports = testUtils