Files
EJClaw/src/restart-context.test.ts
2026-05-27 10:53:31 +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개');
});
});