Web_BLS_Heartbeat_Server/test/stats.test.js

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);
    stats.incDbWriteFailed(4);

    const first = stats.snapshotAndResetMinute();
    assert.equal(first.dbWritten, 3n);
    assert.equal(first.filtered, 2n);
    assert.equal(first.kafkaPulled, 5n);
    assert.equal(first.dbWriteFailed, 4n);

    const second = stats.snapshotAndResetMinute();
    assert.equal(second.dbWritten, 0n);
    assert.equal(second.filtered, 0n);
    assert.equal(second.kafkaPulled, 0n);
    assert.equal(second.dbWriteFailed, 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);
    stats.incDbWriteFailed(2);

    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, 4);
    assert.equal(calls.push[0].level, 'info');
    assert.equal(calls.push[1].level, 'info');
    assert.equal(calls.push[2].level, 'info');
    assert.equal(calls.push[3].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} 数据库写入失败量: 2条$/);
    assert.match(calls.push[2].message, /^\[STATS\] \d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3} 数据过滤量: 8条$/);
    assert.match(calls.push[3].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\}/);
  });
});