105 lines
2.9 KiB
TypeScript
105 lines
2.9 KiB
TypeScript
import test from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import fs from 'node:fs/promises';
|
|
import os from 'node:os';
|
|
import path from 'node:path';
|
|
|
|
import {
|
|
registerAgent,
|
|
extendActivityLease,
|
|
deriveLiveness,
|
|
} from '../../src/lib/agent-registry';
|
|
|
|
async function withTempUserProfile(run: () => Promise<void>): Promise<void> {
|
|
const previous = process.env.USERPROFILE;
|
|
const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'beadboard-agent-liveness-'));
|
|
process.env.USERPROFILE = tempDir;
|
|
|
|
try {
|
|
await run();
|
|
} finally {
|
|
if (previous === undefined) {
|
|
delete process.env.USERPROFILE;
|
|
} else {
|
|
process.env.USERPROFILE = previous;
|
|
}
|
|
|
|
await fs.rm(tempDir, { recursive: true, force: true });
|
|
}
|
|
}
|
|
|
|
test('extendActivityLease emits heartbeat and returns null data (side effect only)', async () => {
|
|
await withTempUserProfile(async () => {
|
|
const start = '2026-02-14T10:00:00.000Z';
|
|
|
|
await registerAgent(
|
|
{ name: 'active-agent', role: 'infra' },
|
|
{ now: () => start }
|
|
);
|
|
|
|
const result = await extendActivityLease(
|
|
{ agent: 'active-agent' },
|
|
{ now: () => start }
|
|
);
|
|
|
|
assert.equal(result.ok, true);
|
|
assert.equal(result.data, null, 'extendActivityLease returns null data - heartbeat is side effect');
|
|
});
|
|
});
|
|
|
|
test('deriveLiveness follows threshold rules (15m/30m default)', () => {
|
|
const now = new Date('2026-02-14T12:00:00Z');
|
|
|
|
// Active: 14 mins ago
|
|
assert.equal(
|
|
deriveLiveness('2026-02-14T11:46:00Z', now),
|
|
'active'
|
|
);
|
|
|
|
// Stale: Exactly 15 mins ago
|
|
assert.equal(
|
|
deriveLiveness('2026-02-14T11:45:00Z', now),
|
|
'stale'
|
|
);
|
|
|
|
// Stale: 29 mins ago
|
|
assert.equal(
|
|
deriveLiveness('2026-02-14T11:31:00Z', now),
|
|
'stale'
|
|
);
|
|
|
|
// Evicted: Exactly 30 mins ago
|
|
assert.equal(
|
|
deriveLiveness('2026-02-14T11:30:00Z', now),
|
|
'evicted'
|
|
);
|
|
|
|
// Evicted: 1 hour ago
|
|
// Note: Since we added Idle at 60m, let's test 59m for Evicted and 60m for Idle
|
|
assert.equal(
|
|
deriveLiveness('2026-02-14T11:01:00Z', now),
|
|
'evicted'
|
|
);
|
|
|
|
// Idle: Exactly 60 mins ago
|
|
assert.equal(
|
|
deriveLiveness('2026-02-14T11:00:00Z', now),
|
|
'idle'
|
|
);
|
|
|
|
// Idle: 2 hours ago
|
|
assert.equal(
|
|
deriveLiveness('2026-02-14T10:00:00Z', now),
|
|
'idle'
|
|
);
|
|
});
|
|
|
|
test('deriveLiveness respects custom staleMinutes', () => {
|
|
const now = new Date('2026-02-14T12:00:00Z');
|
|
const customThreshold = 5; // 5m stale, 10m evicted
|
|
|
|
assert.equal(deriveLiveness('2026-02-14T11:56:00Z', now, customThreshold), 'active');
|
|
assert.equal(deriveLiveness('2026-02-14T11:55:00Z', now, customThreshold), 'stale');
|
|
assert.equal(deriveLiveness('2026-02-14T11:51:00Z', now, customThreshold), 'stale');
|
|
assert.equal(deriveLiveness('2026-02-14T11:50:00Z', now, customThreshold), 'evicted');
|
|
});
|