signal-desktop/libtextsecure/test/websocket_test.js

/* vim: ts=4:sw=4:expandtab
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

describe('TextSecureWebSocket', function() {
    var RealWebSocket = window.WebSocket;
    before(function() { window.WebSocket = MockSocket; });
    after (function() { window.WebSocket = RealWebSocket;  });
    it('connects and disconnects', function(done) {
        var mockServer = new MockServer('ws://localhost:8080');
        mockServer.on('connection', function(server) {
            socket.close();
            server.close();
            done();
        });
        var socket = new TextSecureWebSocket('ws://localhost:8080');
    });

    it('sends and receives', function(done) {
        var mockServer = new MockServer('ws://localhost:8080');
        mockServer.on('connection', function(server) {
            server.on('message', function(data) {
                server.send('ack');
                server.close();
            });
        });
        var socket = new TextSecureWebSocket('ws://localhost:8080');
        socket.onmessage = function(response) {
            assert.strictEqual(response.data, 'ack');
            socket.close();
            done();
        };
        socket.send('syn');

    });

    it('exposes the socket status', function(done) {
        var mockServer = new MockServer('ws://localhost:8082');
        mockServer.on('connection', function(server) {
            assert.strictEqual(socket.getStatus(), WebSocket.OPEN);
            server.close();
            socket.close();
        });
        var socket = new TextSecureWebSocket('ws://localhost:8082');
        socket.onclose = function() {
            assert.strictEqual(socket.getStatus(), WebSocket.CLOSING);
            done();
        };
    });

    it('reconnects', function(done) {
        this.timeout(60000);
        var mockServer = new MockServer('ws://localhost:8082');
        var socket = new TextSecureWebSocket('ws://localhost:8082');
        socket.onclose = function() {
            var mockServer = new MockServer('ws://localhost:8082');
            mockServer.on('connection', function(server) {
                socket.close();
                server.close();
                done();
            });
        };
        mockServer.close();
    });

});