joyent-portal/packages/portal-api/test/data/unit.js

130 lines
4.2 KiB
JavaScript

'use strict';
const Http = require('http');
const { describe, it, expect } = exports.lab = require('lab').script();
const PortalData = require('../../lib/data');
describe('getMetrics()', () => {
it('queries prometheus for the instance metrics', (done) => {
const server = Http.createServer((req, res) => {
const payload = JSON.stringify({ 'status': 'success', 'data': { 'resultType': 'matrix', 'result': [{ 'metric': { '__name__': 'mem_agg_usage', 'instance': 'joyentportal_compose-api_1', 'job': 'triton' }, 'values': [[1502896217.371, '60518400'], [1502899817.371, '60641280'], [1502903417.371, '60575744']] } ] } });
res.writeHead(200, { 'Content-Type': 'application/json' });
res.end(payload);
});
server.listen(9090, (err) => {
expect(err).to.not.exist();
const options = {
deploymentGroupId: 'deploymentGroup',
names: ['AVG_MEM_BYTES'],
instances: ['joyentportal_compose-api_1'],
start: new Date(Date.now() - 10000).toISOString(),
end: new Date(Date.now()).toISOString()
};
const dataThis = {
getServices: ({ deploymentGroupId, name }, next) => {
const instance = {
machine_id: 'machineId'
};
const service = {
instances: () => { return Promise.resolve([instance]); }
};
next(null, [service]);
},
_triton: {
getInstance: (id, next) => {
const tritonInstance = {
primaryIp: 'localhost'
};
next(null, tritonInstance);
}
}
};
PortalData.prototype.getMetrics.call(dataThis, options, (err, metrics) => {
expect(err).to.not.be.an.error();
expect(metrics.length).to.equal(1);
expect(metrics[0].name).to.equal('AVG_MEM_BYTES');
server.close(() => { return done(); });
});
});
});
});
describe('getInstance()', () => {
it('retrieves an instance that can also be used to query for metrics', (done) => {
const server = Http.createServer((req, res) => {
const payload = JSON.stringify({ 'status': 'success', 'data': { 'resultType': 'matrix', 'result': [{ 'metric': { '__name__': 'mem_agg_usage', 'instance': 'joyentportal_compose-api_1', 'job': 'triton' }, 'values': [[1502896217.371, '60518400'], [1502899817.371, '60641280'], [1502903417.371, '60575744']] }] } });
res.writeHead(200, { 'Content-Type': 'application/json' });
res.end(payload);
});
server.listen(9090, (err) => {
expect(err).to.not.exist();
const FakePortalData = function () {
this._db = {
instances: {
single: (options, next) => {
expect(options.id).to.exist();
const instance = {
id: options.id,
deployment_group_id: 'deploymentGroup',
service_id: 'service',
machine_id: 'joyentportal_compose-api_1'
};
next(null, instance);
}
}
};
this._triton = {
getInstance: (id, next) => {
const tritonInstance = {
primaryIp: 'localhost'
};
next(null, tritonInstance);
}
};
};
FakePortalData.prototype = PortalData.prototype;
FakePortalData.prototype.getServices = ({ deploymentGroupId, name }, next) => {
const instance = {
machine_id: 'machineId'
};
const service = {
instances: () => { return Promise.resolve([instance]); }
};
next(null, [service]);
};
const fakePortalData = new FakePortalData();
fakePortalData.getInstance({ id: 'joyentportal_compose-api_1' }, (err, instance) => {
expect(err).to.not.be.an.error();
expect(instance).to.exist();
instance.metrics({ names: ['AVG_MEM_BYTES'], start: new Date(Date.now() - 10000).toISOString(), end: new Date(Date.now()).toISOString() }).catch((err) => {
expect(err).to.not.exist();
}).then((metrics) => {
expect(metrics.length).to.equal(1);
expect(metrics[0].name).to.equal('AVG_MEM_BYTES');
server.close(() => { return done(); });
});
});
});
});
});