Merge pull request #111 from phj1081/codex/owner/ejclaw

Fix paired owner progress identity
This commit is contained in:
Eyejoker
2026-04-30 01:25:08 +09:00
committed by GitHub
4 changed files with 246 additions and 4 deletions

View File

@@ -0,0 +1,144 @@
import { beforeEach, describe, expect, it, vi } from 'vitest';
const runtimeMocks = vi.hoisted(() => ({
executeTurn: vi.fn(async () => ({
outputStatus: 'success' as const,
deliverySucceeded: true,
visiblePhase: 'progress' as const,
})),
}));
vi.mock('./message-runtime-turns.js', () => ({
createRunAgent: vi.fn(() => vi.fn()),
createExecuteTurn: vi.fn(() => runtimeMocks.executeTurn),
isDuplicateOfLastBotFinal: vi.fn(() => false),
labelPairedSenders: vi.fn((_channels, _chatJid, messages) => messages),
}));
import {
_initTestDatabase,
_setRegisteredGroupForTests,
getLatestOpenPairedTaskForChat,
storeChatMetadata,
storeMessage,
} from './db.js';
import { createMessageRuntime } from './message-runtime.js';
import {
clearGlobalFailover,
refreshChannelOwnerCache,
} from './service-routing.js';
import type { Channel, RegisteredGroup } from './types.js';
const chatJid = 'group@test';
const timestamp = '2026-04-30T00:30:00.000Z';
function makeGroup(): RegisteredGroup {
return {
name: 'Test Group',
folder: 'test-group',
trigger: '@Andy',
added_at: timestamp,
requiresTrigger: false,
agentType: 'codex',
workDir: process.cwd(),
};
}
function makeChannel(name: string, ownsJid: boolean): Channel {
return {
name,
connect: vi.fn(),
sendMessage: vi.fn(),
isConnected: vi.fn(() => true),
ownsJid: vi.fn((jid: string) => ownsJid && jid === chatJid),
disconnect: vi.fn(),
setTyping: vi.fn(),
} as unknown as Channel;
}
describe('message-runtime progress entry path', () => {
beforeEach(() => {
_initTestDatabase();
refreshChannelOwnerCache(true);
clearGlobalFailover();
runtimeMocks.executeTurn.mockClear();
runtimeMocks.executeTurn.mockResolvedValue({
outputStatus: 'success',
deliverySucceeded: true,
visiblePhase: 'progress',
});
});
it('passes pairedTurnIdentity through the production owner-turn entry path when no task is open', async () => {
const group = makeGroup();
const ownerChannel = makeChannel('discord', true);
const reviewerChannel = makeChannel('discord-review', false);
const lastAgentTimestamps: Record<string, string> = {};
const saveState = vi.fn();
_setRegisteredGroupForTests(chatJid, {
...group,
agentType: 'claude-code',
});
_setRegisteredGroupForTests(chatJid, group);
refreshChannelOwnerCache(true);
storeChatMetadata(chatJid, timestamp, 'Test Group');
storeMessage({
id: 'human-first-owner-turn',
chat_jid: chatJid,
sender: 'user@test',
sender_name: 'User',
content: '디스코드에는 progress가 보이는데 사이트에는 안 보여요',
timestamp,
is_bot_message: false,
is_from_me: false,
});
const runtime = createMessageRuntime({
assistantName: 'Andy',
idleTimeout: 1_000,
pollInterval: 1_000,
timezone: 'UTC',
triggerPattern: /^@Andy\b/i,
channels: [ownerChannel, reviewerChannel],
queue: {
enqueueMessageCheck: vi.fn(),
registerProcess: vi.fn(),
closeStdin: vi.fn(),
notifyIdle: vi.fn(),
} as any,
getRoomBindings: () => ({ [chatJid]: group }),
getSessions: () => ({}),
getLastTimestamp: () => '',
setLastTimestamp: vi.fn(),
getLastAgentTimestamps: () => lastAgentTimestamps,
saveState,
persistSession: vi.fn(),
clearSession: vi.fn(),
});
const result = await runtime.processGroupMessages(chatJid, {
runId: 'run-first-owner-entry',
reason: 'messages',
});
const freshTask = getLatestOpenPairedTaskForChat(chatJid);
expect(result).toBe(true);
expect(freshTask).toBeDefined();
expect(runtimeMocks.executeTurn).toHaveBeenCalledWith(
expect.objectContaining({
deliveryRole: 'owner',
forcedRole: 'owner',
pairedTurnIdentity: {
turnId: `${freshTask!.id}:${freshTask!.updated_at}:owner-turn`,
taskId: freshTask!.id,
taskUpdatedAt: freshTask!.updated_at,
intentKind: 'owner-turn',
role: 'owner',
},
}),
);
expect(lastAgentTimestamps).toEqual({ [chatJid]: '1' });
expect(saveState).toHaveBeenCalled();
});
});

