X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Ftests%2Fapi%2Futils.js;h=f0b1c3653db6a9e2f0b63c8a63f0f9b5c393fbdb;hb=fbf1134e3e6b386c9ec5a3946c61f3faf84397fb;hp=1b2f6105949b7f7710988e3e431c247157cc40e5;hpb=8c9c1942f6b5675cfca802e2f707bef3c217afbf;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/tests/api/utils.js b/server/tests/api/utils.js index 1b2f61059..f0b1c3653 100644 --- a/server/tests/api/utils.js +++ b/server/tests/api/utils.js @@ -1,8 +1,9 @@ 'use strict' -const child_process = require('child_process') -const exec = child_process.exec -const fork = child_process.fork +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') @@ -11,19 +12,24 @@ const testUtils = { 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(pathUtils.join(__dirname, '../../../bin/clean_test.sh'), callback) + exec('npm run clean:server:test', callback) } function getFriendsList (url, end) { @@ -59,10 +65,57 @@ function getVideosList (url, end) { .end(end) } -function makeFriends (url, expected_status, callback) { +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 = expected_status - expected_status = 204 + callback = expectedStatus + expectedStatus = 204 } const path = '/api/v1/pods/makefriends' @@ -71,7 +124,8 @@ function makeFriends (url, expected_status, callback) { request(url) .get(path) .set('Accept', 'application/json') - .expect(expected_status) + .set('Authorization', 'Bearer ' + accessToken) + .expect(expectedStatus) .end(function (err, res) { if (err) throw err @@ -80,14 +134,20 @@ function makeFriends (url, expected_status, callback) { }) } -function quitFriends (url, callback) { +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') - .expect(204) + .set('Authorization', 'Bearer ' + accessToken) + .expect(expectedStatus) .end(function (err, res) { if (err) throw err @@ -96,17 +156,23 @@ function quitFriends (url, callback) { }) } -function removeVideo (url, id, end) { +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') - .expect(204) + .set('Authorization', 'Bearer ' + token) + .expect(expectedStatus) .end(end) } -function flushAndRunMultipleServers (total_servers, serversRun) { +function flushAndRunMultipleServers (totalServers, serversRun) { let apps = [] let urls = [] let i = 0 @@ -115,13 +181,13 @@ function flushAndRunMultipleServers (total_servers, serversRun) { apps[number - 1] = app urls[number - 1] = url i++ - if (i === total_servers) { + if (i === totalServers) { serversRun(apps, urls) } } flushTests(function () { - for (let j = 1; j <= total_servers; j++) { + for (let j = 1; j <= totalServers; j++) { // For the virtual buffer setTimeout(function () { runServer(j, function (app, url) { @@ -133,12 +199,32 @@ function flushAndRunMultipleServers (total_servers, serversRun) { } function runServer (number, callback) { - const port = 9000 + number - const server_run_string = { + 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' @@ -149,20 +235,33 @@ function runServer (number, callback) { detached: true } - const app = fork(pathUtils.join(__dirname, '../../../server.js'), [], options) - app.stdout.on('data', function onStdout (data) { - let dont_continue = false + 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(server_run_string)) { - if (data.toString().indexOf(key) !== -1) server_run_string[key] = true - if (server_run_string[key] === false) dont_continue = true + 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 (dont_continue === true) return + if (dontContinue === true) return - app.stdout.removeListener('data', onStdout) - callback(app, 'http://localhost:' + port) + server.app.stdout.removeListener('data', onStdout) + callback(server) }) } @@ -177,16 +276,50 @@ function searchVideo (url, search, end) { .end(end) } -function uploadVideo (url, name, description, fixture, 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(204) + .expect(specialStatus) .end(end) }