mirror of
https://github.com/Shik3i/KoalaSync.git
synced 2026-07-26 12:08:15 +00:00
Timo
89 lines
4.0 KiB
JavaScript
89 lines
4.0 KiB
JavaScript
import assert from 'node:assert/strict';
|
|
import {
|
|
buildHealthPayload,
|
|
checkCooldown,
|
|
getCachedPayload,
|
|
isAdminMetricsAuthorized,
|
|
isAdminMetricsTokenStrong
|
|
} from '../server/ops.js';
|
|
|
|
const missingAuth = isAdminMetricsAuthorized(undefined, 'secret-token');
|
|
assert.equal(missingAuth, false, 'missing Authorization header must not authorize metrics');
|
|
|
|
const wrongAuth = isAdminMetricsAuthorized('Bearer wrong-token', 'secret-token');
|
|
assert.equal(wrongAuth, false, 'wrong bearer token must not authorize metrics');
|
|
|
|
const correctAuth = isAdminMetricsAuthorized('Bearer secret-token', 'secret-token');
|
|
assert.equal(correctAuth, true, 'correct bearer token should authorize metrics');
|
|
|
|
const disabledAuth = isAdminMetricsAuthorized('Bearer secret-token', '');
|
|
assert.equal(disabledAuth, false, 'empty admin token disables admin metrics');
|
|
|
|
assert.equal(isAdminMetricsTokenStrong(''), true, 'empty admin token is allowed because metrics stay disabled');
|
|
assert.equal(isAdminMetricsTokenStrong('short-token'), false, 'short admin token should be reported as weak');
|
|
assert.equal(
|
|
isAdminMetricsTokenStrong('a'.repeat(32)),
|
|
true,
|
|
'admin token with at least 32 characters should be considered strong'
|
|
);
|
|
|
|
const cooldowns = new Map();
|
|
assert.equal(checkCooldown(cooldowns, 'socket-1', 10_000, 100_000), true, 'first cooldown check passes');
|
|
assert.equal(checkCooldown(cooldowns, 'socket-1', 10_000, 105_000), false, 'second cooldown check inside window fails');
|
|
assert.equal(checkCooldown(cooldowns, 'socket-1', 10_000, 110_000), true, 'cooldown check after window passes');
|
|
|
|
const cache = new Map();
|
|
let buildCalls = 0;
|
|
const firstCached = getCachedPayload(cache, 'basic-health', 60_000, () => ({ value: ++buildCalls }), 1_000);
|
|
const secondCached = getCachedPayload(cache, 'basic-health', 60_000, () => ({ value: ++buildCalls }), 30_000);
|
|
const expiredCached = getCachedPayload(cache, 'basic-health', 60_000, () => ({ value: ++buildCalls }), 61_001);
|
|
assert.deepEqual(firstCached, { value: 1 }, 'cache should return the builder payload on first request');
|
|
assert.strictEqual(secondCached, firstCached, 'cache should reuse payloads inside the ttl');
|
|
assert.deepEqual(expiredCached, { value: 2 }, 'cache should rebuild payloads after ttl expiry');
|
|
|
|
const roomA = { peers: new Set(['a', 'b']), activeLobby: null };
|
|
const roomB = { peers: new Set(['c', 'd', 'e']), activeLobby: { expectedTitle: 'Episode 2' } };
|
|
const rooms = new Map([['room-a', roomA], ['room-b', roomB]]);
|
|
|
|
const basicHealth = buildHealthPayload({
|
|
rooms,
|
|
connections: 5,
|
|
includeMetrics: false,
|
|
now: 1234,
|
|
uptime: 99,
|
|
memoryUsage: () => ({ rss: 10, heapUsed: 5, heapTotal: 8 }),
|
|
rateLimitSizes: { connections: 1, events: 2, health: 3, adminMetricsAuth: 4, authFailures: 5, roomList: 6 }
|
|
});
|
|
|
|
assert.deepEqual(
|
|
Object.keys(basicHealth).sort(),
|
|
['connections', 'rooms', 'status', 'timestamp', 'uptime'].sort(),
|
|
'basic health should not expose extended metrics'
|
|
);
|
|
|
|
const adminHealth = buildHealthPayload({
|
|
rooms,
|
|
connections: 5,
|
|
includeMetrics: true,
|
|
now: 1234,
|
|
uptime: 99,
|
|
memoryUsage: () => ({ rss: 10, heapUsed: 5, heapTotal: 8 }),
|
|
rateLimitSizes: { connections: 1, events: 2, health: 3, adminMetricsAuth: 4, authFailures: 5, roomList: 6 }
|
|
});
|
|
|
|
assert.equal(adminHealth.peers, 5, 'admin metrics should include aggregate peer count');
|
|
assert.equal(adminHealth.roomsWithLobby, 1, 'admin metrics should count active lobbies');
|
|
assert.equal(adminHealth.avgPeersPerRoom, 2.5, 'admin metrics should include average room size');
|
|
assert.equal(adminHealth.maxPeersInRoom, 3, 'admin metrics should include max room size');
|
|
assert.deepEqual(adminHealth.memory, { rss: 10, heapUsed: 5, heapTotal: 8 }, 'admin metrics should expose process memory');
|
|
assert.deepEqual(
|
|
adminHealth.rateLimits,
|
|
{
|
|
trackedClients: { connections: 1, events: 2, health: 3, adminMetricsAuth: 4, authFailures: 5, roomList: 6 },
|
|
denied: { connections: 0, events: 0, health: 0, adminMetricsAuth: 0, roomList: 0 }
|
|
},
|
|
'admin metrics should expose rate-limit tracking and denial counts'
|
|
);
|
|
|
|
console.log('server ops tests passed');
|