muweb-socket/src/test/socket.ts

WebSocket communication for mudb

import ip = require('ip');
import test = require('tape');

import { MuSocketState } from 'mudb/socket';
import { MuWebSocket } from '../socket';

function noop () { }

function id () {
    return Math.random().toString(36).substr(2);
}

const url = `ws://${ip.address()}:${process.env.PORT}`;

if (typeof process === 'object' && 'exit' in process) {
    test.onFinish(() => process.exit(0));
}

test('socket initial state', (t) => {
    const socket = new MuWebSocket({
        sessionId: id(),
        url,
    });
    t.equals(socket.state, MuSocketState.INIT, 'should be INIT');
    t.end();
});

test('socket.open() - when INIT', (t) => {
    t.plan(1);

    let callsToReady = 0;

    const socket = new MuWebSocket({
        sessionId: id(),
        url,
    });

    socket.open({
        ready: () => {
            ++callsToReady;

            if (callsToReady > 1) {
                t.fail('should not call ready handler more than once');
            }
            t.equals(socket.state, MuSocketState.OPEN, 'should change socket state to OPEN');
        },
        message: noop,
        close: noop,
    });
});

test('socket.open() - when OPEN', (t) => {
    t.plan(1);

    const socket = new MuWebSocket({
        sessionId: id(),
        url,
    });

    socket.open({
        ready: () => {
            t.throws(
                () => socket.open({
                    ready: noop,
                    message: noop,
                    close: noop,
                }),
            );
        },
        message: noop,
        close: noop,
    });
});

test('socket.open() - when CLOSED', (t) => {
    t.plan(1);

    const socket = new MuWebSocket({
        sessionId: id(),
        url,
    });

    socket.open({
        ready: () => socket.close(),
        message: noop,
        close: () => t.throws(
            () => socket.open({
                ready: noop,
                message: noop,
                close: noop,
            }),
        ),
    });
});

test('socket.close() - when OPEN', (t) => {
    t.plan(1);

    const socket = new MuWebSocket({
        sessionId: id(),
        url,
    });

    socket.open({
        ready: () => socket.close(),
        message: noop,
        close: () => t.equals(socket.state, MuSocketState.CLOSED, 'should change socket state to CLOSED'),
    });
});

test('socket.close() - when INIT', (t) => {
    const socket = new MuWebSocket({
        sessionId: id(),
        url,
    });

    socket.close();
    t.equals(socket.state, MuSocketState.CLOSED, 'should change socket state to CLOSED');
    t.end();
});