Skip to content

Instantly share code, notes, and snippets.

@vothanhkiet
Forked from jmyrland/test.js
Created February 18, 2016 08:39
Show Gist options
  • Save vothanhkiet/0e2cdd97746e26e22ba1 to your computer and use it in GitHub Desktop.
Save vothanhkiet/0e2cdd97746e26e22ba1 to your computer and use it in GitHub Desktop.
Socket-io load test?
/**
* Modify the parts you need to get it working.
*/
var should = require('should');
var request = require('../node_modules/request');
var io = require('socket.io-client');
var serverUrl = 'http://localhost';
var options = {
transports: ['xhr-polling'],
'force new connection': true
};
describe("socket-io server", function () {
var clients = [];
it('Should manage 100 rapid connecting/reconnecting users', function (done) {
var connectionInterval = 250;
var numberOfUsersToConnect = 100;
var connections = 0;
var connectionTimestampSum = 0;
for (var i = 1; i <= numberOfUsersToConnect; i++) {
(function (index) {
var userIndex = (index - 1) % (testUsers.length - 1);
setTimeout(function () {
var time = Date.now();
var client = new SocketIoClient(serverUrl, {
onConnected: function () {
connections++;
connectionTimestampSum += Date.now() - time;
log((connections) + "/" + numberOfUsersToConnect + " connected! (" + (Date.now() - time )+ "ms)");
clients.push(client);
if (index == numberOfUsersToConnect) {
log("Average connect time: " + Math.round(connectionTimestampSum / connections) + " ms!");
log("");
setTimeout(function() {
done();
}, 2000);
}
}
});
}, index * connectionInterval);
})(i);
}
});
it('Should successfully send messages between clients', function (done) {
log("");
log("Client length: " + clients.length);
log("");
var chosenMessenger = Math.round(Math.random() % clients.length) % clients.length,
clientsReceived = 0,
sentTimestamp = 0,
receiveTimeSum = 0,
message = "Hello other clients!";
// Bind event to all connected clients:
for (var i = 0; i < clients.length; i++) {
(function (index) {
var event = function (newPost) {
clientsReceived++;
receiveTimeSum += Date.now() - sentTimestamp;
log("Received " + (clientsReceived) + "/" + clients.length + " messages!");
if (clientsReceived == clients.length - 1) {
log("Average receive time: " + Math.round(receiveTimeSum / clientsReceived) + " ms!");
log("");
setTimeout(function () {
done();
}, 2000);
}
}
clients[index].on('newPost all', event);
})(i);
}
// Send the message from one - to all the other clients:
sentTimestamp = Date.now();
clients[chosenMessenger].postMessage(message);
});
});
function log(message) {
process.stdout.write("\n\t\t" + message);
}
function SocketIoClient(connectionString, eventObject) {
// Establish socket connection to the node server:
var socket = this.socket = io.connect(connectionString, options);
this.eventObject = eventObject || {};
this.eventObject.onConnected = eventObject.onConnected || function () { };
this.eventObject.onError = eventObject.onError || function () { };
this.socket.on('connect', function () {
eventObject.onConnected();
});
this.socket.on('error', function (incomingError) {
eventObject.onError(incomingError);
});
this.socket.on('disconnect', function () {
eventObject.onDisconnect();
});
}
SocketIoClient.prototype = {
'disconnect': function() {
this.socket.disconnect();
},
'postMessage': function(message) {
this.socket.emit('post', {
message: message
});
},
'on': function () {
this.socket.on.apply(this.socket, arguments);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment