test-memcache.js

/*
tests for expresso
*/

var util = require('util'),
    memcache = require('memcache'),
    assert = require('assert'),
    port = 11211;

mc = new memcache.Client(port);
mc.on('error', function(e){

	if (e.errno == 111){
		exports['startup test'] = function(){

			assert.ok(false, "You need to have a memcache server running on localhost:11211 for these tests to run");
		}
		return;
	}

	exports['startup test'] = function(){
		assert.ok(false, "Unexpected error during connection: "+util.inspect(e));
	}
});
mc.connect();


mc.addHandler(function() {

	// test nonexistent key is null
	exports['test null value'] = function(beforeExit) {
		var n = 0;
		mc.get('no such key', function(err, r) {
			assert.equal(null, r);
			n++;
		});
		
		beforeExit(function() {
			assert.equal(1, n);
		});
	};
	
	// test set, get and expires
	exports['test set, get, and expires'] = function(beforeExit) {
		var n = 0;
		// set key
		mc.set('set1', 'asdf1', function() {
			n++;
			mc.get('set1', function(err, r) {
				// assert key is found
				assert.equal('asdf1', r);
				n++;
				// assert key expires after 1s
				setTimeout(function() {
					mc.get('set1', function(r) {
						mc.close();
						assert.equal(null, r);
						n++;
					});
				}, 1000);
			});
		}, 1);
		
		beforeExit(function() {
			assert.equal(3, n);
		});
	};

    exports['test set get with integer value'] = function(beforeExit) {
        mc.set('testKey', 123, function() {
            mc.get('testKey', function(err, r) {
                assert.equal(123,r);
            });
        });
    };

	// test set and delete
	exports['test set del'] = function(beforeExit) {
		var n = 0;
		// set key
		mc.set('set2', 'asdf2', function() {
			n++;
			mc.get('set2', function(err, r) {
				// assert key is found
				assert.equal('asdf2', r);
				n++;
				// delete key
				mc.delete('set2', function() {
					mc.get('set2', function(err, r) {
						// assert key is null
						assert.equal(null, r);
						n++;
					});
				});
			});
		}, 0);
		
		beforeExit(function() {
			assert.equal(3, n);
		});
	};

    // test utf8 handling
    exports['utf8'] = function(beforeExit) {
        mc.set('key1', 'привет', function() {
            mc.get('key1', function(err, r) {
                assert.equal('привет', r);
            });
        });
    };


	// test connecting and disconnecting
	exports['con disco'] = function(beforeExit) {

		var n = 0;

		var mc2 = new memcache.Client(port);
		mc2.on('connect', function(){
			n++;
			mc2.close();
		});
		mc2.on('close', function(){
			n++;
		});

		mc2.connect();

		beforeExit(function() {
			assert.equal(2, n);
		});
	};

	// increment / decrement
	exports['inc dec'] = function(beforeExit){

		var n = 0;

		mc.set('inc_bad', 'HELLO', function(err, response){
			assert.equal(response, 'STORED');
			n++;
			mc.increment('inc_bad', 2, function(err, ok){
				n++;
				assert.match(err, /^CLIENT_ERROR/);
				assert.equal(ok, null);
			});
			mc.decrement('inc_bad', 3, function(err, ok){
				n++;
				assert.match(err, /^CLIENT_ERROR/);
				assert.equal(ok, null);
			});
			mc.increment('inc_bad', null, function(err, ok){
				n++;
				assert.match(err, /^CLIENT_ERROR/);
				assert.equal(ok, null);
			});
			mc.decrement('inc_bad', null, function(err, ok){
				n++;
				assert.match(err, /^CLIENT_ERROR/);
				assert.equal(ok, null);
			});
		});

		mc.set('inc_good', '5', function(err, response){
			assert.equal(response, 'STORED');
			n++;
			mc.increment('inc_good', 2, function(err, response){
				n++;
				assert.equal(response, 7);
				mc.increment('inc_good', function(err, response){
					n++;
					assert.equal(response, 8);
					mc.decrement('inc_good', function(err, response){
						n++;
						assert.equal(response, 7);
						mc.decrement('inc_good', 4, function(err, response){
							n++;
							assert.equal(response, 3);
						});
					});
				});
			});
		});

		beforeExit(function(){
			assert.equal(10, n);
		});

	};

	exports['version'] = function(beforeExit){
		var n = 0;

		mc.version(function(error, success){
			n++;
			assert.equal(error, null);
			assert.length(success, 5);
		});

		beforeExit(function(){
			assert.equal(1, n);
		});
	};

	exports['stats'] = function(beforeExit){
		var n = 0;

		mc.stats(function(error, success){
			n++;
			assert.ok(success.pid, "server has a pid");
		});

		mc.stats('settings', function(error, success){
			n++;
			assert.ok(success.maxconns);
		});

		mc.stats('items', function(error, success){ n++; assert.ok(num_keys(success)); });
		mc.stats('sizes', function(error, success){ n++; assert.ok(num_keys(success)); });
		mc.stats('slabs', function(error, success){ n++; assert.ok(num_keys(success)); });

		mc.stats('notreal', function(error, success){
			n++;
			assert.equal(error, 'ERROR');
		});

		beforeExit(function(){
			assert.equal(6, n);
		});
	};

});

function num_keys(a){
	var i=0;
	for (var k in a) i++;
	return i;
}