pubsub-ping-test.js

#!/usr/bin/env node

/* 
 * Spade, pubsub mode events test.
 */
exports.test = function ( done, assertions ) {

    var debug = !! true
        , emptyFn = function () {}
        , log = console.log
        , dbg = debug ? console.log : emptyFn
        , test_utils = require( './deps/test-utils' )
        , inspect = test_utils.inspect
        , format = test_utils.format
        , Spade = require( '../' )
        , client = Spade()
        // expected events
        , evts = []
        // collected events
        , collected = client.logger.collected
        , channels = [ 'd', 'e', 'u', 'c', 'e', 's' ]
        , p = 0
        , exit = typeof done === 'function' ? done : function () {}
        , assert = assertions || require( 'assert' )
        ;
    log( '- a new Spade client was created with custom options:', inspect( client.options ) );

    log( '- enable CLI logging.' );

    client.cli( true, function ( ename, args ) {
        dbg( '  !%s %s', ename, format( ename, args || [] ) );
    }, true );

    log( '- opening client connection.' );

    client.connect( null, function () {
        var i = 0
            ;
        log( '- now client is connected and ready to send.' );

        // push expected events
        evts.push( 'connect', 'reply', 'dbselected', 'scanqueue', 'ready', 'listen' );

        // push expected events, 6 messages from SUBSCRIBE
        for ( ; i < channels.length; ++i ) evts.push( 'message' );

        // push expected reply events form PING, 6 messages

        client.commands.subscribe( channels, function () {
            client.commands.ping( 'Eila', function ( is_err, reply, fn ) {
                if ( is_err ) {
                    log( '- this Redis not support ping in PubSub mode, no matter.. ( got: %s ).', fn( reply ) );
                    evts.push( 'error-reply' );
                } else evts.push( 'reply' );
                ++p;
            } );
        } );
    } );

    log( '- now check if current Redis version supports PING in pubsub mode, wait 2 secs..' );
    /*
     * if process doesn't exit, then PING is supported in PubSub mode,
     * reset and start test.
     */
    setTimeout( function () {

        log( '- ok, now test is running..' );

        log( '- now waiting 2 secs to collect events..' );
     
        setTimeout( function () {
            log( '- now disconnecting client with QUIT.' );

            // push expected connection event
            evts.push( 'reply', 'offline', 'lost' );

            client.commands.quit( function ( is_err, reply, fn ) {
                log( '- QUIT callback.', fn( reply ) );
                assert.ok( fn( reply )[ 0 ] === 'OK' );
                log( '- OK, client was disconnected.' );
            } );

            setTimeout( function () {

                log( '- check collected events for client, should be:', inspect( evts ) );
                assert.deepEqual( collected.events, evts, 'got: ' + inspect( collected.events ) );
                
                log( '- check PING calls, should be:', 6 );
                assert.equal( p, 6 );

                exit();

            }, 1000 );

        }, 2000 );

    }, 2000 );

};

// single test execution with node
if ( process.argv[ 1 ] === __filename ) exports.test = exports.test();