activestack-gateway/src/amqp/test/test-shift.js

Socket server for ActiveStack

require('./harness').run();

var recvCount = 0;
var body = "hello world";

connection.addListener('ready', function () {
  puts("connected to " + connection.serverProperties.product);

  var e = connection.exchange('node-ack-fanout', {type: 'fanout'});
  var q = connection.queue('node-123ack-queue', function() {
    q.bind(e, 'ackmessage.*');
    q.on('queueBindOk', function() {
      q.on('basicConsumeOk', function () {
        puts("publishing 2 json messages");

        e.publish('ackmessage.json1', { name: 'A' });
        e.publish('ackmessage.json2', { name: 'B' });
        e.publish('ackmessage.json3', { name: 'C' });
      });
      
      q.subscribe({ ack: true }, function (json) {
        recvCount++;
        puts('Got message ' + JSON.stringify(json));

        if (recvCount == 1) {
          puts('Got message 1.. waiting');
          assert.equal('A', json.name);
          setTimeout(function () {
            puts('shift!');
            q.shift();
          }, 1000);
        } else if (recvCount == 2) {
          puts('got message 2');
          assert.equal('B', json.name);
          setTimeout(function () {
            puts('reject!');
            q.shift(true);
          }, 1000);
        } else if (recvCount == 3) {
          puts('got message 3');
          assert.equal('C', json.name);
          setTimeout(function () {
            puts('requeue!');
            q.shift(true, true);
          }, 1000);
        } else if (recvCount == 4) {
          puts('got message 4');
          assert.equal('C', json.name);

          puts('closing connection');

          connection.end();
        } else {
          throw new Error('Too many message!');
        }
      });
    });
  });
});


process.addListener('exit', function () {
  assert.equal(4, recvCount);
});