Files
EJClaw/src/restart-context.test.ts
Eyejoker 77c2d031bc Recover only active groups after restart
Limit restart interruption snapshots, announcements, and recovery queueing to groups that were actively processing. Waiting groups remain compatible in stored context but no longer receive restart recovery signals.
2026-05-21 04:54:00 +09:00

122 lines
3.1 KiB
TypeScript

import { describe, expect, it } from 'vitest';
import {
buildRestartAnnouncement,
getRecoverableInterruptedGroups,
getInterruptedRecoveryCandidates,
type RestartContext,
} from './restart-context.js';
import type { RegisteredGroup } from './types.js';
function makeGroup(folder: string): RegisteredGroup {
return {
name: folder,
folder,
trigger: '@Andy',
added_at: new Date().toISOString(),
requiresTrigger: false,
};
}
describe('restart recovery context', () => {
it('returns only registered processing groups and deduplicates by chatJid', () => {
const roomBindings: Record<string, RegisteredGroup> = {
'dc:1': makeGroup('group-one'),
'dc:2': makeGroup('group-two'),
};
const context: RestartContext = {
chatJid: 'dc:main',
summary: 'restart',
verify: [],
writtenAt: new Date().toISOString(),
interruptedGroups: [
{
chatJid: 'dc:1',
groupName: 'one',
status: 'processing',
elapsedMs: 1000,
pendingMessages: true,
pendingTasks: 0,
},
{
chatJid: 'dc:2',
groupName: 'two-waiting',
status: 'waiting',
elapsedMs: null,
pendingMessages: false,
pendingTasks: 1,
},
{
chatJid: 'dc:2',
groupName: 'two-idle',
status: 'idle',
elapsedMs: null,
pendingMessages: false,
pendingTasks: 0,
},
{
chatJid: 'dc:3',
groupName: 'missing',
status: 'processing',
elapsedMs: 500,
pendingMessages: true,
pendingTasks: 0,
},
{
chatJid: 'dc:1',
groupName: 'one-duplicate',
status: 'processing',
elapsedMs: 2000,
pendingMessages: false,
pendingTasks: 1,
},
],
};
expect(getInterruptedRecoveryCandidates(context, roomBindings)).toEqual([
{
chatJid: 'dc:1',
groupFolder: 'group-one',
status: 'processing',
pendingMessages: true,
pendingTasks: 0,
},
]);
});
it('returns empty when there is no explicit restart context', () => {
expect(getInterruptedRecoveryCandidates(null, {})).toEqual([]);
});
it('uses only processing groups for restart announcements', () => {
const context: RestartContext = {
chatJid: 'dc:main',
summary: 'restart',
verify: [],
writtenAt: '2026-05-21T00:00:00.000Z',
interruptedGroups: [
{
chatJid: 'dc:processing',
groupName: 'processing',
status: 'processing',
elapsedMs: 1000,
pendingMessages: false,
pendingTasks: 0,
},
{
chatJid: 'dc:waiting',
groupName: 'waiting',
status: 'waiting',
elapsedMs: null,
pendingMessages: true,
pendingTasks: 1,
},
],
};
expect(getRecoverableInterruptedGroups(context)).toHaveLength(1);
expect(buildRestartAnnouncement(context)).toContain('중단 작업 감지: 1개');
});
});