71 lines
2.5 KiB
JavaScript
71 lines
2.5 KiB
JavaScript
|
import assert from 'node:assert/strict';
|
||
|
|
import { StatsCounters, StatsReporter } from '../src/stats/statsManager.js';
|
||
|
|
import { HeartbeatProcessor } from '../src/processor/heartbeatProcessor.js';
|
||
|
|
|
||
|
|
describe('StatsCounters', () => {
|
||
|
|
it('snapshots and resets minute counters atomically', () => {
|
||
|
|
const stats = new StatsCounters();
|
||
|
|
stats.incDbWritten(3);
|
||
|
|
stats.incFiltered(2);
|
||
|
|
stats.incKafkaPulled(5);
|
||
|
|
|
||
|
|
const first = stats.snapshotAndResetMinute();
|
||
|
|
assert.equal(first.dbWritten, 3n);
|
||
|
|
assert.equal(first.filtered, 2n);
|
||
|
|
assert.equal(first.kafkaPulled, 5n);
|
||
|
|
|
||
|
|
const second = stats.snapshotAndResetMinute();
|
||
|
|
assert.equal(second.dbWritten, 0n);
|
||
|
|
assert.equal(second.filtered, 0n);
|
||
|
|
assert.equal(second.kafkaPulled, 0n);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('StatsReporter', () => {
|
||
|
|
it('writes three [STATS] info logs to redis console', () => {
|
||
|
|
const stats = new StatsCounters();
|
||
|
|
stats.incDbWritten(7);
|
||
|
|
stats.incFiltered(8);
|
||
|
|
stats.incKafkaPulled(9);
|
||
|
|
|
||
|
|
const calls = { push: [] };
|
||
|
|
const redis = {
|
||
|
|
isEnabled: () => true,
|
||
|
|
pushConsoleLog: ({ level, message, metadata }) => {
|
||
|
|
calls.push.push({ level, message, metadata });
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
const reporter = new StatsReporter({ redis, stats });
|
||
|
|
reporter.flushOnce();
|
||
|
|
|
||
|
|
assert.equal(calls.push.length, 3);
|
||
|
|
assert.equal(calls.push[0].level, 'info');
|
||
|
|
assert.equal(calls.push[1].level, 'info');
|
||
|
|
assert.equal(calls.push[2].level, 'info');
|
||
|
|
assert.match(calls.push[0].message, /^\[STATS\] \d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3} 数据库写入量: 7条$/);
|
||
|
|
assert.match(calls.push[1].message, /^\[STATS\] \d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3} 数据过滤量: 8条$/);
|
||
|
|
assert.match(calls.push[2].message, /^\[STATS\] \d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3} Kafka拉取量: 9条$/);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('HeartbeatProcessor db write error logging', () => {
|
||
|
|
it('emits [ERROR] warn log with raw data', () => {
|
||
|
|
const calls = { warn: [] };
|
||
|
|
const redis = {
|
||
|
|
isEnabled: () => true,
|
||
|
|
pushConsoleLog: ({ level, message }) => {
|
||
|
|
if (level === 'warn') calls.warn.push(message);
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
const processor = new HeartbeatProcessor({ batchSize: 1, batchTimeout: 10 }, {}, { redis });
|
||
|
|
processor._emitDbWriteError(new Error('boom'), [{ a: 1 }]);
|
||
|
|
|
||
|
|
assert.equal(calls.warn.length >= 1, true);
|
||
|
|
assert.match(calls.warn[0], /^\[ERROR\] \d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3} db_write_failed: /);
|
||
|
|
assert.match(calls.warn[0], /"errorId":"db_write_failed"/);
|
||
|
|
assert.match(calls.warn[0], /"rawData":\{"a":1\}/);
|
||
|
|
});
|
||
|
|
});
|