rooms.getAll test

src/socket.io/admin/rooms.js

@@ -1,6 +1,7 @@
 'use strict';
 
 
+var async = require('async');
 var os = require('os');
 var nconf = require('nconf');
 var winston = require('winston');
@@ -68,7 +69,7 @@ SocketRooms.getAll = function (socket, data, callback) {
 		category: 0
 	};
 
-	for(var instance in stats) {
+	for (var instance in stats) {
 		if (stats.hasOwnProperty(instance)) {
 			totals.onlineGuestCount += stats[instance].onlineGuestCount;
 			totals.onlineRegisteredCount += stats[instance].onlineRegisteredCount;
@@ -99,10 +100,11 @@ SocketRooms.getAll = function (socket, data, callback) {
 		return topic.tid;
 	});
 
-	topics.getTopicsFields(topTenTids, ['title'], function (err, titles) {
+	async.waterfall([
-		if (err) {
+		function (next) {
-			return callback(err);
+			topics.getTopicsFields(topTenTids, ['title'], next);
-		}
+		},
+		function (titles, next) {
 			totals.topics = {};
 			topTenTopics.forEach(function (topic, index) {
 				totals.topics[topic.tid] = {
@@ -110,21 +112,21 @@ SocketRooms.getAll = function (socket, data, callback) {
 					title: validator.escape(String(titles[index].title))
 				};
 			});
-
+			next(null, totals);
-		callback(null, totals);
+		}
-	});
+	], callback);
 };
 
 SocketRooms.getOnlineUserCount = function (io) {
-	if (!io) {
-		return 0;
-	}
 	var count = 0;
+
+	if (io) {
 		for (var key in io.sockets.adapter.rooms) {
 			if (io.sockets.adapter.rooms.hasOwnProperty(key) && key.startsWith('uid_')) {
 				++ count;
 			}
 		}
+	}
 
 	return count;
 };
@@ -132,25 +134,29 @@ SocketRooms.getOnlineUserCount = function (io) {
 SocketRooms.getLocalStats = function (callback) {
 	var io = require('../index').server;
 
-	if (!io) {
-		return callback();
-	}
-
-	var roomClients = io.sockets.adapter.rooms;
 	var socketData = {
-		onlineGuestCount: roomClients.online_guests ? roomClients.online_guests.length : 0,
+		onlineGuestCount: 0,
-		onlineRegisteredCount: SocketRooms.getOnlineUserCount(io),
+		onlineRegisteredCount: 0,
-		socketCount: Object.keys(io.sockets.sockets).length,
+		socketCount: 0,
 		users: {
-			categories: roomClients.categories ? roomClients.categories.length : 0,
+			categories: 0,
-			recent: roomClients.recent_topics ? roomClients.recent_topics.length : 0,
+			recent: 0,
-			unread: roomClients.unread_topics ? roomClients.unread_topics.length : 0,
+			unread: 0,
 			topics: 0,
 			category: 0
 		},
 		topics: {}
 	};
 
+	if (io) {
+		var roomClients = io.sockets.adapter.rooms;
+		socketData.onlineGuestCount = roomClients.online_guests ? roomClients.online_guests.length : 0;
+		socketData.onlineRegisteredCount = SocketRooms.getOnlineUserCount(io);
+		socketData.socketCount = Object.keys(io.sockets.sockets).length;
+		socketData.users.categories = roomClients.categories ? roomClients.categories.length : 0;
+		socketData.users.recent = roomClients.recent_topics ? roomClients.recent_topics.length : 0;
+		socketData.users.unread = roomClients.unread_topics ? roomClients.unread_topics.length : 0;
+
 		var topTenTopics = [];
 		var tid;
 
@@ -171,6 +177,8 @@ SocketRooms.getLocalStats = function (callback) {
 		}).slice(0, 10);
 
 		socketData.topics = topTenTopics;
+	}
+
 	callback(null, socketData);
 };
 

test/socket.io.js

@@ -380,6 +380,48 @@ describe('socket.io', function () {
 		});
 	});
 
+	it('should get room stats', function (done) {
+		var socketAdmin = require('../src/socket.io/admin');
+
+		io.emit('meta.rooms.enter', {enter: 'topic_1'}, function (err) {
+			assert.ifError(err);
+			socketAdmin.rooms.getAll({uid: 10}, {}, function (err) {
+				assert.ifError(err);
+				setTimeout(function () {
+					socketAdmin.rooms.getAll({uid: 10}, {}, function (err, data) {
+						assert.ifError(err);
+						assert(data.hasOwnProperty('onlineGuestCount'));
+						assert(data.hasOwnProperty('onlineRegisteredCount'));
+						assert(data.hasOwnProperty('socketCount'));
+						assert(data.hasOwnProperty('topics'));
+						assert(data.hasOwnProperty('users'));
+						assert.equal(data.topics['1'].title, 'test topic title')
+						done();
+					});
+				}, 1000);
+			});
+		});
+	});
+
+	it('should get room stats', function (done) {
+		var socketAdmin = require('../src/socket.io/admin');
+
+		io.emit('meta.rooms.enter', {enter: 'category_1'}, function (err) {
+			assert.ifError(err);
+			socketAdmin.rooms.getAll({uid: 10}, {}, function (err) {
+				assert.ifError(err);
+				setTimeout(function () {
+					socketAdmin.rooms.getAll({uid: 10}, {}, function (err, data) {
+						assert.ifError(err);
+						assert.equal(data.users.category, 1);
+						done();
+					});
+				}, 1000);
+			});
+		});
+	});
+
+
 	after(function (done) {
 		db.emptydb(done);
 	});