Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -353,6 +353,7 @@ mock.module("../memory/conversation-crud.js", () => ({
// Always return undefined for the row read so the test fails fast unless
// executeSubagentSpawn reads from context.overrideProfile first.
getConversationOverrideProfile: () => undefined,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

import { getSubagentManager } from "../subagent/index.js";
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ import type { Message } from "../providers/types.js";

mock.module("../memory/conversation-crud.js", () => ({
getConversationOverrideProfile: () => undefined,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../config/loader.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ let mockOverrideProfile: string | undefined = undefined;

mock.module("../memory/conversation-crud.js", () => ({
getConversationOverrideProfile: (_id: string) => mockOverrideProfile,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

// Mutable stub for `getConfig().llm` consumed by `RetryProvider`'s
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/annotate-risk-options.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@ mock.module("../memory/conversation-crud.js", () => ({
updates.push({ id, content });
},
provenanceFromTrustContext: () => ({}),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/llm-request-log-store.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/approval-cascade.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -151,6 +151,7 @@ mock.module("../memory/conversation-crud.js", () => ({
addMessage: () => ({ id: `msg-${Date.now()}` }),
updateConversationUsage: () => {},
updateConversationTitle: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,7 @@ mock.module("../memory/conversation-crud.js", () => ({
setLastNotifiedInferenceProfile: mock(() => {}),
setConversationHistoryStrippedAt: mock(() => {}),
wipeConversation: mock(() => ({ memoryIds: [] })),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-title-service.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/btw-routes.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ const mockAddMessage = mock(() => {});

mock.module("../memory/conversation-crud.js", () => ({
addMessage: mockAddMessage,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

const MOCK_TOOLS = [
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/channel-reply-delivery.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,7 @@ mock.module("../memory/conversation-crud.js", () => ({
: {};
row.metadata = JSON.stringify({ ...existing, ...updates });
},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/attachments-store.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/compaction-events.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -150,6 +150,7 @@ mock.module("../memory/conversation-crud.js", () => ({
addMessage: () => ({ id: `msg-${Date.now()}` }),
updateConversationUsage: () => {},
updateConversationTitle: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ mock.module("../util/logger.js", () => ({

mock.module("../memory/conversation-crud.js", () => ({
getMessages: () => [],
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/attachments-store.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ mock.module("../util/logger.js", () => ({

mock.module("../memory/conversation-crud.js", () => ({
getMessages: () => [],
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/attachments-store.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -138,6 +138,7 @@ mock.module("../memory/conversation-crud.js", () => ({
updateConversationTitle: () => {},
getMessageById: () => null,
getLastUserTimestampBefore: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,7 @@ mock.module("../memory/conversation-crud.js", () => ({
setConversationHistoryStrippedAt: () => {},
updateConversationContextWindow: () => {},
updateConversationSlackContextWatermark: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

import { runAgentLoopImpl } from "../daemon/conversation-agent-loop.js";
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -170,6 +170,7 @@ mock.module("../memory/conversation-crud.js", () => ({
getMessageById: () => null,
getLastUserTimestampBefore: () => 0,
setLastNotifiedInferenceProfile: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-disk-view.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -198,6 +198,7 @@ mock.module("../memory/conversation-crud.js", () => ({
setLastNotifiedInferenceProfile: () => {},
getLastUserTimestampBefore: () => 0,
getConversationOverrideProfileFromRow: () => undefined,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

afterAll(() => {
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-agent-loop.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -197,6 +197,7 @@ mock.module("../memory/conversation-crud.js", () => ({
getConversationOriginChannel: () => null,
getMessageById: () => mockMessageById,
getLastUserTimestampBefore: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

afterAll(() => {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ mock.module("../memory/conversation-crud.js", () => ({
getMessages: mockGetMessages,
createConversation: mockCreateConversation,
addMessage: mockAddMessage,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../export/transcript-formatter.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,7 @@ mock.module("../memory/conversation-crud.js", () => ({
addMessage: async () => ({ id: "persisted-1" }),
setConversationOriginChannelIfUnset: () => {},
setConversationOriginInterfaceIfUnset: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,7 @@ mock.module("../memory/conversation-crud.js", () => ({
addMessage: () => ({ id: `msg-${Date.now()}` }),
updateConversationUsage: () => {},
updateConversationTitle: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ mock.module("../memory/conversation-crud.js", () => ({
},
relinkAttachments: () => 0,
deleteLastExchange: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-lifecycle.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,7 @@ mock.module("../memory/conversation-crud.js", () => ({
},
setConversationOriginChannelIfUnset: () => {},
setConversationOriginInterfaceIfUnset: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -104,6 +104,7 @@ mock.module("../memory/conversation-crud.js", () => ({
},
setConversationOriginChannelIfUnset: () => {},
setConversationOriginInterfaceIfUnset: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -87,6 +87,7 @@ mock.module("../memory/conversation-crud.js", () => ({
setConversationHistoryStrippedAt: () => {},
setConversationOriginChannelIfUnset: () => {},
setConversationOriginInterfaceIfUnset: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-pairing.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,7 @@ mock.module("../memory/conversation-crud.js", () => ({
createConversation: createConversationMock,
addMessage: addMessageMock,
getConversation: getConversationMock,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

/** Simulated bindings for external-conversation-store mock. */
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ mock.module("../memory/conversation-crud.js", () => ({
trustContext: undefined,
}),
addMessage: () => ({ id: "msg-mock" }),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/canonical-guardian-store.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -183,6 +183,7 @@ mock.module("../memory/conversation-crud.js", () => ({
updateConversationTitle: () => {},
getMessageById: () => null,
getLastUserTimestampBefore: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -185,6 +185,7 @@ mock.module("../memory/conversation-crud.js", () => ({
provenanceFromTrustContext: () => ({}),
getMessageById: () => null,
getLastUserTimestampBefore: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-queue.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -193,6 +193,7 @@ mock.module("../memory/conversation-crud.js", () => ({
updateConversationTitle: () => {},
getMessageById: () => null,
getLastUserTimestampBefore: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ mock.module("../memory/conversation-crud.js", () => ({
content: string,
metadata?: Record<string, unknown>,
) => addMessageMock(conversationId, role, content, metadata),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../runtime/local-actor-identity.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,7 @@ mock.module("../memory/conversation-crud.js", () => ({
: { provenanceTrustClass: "unknown" },
setConversationOriginChannelIfUnset: () => {},
setConversationOriginInterfaceIfUnset: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../daemon/conversation-process.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-slash-queue.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -135,6 +135,7 @@ mock.module("../memory/conversation-crud.js", () => ({
updateConversationTitle: () => {},
getMessageById: () => null,
getLastUserTimestampBefore: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-slash-unknown.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -136,6 +136,7 @@ mock.module("../memory/conversation-crud.js", () => ({
updateConversationTitle: () => {},
getMessageById: () => null,
getLastUserTimestampBefore: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -150,6 +150,7 @@ mock.module("../memory/conversation-crud.js", () => ({
addMessage: () => ({ id: `msg-${Date.now()}` }),
updateConversationUsage: () => {},
updateConversationTitle: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ mock.module("../memory/conversation-crud.js", () => ({
getMessages: (conversationId: string) => getMessagesImpl(conversationId),
updateMessageContent: (id: string, content: string) =>
updateMessageContentSpy(id, content),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

// Imports must come AFTER mock.module so the surface module picks up
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-title-service.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ mock.module("../memory/conversation-crud.js", () => ({
getConversation: mockGetConversation,
getMessages: mockGetMessages,
updateConversationTitle: mockUpdateConversationTitle,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../providers/provider-send-message.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/conversation-usage.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ mock.module("../memory/conversation-crud.js", () => ({
estimatedCost,
});
},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

import { recordUsage } from "../daemon/conversation-usage.js";
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -111,6 +111,7 @@ mock.module("../memory/conversation-crud.js", () => ({
updateConversationContextWindow: () => {},
deleteMessageById: () => ({ segmentIds: [], deletedSummaryIds: [] }),
deleteLastExchange: () => 0,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -145,6 +145,7 @@ mock.module("../memory/conversation-crud.js", () => ({
setLastNotifiedInferenceProfile: () => {},
getConversationOverrideProfileFromRow: () => undefined,
updateMessageMetadata: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,7 @@ mock.module("../memory/conversation-crud.js", () => ({
setLastNotifiedInferenceProfile: () => {},
getConversationOverrideProfileFromRow: () => undefined,
updateMessageMetadata: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-queries.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/dm-persistence.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ mock.module("../memory/conversation-crud.js", () => ({
provenanceFromTrustContext: () => ({}),
setConversationOriginChannelIfUnset: () => {},
setConversationOriginInterfaceIfUnset: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-disk-view.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/filing-service.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ mock.module("../memory/conversation-crud.js", () => ({
createdConversations.push(opts);
return { id: `conv-${++conversationIdCounter}`, ...opts };
},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

// Mock logger
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,7 @@ mock.module("../runtime/pending-interactions.js", () => ({

mock.module("../memory/conversation-crud.js", () => ({
addMessage: mock(async () => ({ id: "persisted-message-id" })),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../config/loader.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/heartbeat-disk-pressure.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,7 @@ mock.module("../memory/conversation-crud.js", () => ({
// addMessage. Disk-pressure short-circuits before addMessage ever runs,
// but the mock module must still expose every name the real module does.
addMessage: () => Promise.resolve({ id: "mock-msg-id" }),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

const mockProcessMessage = mock(() => Promise.resolve({ messageId: "msg-1" }));
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/heartbeat-service.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -159,6 +159,7 @@ mock.module("../memory/conversation-crud.js", () => ({
createdConversations.push(opts);
return { id: `conv-${++conversationIdCounter}`, ...opts };
},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

// Mock logger — capture warn calls for unreachable-credential assertions
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/http-user-message-parity.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,7 @@ mock.module("../memory/conversation-crud.js", () => ({
content: string,
metadata?: Record<string, unknown>,
) => addMessageMock(conversationId, role, content, metadata),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../runtime/local-actor-identity.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/inbound-slack-persistence.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@ mock.module("../memory/conversation-crud.js", () => ({
provenanceFromTrustContext: () => ({}),
setConversationOriginChannelIfUnset: () => {},
setConversationOriginInterfaceIfUnset: () => {},
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-disk-view.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/messaging-send-tool.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,7 @@ const getBindingByChannelChatMock = mock(
mock.module("../memory/conversation-crud.js", () => ({
addMessage: addMessageMock,
getConversation: getConversationMock,
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/conversation-disk-view.js", () => ({
Expand Down
1 change: 1 addition & 0 deletions assistant/src/__tests__/outbound-slack-persistence.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -111,6 +111,7 @@ mock.module("../memory/conversation-crud.js", () => ({
// The handler treats provenance as a flat spread; returning {} keeps the
// metadata snapshot focused on the fields under test.
provenanceFromTrustContext: () => ({}),
reserveMessage: mock(async () => ({ id: "msg-reserve" })),
}));

mock.module("../memory/llm-request-log-store.js", () => ({
Expand Down
Loading
Loading