mirror of
https://github.com/Unleash/unleash.git
synced 2024-10-18 20:09:08 +02:00
ff7be7696c
Co-authored-by: Ivar Conradi Østhus <ivarconr@gmail.com>
97 lines
2.6 KiB
TypeScript
97 lines
2.6 KiB
TypeScript
import EventEmitter from 'events';
|
|
|
|
import { ClientMetricsStore } from './client-metrics-store';
|
|
import getLogger from '../../test/fixtures/no-logger';
|
|
|
|
function getMockDb() {
|
|
const list = [
|
|
{ id: 4, metrics: { appName: 'test' } },
|
|
{ id: 3, metrics: { appName: 'test' } },
|
|
{ id: 2, metrics: { appName: 'test' } },
|
|
];
|
|
return {
|
|
getMetricsLastHour() {
|
|
return Promise.resolve([{ id: 1, metrics: { appName: 'test' } }]);
|
|
},
|
|
|
|
getNewMetrics() {
|
|
return Promise.resolve([list.pop() || { id: 0 }]);
|
|
},
|
|
destroy() {
|
|
// noop
|
|
},
|
|
};
|
|
}
|
|
|
|
test('should call database on startup', (done) => {
|
|
jest.useFakeTimers('modern');
|
|
const mock = getMockDb();
|
|
const ee = new EventEmitter();
|
|
const store = new ClientMetricsStore(mock as any, ee, getLogger);
|
|
|
|
jest.runAllTicks();
|
|
|
|
expect.assertions(2);
|
|
|
|
store.on('metrics', (metrics) => {
|
|
expect(store.highestIdSeen).toBe(1);
|
|
expect(metrics.appName).toBe('test');
|
|
store.destroy();
|
|
|
|
done();
|
|
});
|
|
});
|
|
|
|
test('should start poller even if initial database fetch fails', (done) => {
|
|
jest.useFakeTimers('modern');
|
|
getLogger.setMuteError(true);
|
|
const mock = getMockDb();
|
|
mock.getMetricsLastHour = () => Promise.reject(new Error('oops'));
|
|
const ee = new EventEmitter();
|
|
const store = new ClientMetricsStore(mock as any, ee, getLogger, 100);
|
|
jest.runAllTicks();
|
|
|
|
const metrics = [];
|
|
store.on('metrics', (m) => metrics.push(m));
|
|
|
|
store.on('ready', () => {
|
|
jest.useFakeTimers('modern');
|
|
expect(metrics).toHaveLength(0);
|
|
jest.advanceTimersByTime(300);
|
|
jest.useRealTimers();
|
|
process.nextTick(() => {
|
|
expect(metrics).toHaveLength(3);
|
|
expect(store.highestIdSeen).toBe(4);
|
|
store.destroy();
|
|
done();
|
|
});
|
|
});
|
|
getLogger.setMuteError(false);
|
|
});
|
|
|
|
test('should poll for updates', (done) => {
|
|
jest.useFakeTimers('modern');
|
|
const mock = getMockDb();
|
|
const ee = new EventEmitter();
|
|
const store = new ClientMetricsStore(mock as any, ee, getLogger, 100);
|
|
jest.runAllTicks();
|
|
|
|
const metrics = [];
|
|
store.on('metrics', (m) => metrics.push(m));
|
|
|
|
expect(metrics).toHaveLength(0);
|
|
|
|
store.on('ready', () => {
|
|
jest.useFakeTimers('modern');
|
|
expect(metrics).toHaveLength(1);
|
|
jest.advanceTimersByTime(300);
|
|
jest.useRealTimers();
|
|
process.nextTick(() => {
|
|
expect(metrics).toHaveLength(4);
|
|
expect(store.highestIdSeen).toBe(4);
|
|
store.destroy();
|
|
done();
|
|
});
|
|
});
|
|
});
|