occaecatidicta/packages/omelox/test/service/connectionService.ts

import * as should from 'should';
// import { describe, it } from "mocha-typescript"
let ConnectionService = require('../../lib/common/service/connectionService').ConnectionService;

let mockApp = {
    settings: {
        serverId: 'connector-server-1'
    },

    get: function (key: 'serverId') {
        return this.settings[key];
    },

    getServerId: function () {
        return this.get('serverId');
    }
};

describe('connection service test', function () {
    describe('#addLoginedUser', function () {
        it('should add logined user and could fetch it later', function () {
            let service = new ConnectionService(mockApp);
            should.exist(service);
            service.loginedCount.should.equal(0);

            let uid = 'uid1';
            let info = { msg: 'some other message' };
            service.addLoginedUser(uid, info);

            service.loginedCount.should.equal(1);
            let record = service.logined[uid];
            should.exist(record);
            record.should.eql(info);
        });
    });

    describe('#increaseConnectionCount', function () {
        it('should increate connection count and could fetch it later', function () {
            let service = new ConnectionService(mockApp);
            should.exist(service);
            service.connCount.should.equal(0);

            service.increaseConnectionCount();
            service.connCount.should.equal(1);
        });
    });

    describe('#removeLoginedUser', function () {
        it('should remove logined user info with the uid', function () {
            let service = new ConnectionService(mockApp);
            should.exist(service);
            service.loginedCount.should.equal(0);

            let uid = 'uid1';
            let info = { msg: 'some other message' };
            service.addLoginedUser(uid, info);

            service.loginedCount.should.equal(1);
            let record = service.logined[uid];
            should.exist(record);

            let uid2 = 'uid2';
            service.removeLoginedUser(uid2);
            service.loginedCount.should.equal(1);
            record = service.logined[uid];
            should.exist(record);

            service.removeLoginedUser(uid);
            service.loginedCount.should.equal(0);
            record = service.logined[uid];
            should.not.exist(record);
        });
    });

    describe('#decreaseConnectionCount', function () {
        it('should decrease connection count only if uid is empty', function () {
            let service = new ConnectionService(mockApp);
            should.exist(service);

            service.increaseConnectionCount();
            service.connCount.should.equal(1);
            service.decreaseConnectionCount();
            service.connCount.should.equal(0);
        });

        it('should keep zero if connection count become zero', function () {
            let service = new ConnectionService(mockApp);
            should.exist(service);

            service.connCount.should.equal(0);
            service.decreaseConnectionCount();
            service.connCount.should.equal(0);
        });

        it('should remove the logined info if uid is specified', function () {
            let service = new ConnectionService(mockApp);
            should.exist(service);

            service.increaseConnectionCount();

            let uid = 'uid1';
            let info = { msg: 'some other message' };
            service.addLoginedUser(uid, info);

            service.connCount.should.equal(1);
            service.logined[uid].should.eql(info);

            service.decreaseConnectionCount(uid);

            service.connCount.should.equal(0);
            should.not.exist(service.logined[uid]);
        });
    });

    it('should getStatisticsInfo', function (done: MochaDone) {
        let service = new ConnectionService(mockApp);
        let uid1 = 'uid1', uid2 = 'uid2';
        let info1 = { msg: 'msg1' }, info2 = { msg: 'msg2' };

        service.increaseConnectionCount();
        service.increaseConnectionCount();
        service.increaseConnectionCount();

        service.addLoginedUser(uid1, info1);
        service.addLoginedUser(uid2, info2);


        let sinfo = service.getStatisticsInfo();

        sinfo.should.have.property('serverId', 'connector-server-1');
        sinfo.should.have.property('totalConnCount', 3);
        sinfo.should.have.property('loginedCount', 2);

        let infos = sinfo.loginedList;
        should.exist(infos);
        infos.length.should.equal(2);
        infos.should.containEql(info1);
        infos.should.containEql(info2);

        done();
    });
});