250 lines
8.3 KiB
TypeScript
250 lines
8.3 KiB
TypeScript
import * as fs from 'fs/promises';
|
|
import * as os from 'os';
|
|
import * as path from 'path';
|
|
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
|
|
import { setClaudeBasePathOverride } from '../../../../src/main/utils/pathDecoder';
|
|
import { TeamTaskActivityIntervalService } from '../../../../src/main/services/team/TeamTaskActivityIntervalService';
|
|
|
|
let tempDir = '';
|
|
|
|
async function writeTask(teamName: string, task: Record<string, unknown>): Promise<void> {
|
|
const taskDir = path.join(tempDir, 'tasks', teamName);
|
|
const taskId = String(task.id);
|
|
await fs.mkdir(taskDir, { recursive: true });
|
|
await fs.writeFile(path.join(taskDir, `${taskId}.json`), JSON.stringify(task, null, 2), 'utf8');
|
|
}
|
|
|
|
async function readTask(teamName: string, taskId: string): Promise<Record<string, unknown>> {
|
|
return JSON.parse(
|
|
await fs.readFile(path.join(tempDir, 'tasks', teamName, `${taskId}.json`), 'utf8')
|
|
) as Record<string, unknown>;
|
|
}
|
|
|
|
describe('TeamTaskActivityIntervalService', () => {
|
|
beforeEach(async () => {
|
|
tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'team-task-activity-'));
|
|
setClaudeBasePathOverride(tempDir);
|
|
});
|
|
|
|
afterEach(async () => {
|
|
vi.useRealTimers();
|
|
setClaudeBasePathOverride(null);
|
|
if (tempDir) {
|
|
await fs.rm(tempDir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
it('pauses all active work and review intervals for a team without changing task status', async () => {
|
|
await writeTask('alpha', {
|
|
id: 'task-1',
|
|
subject: 'Build',
|
|
owner: 'bob',
|
|
status: 'in_progress',
|
|
workIntervals: [{ startedAt: '2026-05-08T10:00:00.000Z' }],
|
|
reviewIntervals: [{ reviewer: 'alice', startedAt: '2026-05-08T10:05:00.000Z' }],
|
|
historyEvents: [],
|
|
});
|
|
|
|
const result = new TeamTaskActivityIntervalService().pauseActiveIntervalsForTeam(
|
|
'alpha',
|
|
'2026-05-08T10:10:00.000Z'
|
|
);
|
|
const task = await readTask('alpha', 'task-1');
|
|
|
|
expect(result.changedTasks).toBe(1);
|
|
expect(task.status).toBe('in_progress');
|
|
expect(task.workIntervals).toEqual([
|
|
{ startedAt: '2026-05-08T10:00:00.000Z', completedAt: '2026-05-08T10:10:00.000Z' },
|
|
]);
|
|
expect(task.reviewIntervals).toEqual([
|
|
{
|
|
reviewer: 'alice',
|
|
startedAt: '2026-05-08T10:05:00.000Z',
|
|
completedAt: '2026-05-08T10:10:00.000Z',
|
|
},
|
|
]);
|
|
});
|
|
|
|
it('pauses only the selected member work and review intervals', async () => {
|
|
await writeTask('alpha', {
|
|
id: 'bob-task',
|
|
subject: 'Bob',
|
|
owner: 'bob',
|
|
status: 'in_progress',
|
|
workIntervals: [{ startedAt: '2026-05-08T10:00:00.000Z' }],
|
|
reviewIntervals: [{ reviewer: 'alice', startedAt: '2026-05-08T10:01:00.000Z' }],
|
|
historyEvents: [],
|
|
});
|
|
await writeTask('alpha', {
|
|
id: 'tom-task',
|
|
subject: 'Tom',
|
|
owner: 'tom',
|
|
status: 'in_progress',
|
|
workIntervals: [{ startedAt: '2026-05-08T10:00:00.000Z' }],
|
|
reviewIntervals: [{ reviewer: 'bob', startedAt: '2026-05-08T10:02:00.000Z' }],
|
|
historyEvents: [],
|
|
});
|
|
|
|
const result = new TeamTaskActivityIntervalService().pauseActiveIntervalsForMember(
|
|
'alpha',
|
|
'bob',
|
|
'2026-05-08T10:05:00.000Z'
|
|
);
|
|
|
|
expect(result.changedTasks).toBe(2);
|
|
expect((await readTask('alpha', 'bob-task')).workIntervals).toEqual([
|
|
{ startedAt: '2026-05-08T10:00:00.000Z', completedAt: '2026-05-08T10:05:00.000Z' },
|
|
]);
|
|
expect((await readTask('alpha', 'bob-task')).reviewIntervals).toEqual([
|
|
{ reviewer: 'alice', startedAt: '2026-05-08T10:01:00.000Z' },
|
|
]);
|
|
expect((await readTask('alpha', 'tom-task')).workIntervals).toEqual([
|
|
{ startedAt: '2026-05-08T10:00:00.000Z' },
|
|
]);
|
|
expect((await readTask('alpha', 'tom-task')).reviewIntervals).toEqual([
|
|
{
|
|
reviewer: 'bob',
|
|
startedAt: '2026-05-08T10:02:00.000Z',
|
|
completedAt: '2026-05-08T10:05:00.000Z',
|
|
},
|
|
]);
|
|
});
|
|
|
|
it('resumes active work and current review intervals for the selected member', async () => {
|
|
await writeTask('alpha', {
|
|
id: 'task-1',
|
|
subject: 'Build',
|
|
owner: 'bob',
|
|
status: 'in_progress',
|
|
workIntervals: [
|
|
{ startedAt: '2026-05-08T10:00:00.000Z', completedAt: '2026-05-08T10:05:00.000Z' },
|
|
],
|
|
reviewIntervals: [
|
|
{
|
|
reviewer: 'bob',
|
|
startedAt: '2026-05-08T10:06:00.000Z',
|
|
completedAt: '2026-05-08T10:08:00.000Z',
|
|
},
|
|
],
|
|
historyEvents: [
|
|
{
|
|
id: 'event-review-started',
|
|
type: 'review_started',
|
|
timestamp: '2026-05-08T10:06:00.000Z',
|
|
actor: 'bob',
|
|
},
|
|
],
|
|
});
|
|
|
|
const result = new TeamTaskActivityIntervalService().resumeActiveIntervalsForMember(
|
|
'alpha',
|
|
'bob',
|
|
'2026-05-08T10:20:00.000Z'
|
|
);
|
|
const task = await readTask('alpha', 'task-1');
|
|
|
|
expect(result.changedTasks).toBe(1);
|
|
expect(task.workIntervals).toEqual([
|
|
{ startedAt: '2026-05-08T10:00:00.000Z', completedAt: '2026-05-08T10:05:00.000Z' },
|
|
{ startedAt: '2026-05-08T10:20:00.000Z' },
|
|
]);
|
|
expect(task.reviewIntervals).toEqual([
|
|
{
|
|
reviewer: 'bob',
|
|
startedAt: '2026-05-08T10:06:00.000Z',
|
|
completedAt: '2026-05-08T10:08:00.000Z',
|
|
},
|
|
{ reviewer: 'bob', startedAt: '2026-05-08T10:20:00.000Z' },
|
|
]);
|
|
});
|
|
|
|
it('repairs stale open intervals using last runtime evidence plus a small grace window', async () => {
|
|
vi.useFakeTimers();
|
|
vi.setSystemTime(new Date('2026-05-08T12:00:00.000Z'));
|
|
await writeTask('alpha', {
|
|
id: 'task-1',
|
|
subject: 'Build',
|
|
owner: 'bob',
|
|
status: 'in_progress',
|
|
workIntervals: [{ startedAt: '2026-05-08T10:00:00.000Z' }],
|
|
reviewIntervals: [{ reviewer: 'alice', startedAt: '2026-05-08T10:10:00.000Z' }],
|
|
historyEvents: [],
|
|
});
|
|
|
|
const result = new TeamTaskActivityIntervalService().repairStaleIntervalsAfterCrash('alpha', {
|
|
version: 2,
|
|
teamName: 'alpha',
|
|
updatedAt: '2026-05-08T10:31:00.000Z',
|
|
launchPhase: 'active',
|
|
expectedMembers: ['bob', 'alice'],
|
|
members: {
|
|
bob: {
|
|
name: 'bob',
|
|
launchState: 'confirmed_alive',
|
|
agentToolAccepted: true,
|
|
runtimeAlive: true,
|
|
bootstrapConfirmed: true,
|
|
hardFailure: false,
|
|
runtimeLastSeenAt: '2026-05-08T10:30:00.000Z',
|
|
lastEvaluatedAt: '2026-05-08T10:31:00.000Z',
|
|
},
|
|
alice: {
|
|
name: 'alice',
|
|
launchState: 'confirmed_alive',
|
|
agentToolAccepted: true,
|
|
runtimeAlive: true,
|
|
bootstrapConfirmed: true,
|
|
hardFailure: false,
|
|
lastHeartbeatAt: '2026-05-08T10:20:00.000Z',
|
|
lastEvaluatedAt: '2026-05-08T10:31:00.000Z',
|
|
},
|
|
},
|
|
summary: { confirmedCount: 2, pendingCount: 0, failedCount: 0, runtimeAlivePendingCount: 0 },
|
|
teamLaunchState: 'clean_success',
|
|
});
|
|
const task = await readTask('alpha', 'task-1');
|
|
|
|
expect(result.changedTasks).toBe(1);
|
|
expect(task.workIntervals).toEqual([
|
|
{ startedAt: '2026-05-08T10:00:00.000Z', completedAt: '2026-05-08T10:30:05.000Z' },
|
|
]);
|
|
expect(task.reviewIntervals).toEqual([
|
|
{
|
|
reviewer: 'alice',
|
|
startedAt: '2026-05-08T10:10:00.000Z',
|
|
completedAt: '2026-05-08T10:20:05.000Z',
|
|
},
|
|
]);
|
|
});
|
|
|
|
it('repairs stale open intervals near their start time when no runtime evidence exists', async () => {
|
|
vi.useFakeTimers();
|
|
vi.setSystemTime(new Date('2026-05-08T12:00:00.000Z'));
|
|
await writeTask('alpha', {
|
|
id: 'task-1',
|
|
subject: 'Build',
|
|
owner: 'bob',
|
|
status: 'in_progress',
|
|
workIntervals: [{ startedAt: '2026-05-08T10:00:00.000Z' }],
|
|
reviewIntervals: [{ reviewer: 'alice', startedAt: '2026-05-08T10:10:00.000Z' }],
|
|
historyEvents: [],
|
|
});
|
|
|
|
const result = new TeamTaskActivityIntervalService().repairStaleIntervalsAfterCrash('alpha');
|
|
const task = await readTask('alpha', 'task-1');
|
|
|
|
expect(result.changedTasks).toBe(1);
|
|
expect(task.workIntervals).toEqual([
|
|
{ startedAt: '2026-05-08T10:00:00.000Z', completedAt: '2026-05-08T10:00:05.000Z' },
|
|
]);
|
|
expect(task.reviewIntervals).toEqual([
|
|
{
|
|
reviewer: 'alice',
|
|
startedAt: '2026-05-08T10:10:00.000Z',
|
|
completedAt: '2026-05-08T10:10:05.000Z',
|
|
},
|
|
]);
|
|
});
|
|
});
|