View File

@@ -0,0 +1,98 @@
import { beforeEach, describe, expect, it, vi } from 'vitest';
import { _initTestDatabase, getLatestOpenPairedTaskForChat } from './db.js';
import { runQueuedGroupTurn } from './message-runtime-queue.js';
import { resetPairedFollowUpScheduleState } from './paired-follow-up-scheduler.js';
import type { Channel, RegisteredGroup } from './types.js';
function makeGroup(): RegisteredGroup {
return {
name: 'Test Group',
folder: 'test-group',
trigger: '@Andy',
added_at: '2026-03-30T00:00:00.000Z',
requiresTrigger: false,
agentType: 'codex',
workDir: '/repo',
};
}
function makeChannel(): Channel {
return {
name: 'discord-review',
connect: vi.fn(),
sendMessage: vi.fn(),
isConnected: vi.fn(() => true),
ownsJid: vi.fn(() => true),
disconnect: vi.fn(),
} as unknown as Channel;
}
describe('message-runtime queue progress persistence', () => {
beforeEach(() => {
_initTestDatabase();
resetPairedFollowUpScheduleState();
});
it('creates a paired owner task before the first queued turn so progress has a turn identity', async () => {
const executeTurn = vi.fn(async () => ({
outputStatus: 'success' as const,
deliverySucceeded: true,
visiblePhase: 'progress',
}));
const lastAgentTimestamps: Record<string, string> = {};
const saveState = vi.fn();
const outcome = await runQueuedGroupTurn({
chatJid: 'group@test',
group: makeGroup(),
runId: 'run-new-paired-owner',
log: { info: vi.fn(), warn: vi.fn(), error: vi.fn() } as any,
timezone: 'UTC',
missedMessages: [
{
id: 'human-new-task-1',
chat_jid: 'group@test',
sender: 'user@test',
sender_name: 'User',
content: '새 작업입니다',
timestamp: '2026-03-30T00:00:03.000Z',
seq: 48,
is_bot_message: false,
},
],
task: null,
roleToChannel: {
owner: null,
reviewer: makeChannel(),
arbiter: null,
},
ownerChannel: makeChannel(),
lastAgentTimestamps,
saveState,
executeTurn,
getFixedRoleChannelName: () => 'discord-review',
labelPairedSenders: (_chatJid, messages) => messages,
formatMessages: () => 'formatted prompt',
});
const freshTask = getLatestOpenPairedTaskForChat('group@test');
expect(outcome).toBe(true);
expect(freshTask).toBeDefined();
expect(executeTurn).toHaveBeenCalledWith(
expect.objectContaining({
deliveryRole: 'owner',
forcedRole: 'owner',
pairedTurnIdentity: {
turnId: `${freshTask!.id}:${freshTask!.updated_at}:owner-turn`,
taskId: freshTask!.id,
taskUpdatedAt: freshTask!.updated_at,
intentKind: 'owner-turn',
role: 'owner',
},
}),
);
expect(lastAgentTimestamps).toEqual({ 'group@test': '48' });
expect(saveState).toHaveBeenCalled();
});
});

View File

@@ -166,11 +166,11 @@ export async function runQueuedGroupTurn(args: {
) )
: !isBotOnlyPairedRoomTurn(chatJid, missedMessages); : !isBotOnlyPairedRoomTurn(chatJid, missedMessages);
let fallbackMessages = missedMessages; let fallbackMessages = missedMessages;
if (currentTask && hasHumanMsg) { if (currentTask !== undefined && hasHumanMsg) {
const resolvedTask = resolveOwnerTaskForHumanMessage({ const resolvedTask = resolveOwnerTaskForHumanMessage({
group, group,
chatJid, chatJid,
existingTask: currentTask, existingTask: currentTask ?? null,
}); });
currentTask = resolvedTask.task; currentTask = resolvedTask.task;
if (resolvedTask.supersededTask) { if (resolvedTask.supersededTask) {

View File

@@ -490,8 +490,8 @@ export function createMessageRuntime(deps: MessageRuntimeDeps): {
timezone: deps.timezone, timezone: deps.timezone,
missedMessages, missedMessages,
task: hasReviewerLease(chatJid) task: hasReviewerLease(chatJid)
? getLatestOpenPairedTaskForChat(chatJid) ? (getLatestOpenPairedTaskForChat(chatJid) ?? null)
: null, : undefined,
roleToChannel, roleToChannel,
ownerChannel: channel, ownerChannel: channel,
lastAgentTimestamps: deps.getLastAgentTimestamps(), lastAgentTimestamps: deps.getLastAgentTimestamps(),