NodeBB/test/helpers/index.js

'use strict';

var request = require('request');
var nconf = require('nconf');
var fs = require('fs');

var myXhr = require('../mocks/newXhr');
var utils = require('../../public/src/utils');

var helpers = module.exports;

helpers.loginUser = function (username, password, callback) {
	var jar = request.jar();
	request({
		url: nconf.get('url') + '/api/config',
		json: true,
		jar: jar,
	}, function (err, res, body) {
		if (err || res.statusCode !== 200) {
			return callback(err || new Error('[[error:invalid-response]]'));
		}

		request.post(nconf.get('url') + '/login', {
			form: {
				username: username,
				password: password,
			},
			json: true,
			jar: jar,
			headers: {
				'x-csrf-token': body.csrf_token,
			},
		}, function (err, res) {
			if (err || res.statusCode !== 200) {
				return callback(err || new Error('[[error:invalid-response]]'));
			}
			myXhr.callbacks.test2 = function () {
				this.setDisableHeaderCheck(true);
				var stdOpen = this.open;
				this.open = function () {
					stdOpen.apply(this, arguments);
					this.setRequestHeader('Cookie', res.headers['set-cookie'][0].split(';')[0]);
					this.setRequestHeader('Origin', nconf.get('url'));
				};
			};

			var socketClient = require('socket.io-client');

			var io = socketClient.connect(nconf.get('url'), { forceNew: true, multiplex: false });
			io.on('connect', function () {
				callback(null, jar, io, body.csrf_token);
			});

			io.on('error', function (err) {
				callback(err);
			});
		});
	});
};

helpers.initSocketIO = function (callback) {
	var jar;
	request.get({
		url: nconf.get('url') + '/api/config',
		jar: jar,
		json: true,
	}, function (err, res, body) {
		if (err) {
			return callback(err);
		}

		myXhr.callbacks.test2 = function () {
			this.setDisableHeaderCheck(true);
			var stdOpen = this.open;
			this.open = function () {
				stdOpen.apply(this, arguments);
				this.setRequestHeader('Cookie', res.headers['set-cookie'][0].split(';')[0]);
				this.setRequestHeader('Origin', nconf.get('url'));
			};
		};

		var io = require('socket.io-client')(nconf.get('url'), { forceNew: true });

		io.on('connect', function () {
			callback(null, jar, io);
		});

		io.on('error', function (err) {
			callback(err);
		});
	});
};


helpers.uploadFile = function (uploadEndPoint, filePath, body, jar, csrf_token, callback) {
	var formData = {
		files: [
			fs.createReadStream(filePath),
			fs.createReadStream(filePath), // see https://github.com/request/request/issues/2445
		],
	};
	formData = utils.merge(formData, body);
	request.post({
		url: uploadEndPoint,
		formData: formData,
		json: true,
		jar: jar,
		headers: {
			'x-csrf-token': csrf_token,
		},
	}, function (err, res, body) {
		if (err) {
			return callback(err);
		}
		callback(err, res, body);
	});
};

helpers.registerUser = function (data, callback) {
	var jar = request.jar();
	request({
		url: nconf.get('url') + '/api/config',
		json: true,
		jar: jar,
	}, function (err, response, body) {
		if (err) {
			return callback(err);
		}

		request.post(nconf.get('url') + '/register', {
			form: data,
			json: true,
			jar: jar,
			headers: {
				'x-csrf-token': body.csrf_token,
			},
		}, function (err, res, body) {
			if (err) {
				return callback(err);
			}

			callback(null, jar);
		});
	});
};

// http://stackoverflow.com/a/14387791/583363
helpers.copyFile = function (source, target, callback) {
	var cbCalled = false;

	var rd = fs.createReadStream(source);
	rd.on('error', function (err) {
		done(err);
	});
	var wr = fs.createWriteStream(target);
	wr.on('error', function (err) {
		done(err);
	});
	wr.on('close', function () {
		done();
	});
	rd.pipe(wr);

	function done(err) {
		if (!cbCalled) {
			callback(err);
			cbCalled = true;
		}
	}
};