82 lines
2.5 KiB
JavaScript
82 lines
2.5 KiB
JavaScript
|
var assert = require('assert'),
|
||
|
|
_ = require('underscore'),
|
||
|
|
HealthCheck = require('../../app/monitoring/health_check');
|
||
|
|
|
||
|
|
var metadataBackend = {};
|
||
|
|
|
||
|
|
function PSQL(dbParams) {
|
||
|
|
this.params = dbParams;
|
||
|
|
}
|
||
|
|
|
||
|
|
var healthCheck = new HealthCheck(metadataBackend, PSQL);
|
||
|
|
|
||
|
|
suite('health checks', function() {
|
||
|
|
|
||
|
|
beforeEach(function(done) {
|
||
|
|
mockGetAllUserDBParams(function(username, callback) {
|
||
|
|
callback(null, {user: 'fake', dbname: 'fake'});
|
||
|
|
});
|
||
|
|
|
||
|
|
mockQuery(function(query, callback) {
|
||
|
|
callback(null, {rows: [{},{},{}]});
|
||
|
|
});
|
||
|
|
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
|
||
|
|
test('happy case, everything goes OK', function(done) {
|
||
|
|
healthCheck.check('fake', 'select 1::text', function(err, result) {
|
||
|
|
assert.ok(result.redis.ok);
|
||
|
|
assert.ok(result.redis.elapsed >= 0);
|
||
|
|
assert.equal(result.redis.count, 2);
|
||
|
|
|
||
|
|
assert.ok(result.postgresql.ok);
|
||
|
|
assert.ok(result.postgresql.elapsed >= 0);
|
||
|
|
assert.ok(result.postgresql.count, 3);
|
||
|
|
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
test('error in metadataBackend reports as false and does not report postgresql except ok=false', function(done) {
|
||
|
|
mockGetAllUserDBParams(function() {
|
||
|
|
throw "Error";
|
||
|
|
});
|
||
|
|
healthCheck.check('fake', 'select 1::text', function(err, result) {
|
||
|
|
assert.equal(result.redis.ok, false);
|
||
|
|
assert.ok(result.redis.elapsed >= 0);
|
||
|
|
assert.ok(_.isUndefined(result.redis.count));
|
||
|
|
|
||
|
|
assert.equal(result.postgresql.ok, false);
|
||
|
|
assert.ok(_.isUndefined(result.postgresql.elapsed));
|
||
|
|
assert.ok(_.isUndefined(result.postgresql.count));
|
||
|
|
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
test('error in metadataBackend reports as false and does not report postgresql except ok=false', function(done) {
|
||
|
|
mockQuery(function() {
|
||
|
|
throw "Error";
|
||
|
|
});
|
||
|
|
healthCheck.check('fake', 'select 1::text', function(err, result) {
|
||
|
|
assert.ok(result.redis.ok);
|
||
|
|
assert.ok(result.redis.elapsed >= 0);
|
||
|
|
assert.equal(result.redis.count, 2);
|
||
|
|
|
||
|
|
assert.equal(result.postgresql.ok, false);
|
||
|
|
assert.ok(_.isUndefined(result.postgresql.elapsed));
|
||
|
|
assert.ok(_.isUndefined(result.postgresql.count));
|
||
|
|
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
function mockGetAllUserDBParams(func) {
|
||
|
|
metadataBackend.getAllUserDBParams = func;
|
||
|
|
}
|
||
|
|
|
||
|
|
function mockQuery(func) {
|
||
|
|
PSQL.prototype.query = func;
|
||
|
|
}
|
||
|
|
});
|